import org.gradle.api.tasks.SourceSetContainer;

import static org.elasticsearch.gradle.internal.test.rest.RestTestUtil.registerTestTask;
import static org.elasticsearch.gradle.internal.test.rest.RestTestUtil.setupTestDependenciesDefaults;

/**
 * Apply this plugin to run the Java based REST tests.
 */
public class InternalJavaRestTestPlugin implements Plugin<Project> {

    public static final String SOURCE_SET_NAME = "javaRestTest";

import org.gradle.api.tasks.SourceSetContainer;

import static org.elasticsearch.gradle.internal.test.rest.RestTestUtil.registerTestTask;
import static org.elasticsearch.gradle.internal.test.rest.RestTestUtil.setupTestDependenciesDefaults;

/**
 * Apply this plugin to run the YAML based REST tests.
 */
public class InternalYamlRestTestPlugin implements Plugin<Project> {

    public static final String SOURCE_SET_NAME = "yamlRestTest";

                }
            });
        });
    }

    /**
     * Setup the dependencies needed for the REST tests.
     */
    public static void setupTestDependenciesDefaults(Project project, SourceSet sourceSet) {
        project.getDependencies().add(sourceSet.getImplementationConfigurationName(), project.project(":test:framework"));
    }

        });

        // create a compileOnly configuration as others might expect it
        project.getConfigurations().create("compileOnly");
        RestTestUtil.setupTestDependenciesDefaults(project, testSourceSet);

        EclipseModel eclipse = project.getExtensions().getByType(EclipseModel.class);
        eclipse.getClasspath().setSourceSets(Arrays.asList(testSourceSet));
        eclipse.getClasspath()