public InternalTestArtifactExtension(Project project, ProviderFactory providerFactory) {
        this.project = project;
        this.providerFactory = providerFactory;
    }

    public void registerTestArtifactFromSourceSet(SourceSet sourceSet) {
        String name = sourceSet.getName();
        JavaPluginExtension javaPluginExtension = project.getExtensions().getByType(JavaPluginExtension.class);

        SourceSet testSourceSet = project.getExtensions().getByType(SourceSetContainer.class).getByName("test");
        testArtifactExtension.registerTestArtifactFromSourceSet(testSourceSet);
    }