From 103c2d5cbfb0f2ecaa418ae8d16bcea68c3ac8ac Mon Sep 17 00:00:00 2001 From: Marcin Zajaczkowski Date: Sun, 15 Feb 2015 22:41:48 +0100 Subject: [PATCH] Add test class suffix generation No tests due to broken windows theory... --- .../main/groovy/io/coderate/accurest/TestGenerator.groovy | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/accurest-core/src/main/groovy/io/coderate/accurest/TestGenerator.groovy b/accurest-core/src/main/groovy/io/coderate/accurest/TestGenerator.groovy index 2042f25461..e13a2ea523 100644 --- a/accurest-core/src/main/groovy/io/coderate/accurest/TestGenerator.groovy +++ b/accurest-core/src/main/groovy/io/coderate/accurest/TestGenerator.groovy @@ -51,7 +51,8 @@ class TestGenerator { if (containsStubs(it)) { def testBaseDir = Paths.get(targetDirectory, NamesUtil.packageToDirectory(packageName)) Files.createDirectories(testBaseDir) - def classPath = Paths.get(testBaseDir.toString(), capitalize(it.name) + getTestClassExtension()).toAbsolutePath() + def classPath = Paths.get(testBaseDir.toString(), capitalize(it.name) + getTestClassSuffix() + getTestClassExtension()) + .toAbsolutePath() def classBytes = buildClass(it, packageName).bytes Files.write(classPath, classBytes, StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING) counter.incrementAndGet() @@ -60,6 +61,10 @@ class TestGenerator { } } + private String getTestClassSuffix() { + return configProperties.targetFramework == TestFramework.SPOCK ? 'Spec' : 'Test' + } + private String getTestClassExtension() { return configProperties.targetFramework == TestFramework.SPOCK ? '.groovy' : '.java' }