/*
 * Copyright 2018 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.gradle.api.internal.artifacts.transform;

import org.gradle.internal.execution.history.ExecutionHistoryStore;

import javax.annotation.Nullable;
import java.io.File;

public class TestTransformationWorkspaceProvider implements CachingTransformationWorkspaceProvider {
    private final File transformationsStoreDirectory;
    private final ExecutionHistoryStore executionHistoryStore;

    public TestTransformationWorkspaceProvider(File transformationsStoreDirectory, ExecutionHistoryStore executionHistoryStore) {
        this.transformationsStoreDirectory = transformationsStoreDirectory;
        this.executionHistoryStore = executionHistoryStore;
    }

    @Override
    public ExecutionHistoryStore getExecutionHistoryStore() {
        return executionHistoryStore;
    }

    @Override
    public <T> T withWorkspace(TransformationWorkspaceIdentity identity, TransformationWorkspaceAction<T> workspaceAction) {
        String identityString = identity.getIdentity();
        return workspaceAction.useWorkspace(identityString, new File(transformationsStoreDirectory, identityString));
    }

    @Nullable
    @Override
    public <T> T getCachedResult(TransformationWorkspaceIdentity identity) {
        return null;
    }
}

@@ -664,16 +664,65 @@
 sform;%0A%0A
+import org.gradle.internal.execution.UnitOfWork;%0A
 import o
@@ -1496,39 +1496,27 @@
 rkspace(
-TransformationWorkspace
+UnitOfWork.
 Identity
@@ -1622,24 +1622,24 @@
 tity.get
-Identity
+UniqueId
 ();%0A    
@@ -1820,39 +1820,27 @@
 dResult(
-TransformationWorkspace
+UnitOfWork.
 Identity

Fix compile