Extract readSource

1 files changed, 8 insertions, 5 deletions

diff --git a/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java b/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
index b40ca3f..8a4c7e6 100644
--- a/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
@@ -828,15 +828,18 @@ public class JavaReaderTests {
 
     private List<ClassNode> readClasses(String classBody) {
         String javaClass = generateClassSource(classBody);
-        try {
+        return readSource("com/example/Example.java", javaClass);
+    }
 
+    private List<ClassNode> readSource(String path, String contents) {
+        try {
             Path directoryPath = Files.createTempDirectory(null);
-            Path sourcePath = directoryPath.resolve("com/example/Example.java");
             try {
-                Files.createDirectories(directoryPath.resolve("com/example"));
-                Files.write(sourcePath, list(javaClass));
+                Path sourcePath = directoryPath.resolve(path);
+                Files.createDirectories(sourcePath.getParent());
+                Files.write(sourcePath, list(contents));
 
-                return JavaReader.readClassFromFile(list(directoryPath), sourcePath);
+                return JavaReader.readClassFromFile(list(directoryPath), directoryPath.resolve(path));
             } finally {
                 deleteRecursively(directoryPath.toFile());
             }