fix: Extension register

build.gradle.kts

@@ -19,6 +19,9 @@ dependencies {
 
     compileOnly(gradleApi())
     testImplementation(gradleTestKit())
+
+    testImplementation("org.junit.jupiter:junit-jupiter-api:5.8.1")
+    testRuntimeOnly("org.junit.jupiter:junit-jupiter-engine:5.8.1")
 }
 
 gradlePlugin {
@@ -34,6 +37,10 @@ gradlePlugin {
     }
 }
 
+tasks.test {
+    useJUnitPlatform()
+}
+
 publishing {
     repositories {
         mavenLocal()

src/main/java/studio/o7/remora/RemoraPlugin.java

@@ -25,7 +25,7 @@ public void apply(Project project) {
         applyNecessaryPlugins(logger, project);
         applyNecessaryRepositories(logger, project);
 
-        RemoraExtension extension = setupExtension(logger, project.getExtensions());
+        RemoraExtension extension = setupExtension(logger, project);
 
         applyNecessaryProjectConfiguration(logger, project, extension);
         applyJavaPluginConfiguration(logger, project, extension);
@@ -36,9 +36,9 @@ public void apply(Project project) {
         // applyPlaceholderConfiguration(logger, project, extension);
     }
 
-    public static RemoraExtension setupExtension(@NonNull Logger logger, @NonNull ExtensionContainer extensions) {
+    public static RemoraExtension setupExtension(@NonNull Logger logger, @NonNull Project project) {
         logger.info("Setting up the `remora` extension");
-        return extensions.create("remora", RemoraExtension.class);
+        return project.getExtensions().create("remora", RemoraExtension.class, project.getObjects());
     }
 
     public static void applyNecessaryProjectConfiguration(@NonNull Logger logger, @NonNull Project project, @NonNull RemoraExtension extension) {

src/main/java/studio/o7/remora/extensions/RemoraExtension.java

@@ -2,6 +2,7 @@
 
 import lombok.Getter;
 import lombok.Setter;
+import org.gradle.api.Action;
 import org.gradle.api.model.ObjectFactory;
 import org.gradle.jvm.toolchain.JavaLanguageVersion;
 import studio.o7.remora.extensions.framework.FrameworkExtension;
@@ -26,4 +27,8 @@ public class RemoraExtension {
     public RemoraExtension(ObjectFactory factory) {
         this.framework = factory.newInstance(FrameworkExtension.class);
     }
+
+    public void framework(Action<? super FrameworkExtension> action) {
+        action.execute(this.framework);
+    }
 }

src/main/java/studio/o7/remora/extensions/framework/FrameworkExtension.java

@@ -2,6 +2,7 @@
 
 import lombok.Getter;
 import lombok.Setter;
+import org.gradle.api.Action;
 import org.gradle.api.model.ObjectFactory;
 
 import javax.inject.Inject;
@@ -15,4 +16,8 @@ public class FrameworkExtension {
     public FrameworkExtension(ObjectFactory factory) {
         lombok = factory.newInstance(LombokExtension.class);
     }
+
+    public void lombok(Action<? super LombokExtension> action) {
+        action.execute(this.lombok);
+    }
 }

src/test/java/studio/o7/remora/RemoraPluginTest.java

@@ -0,0 +1,8 @@
+package studio.o7.remora;
+
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class RemoraPluginTest {
+
+}

src/test/java/studio/o7/remora/extensions/RemoraExtensionTest.java

@@ -0,0 +1,32 @@
+package studio.o7.remora.extensions;
+
+import org.gradle.api.Project;
+import org.gradle.testfixtures.ProjectBuilder;
+import org.junit.jupiter.api.Test;
+import studio.o7.remora.RemoraPlugin;
+import studio.o7.remora.extensions.framework.FrameworkExtension;
+import studio.o7.remora.extensions.framework.LombokExtension;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class RemoraExtensionTest {
+
+    @Test
+    public void testExtension() {
+        Project project = ProjectBuilder.builder().build();
+        project.getPlugins().apply(RemoraPlugin.class);
+
+        RemoraExtension extension = project.getExtensions().findByType(RemoraExtension.class);
+        assertNotNull(extension);
+
+        FrameworkExtension framework = extension.getFramework();
+        assertNotNull(framework);
+
+        LombokExtension lombok = framework.getLombok();
+        assertNotNull(lombok);
+
+        String version = lombok.getVersion();
+        assertNotNull(version);
+        assertEquals("1.18.36", version);
+    }
+}