refactor unit tests

Author: Matthias Balke

plugin-gradle/src/test/java/com/diffplug/gradle/spotless/Antlr4TaskTest.java

@@ -30,50 +30,59 @@
 import com.diffplug.spotless.TestProvisioner;
 import com.diffplug.spotless.antlr4.Antlr4FormatterStep;
 
-public class Antlr4TaskTest extends ResourceHarness {
+public class Antlr4TaskTest extends GradleIntegrationHarness {
 
 	private Provisioner provisioner = TestProvisioner.mavenCentral();
 
-	private SpotlessTask checkTask;
-	private SpotlessTask applyTask;
-
-	@Before
-	public void createTask() {
-		Project project = ProjectBuilder.builder().build();
-		checkTask = project.getTasks().create("checkTaskUnderTest", SpotlessTask.class);
-		checkTask.setCheck();
-		applyTask = project.getTasks().create("applyTaskUnderTest", SpotlessTask.class);
-		applyTask.setApply();
-	}
-
 	@Test(expected = GradleException.class)
 	public void testFormatCheckFail() throws Exception {
-		String unformatted = getTestResource("antlr4/Hello.unformatted.g4");
+		setFile("build.gradle").toLines(
+			"buildscript { repositories { mavenCentral() } }",
+			"plugins {",
+			"    id 'com.diffplug.gradle.spotless'",
+			"}",
+			"spotless {",
+			"    antlr4 {",
+			"        antlr4Formatter()",
+			"    }",
+			"}");
 
-		checkTask.addStep(Antlr4FormatterStep.create(provisioner));
-		checkTask.setTarget(Collections.singleton(setFile("testFile.g4").toContent(unformatted)));
-		execute(checkTask);
+		setFile("src/main/antlr/Hello.g4").toResource("antlr4/Hello.unformatted.g4");
+		gradleRunner().withArguments("spotlessCheck").build();
 	}
 
 	@Test
 	public void testFormatedCheckPass() throws Exception {
-		String formatted = getTestResource("antlr4/Hello.formatted.g4");
+		setFile("build.gradle").toLines(
+			"buildscript { repositories { mavenCentral() } }",
+			"plugins {",
+			"    id 'com.diffplug.gradle.spotless'",
+			"}",
+			"spotless {",
+			"    antlr4 {",
+			"        antlr4Formatter()",
+			"    }",
+			"}");
 
-		checkTask.addStep(Antlr4FormatterStep.create(provisioner));
-		checkTask.setTarget(Collections.singleton(setFile("testFile.g4").toContent(formatted)));
-		execute(checkTask);
+		setFile("src/main/antlr/Hello.g4").toResource("antlr4/Hello.formatted.g4");
+		gradleRunner().withArguments("spotlessCheck").build();
 	}
 
 	@Test
 	public void testFormatApplyPass() throws Exception {
-		String unformatted = getTestResource("antlr4/Hello.unformatted.g4");
-		String formatted = getTestResource("antlr4/Hello.formatted.g4");
-		String testFile = "testFile.g4";
-
-		applyTask.addStep(Antlr4FormatterStep.create(provisioner));
-		applyTask.setTarget(Collections.singleton(setFile(testFile).toContent(unformatted)));
-		execute(applyTask);
+		setFile("build.gradle").toLines(
+			"buildscript { repositories { mavenCentral() } }",
+			"plugins {",
+			"    id 'com.diffplug.gradle.spotless'",
+			"}",
+			"spotless {",
+			"    antlr4 {",
+			"        antlr4Formatter()",
+			"    }",
+			"}");
 
-		assertFile(testFile).hasContent(formatted);
+		setFile("src/main/antlr/Hello.g4").toResource("antlr4/Hello.unformatted.g4");
+		gradleRunner().withArguments("spotlessApply").build();
+		assertFile("src/main/antlr/Hello.g4").sameAsResource("antlr4/Hello.formatted.g4");
 	}
 }