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

public class FetchTaskAction implements BuildAction<GradleTask> {
    @Override
    public GradleTask execute(BuildController controller) {
        GradleProject project = controller.getModel(GradleProject.class);

@TargetGradleVersion('>=3.1')
class BuildActionCrossVersionSpec extends ToolingApiSpecification {
    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()
    }