import org.gradle.tooling.model.idea.IdeaProject;

public class FetchIdeaModel implements BuildAction<IdeaProject> {
    public IdeaProject execute(BuildController controller) {
        return controller.getModel(IdeaProject.class);
    }

public class CustomBuildAction implements BuildAction<EclipseProject> {

    @Override
    public EclipseProject execute(BuildController controller) {
        return controller.getModel(EclipseProject.class);
    }

    void setMessagePrefix(String value)
}

class SomeToolingModelBuildAction implements BuildAction<SomeToolingModel> {
    @Override
    SomeToolingModel execute(BuildController controller) {
        return controller.getModel(SomeToolingModel)
    }

public class ActionDiscardsConfigurationFailure implements BuildAction<String> {
    @Override
    public String execute(BuildController controller) {
        try {
            controller.getModel(GradleProject.class);
            throw new IllegalStateException("should fail");
        } catch (Exception e) {
            // Ignore
        }
        return "result";
    }

import org.gradle.tooling.BuildController;

public class InvalidModel2Action implements BuildAction<CustomModel2> {
    @Override
    public CustomModel2 execute(BuildController controller) {
        return controller.getModel(CustomModel2.class, CustomParameter.class, new Action<CustomParameter>() {
            @Override
            public void execute(CustomParameter customParameter) {}
        });
    }

                        .getCanonicalPath());
            } catch (IOException e) {
                throw new RuntimeException(e);
            }

            project.getModel().addDependency(dependency);
        }

        @Override
        public void afterSessionStart(MavenSession session) {
            session.getUserProperties().setProperty("injected", "bar");
        }
    }

        }

        dependencies {
            implementation(localGroovy())
        }
        """

        when:
        IdeaProject ideaProject = toolingApi.withConnection { connection -> connection.getModel(IdeaProject) }

        then:
        def ideaModule = ideaProject.modules.find { it.name == 'root' }
        def dependency = ideaModule.dependencies.find { it.file.name.contains("groovy-") }

        this.type = type;
        this.value = value;
    }

    @Override
    public T execute(BuildController controller) {
        controller.send(new CustomModel(value));
        return controller.getModel(type);
    }

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

public class HangingBuildAction implements BuildAction<Void> {
    public Void execute(BuildController controller) {
        controller.getModel(GradleProject.class);
        throw new GradleException("Should be cancelled before the end of action.");
    }

    @Override
    public ProjectPublications execute(BuildController controller) {
        GradleBuild buildModel = controller.getBuildModel();
        return controller.getModel(buildModel.getRootProject(), ProjectPublications.class);
    }