import org.gradle.tooling.BuildAction;
import org.gradle.tooling.BuildController;
import org.gradle.tooling.model.GradleTask;

public class ReturnValueAction implements BuildAction<GradleTask> {
    private final GradleTask task;

    public ReturnValueAction(GradleTask task) {
        this.task = task;
    }

    @Override
    public GradleTask execute(BuildController controller) {
        return task;
    }

    def "can round trip a model queried by a build action"() {
        when:
        GradleTask task = withConnection { c -> c.action(new FetchTaskAction()).run() }
        task = withConnection { c -> c.action(new ReturnValueAction(task)).run() }
        withConnection { c -> c.newBuild().forTasks(task).run() }

        then:
        noExceptionThrown()
    }