SpringCloud/spring-cloud-contract
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Changed generate method name

Browse Source
This commit is contained in:
Jakub Kubrynski
2014-12-07 00:50:01 +01:00
parent e83455650e
commit 7a68e8d7b7
2 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/groovy/eu/coderate/accurest/TestGenerator.groovy
View File

@@ -4,7 +4,7 @@ import eu.coderate.accurest.builder.ClassBuilder
import eu.coderate.accurest.util.NamesUtil
import static eu.coderate.accurest.builder.ClassBuilder.createClass
import static eu.coderate.accurest.builder.MethodBuilder.createVoidMethod
import static eu.coderate.accurest.builder.MethodBuilder.createTestMethod
/**
* @author Jakub Kubrynski
@@ -50,7 +50,7 @@ class TestGenerator {
.addStaticImport('com.jayway.restassured.matcher.RestAssuredMatchers.*')
.addRule('com.test.MyRule')
directory.listFiles().each {
clazz.addMethod(createVoidMethod(it))
clazz.addMethod(createTestMethod(it))
}
return clazz.build()
}

10
src/main/groovy/eu/coderate/accurest/builder/MethodBuilder.groovy
View File

@@ -9,24 +9,22 @@ import groovy.json.JsonSlurper
class MethodBuilder {
private final String methodName
private final String returnType
private final Map stubContent
private MethodBuilder(String returnType, String methodName, Map stubContent) {
private MethodBuilder(String methodName, Map stubContent) {
this.stubContent = stubContent
this.returnType = returnType
this.methodName = methodName
}
static MethodBuilder createVoidMethod(File stubsFile) {
static MethodBuilder createTestMethod(File stubsFile) {
Map stubContent = new JsonSlurper().parse(stubsFile)
String methodName = NamesUtil.uncapitalize(NamesUtil.toLastDot(NamesUtil.afterLast(stubsFile.path, File.separator)))
return new MethodBuilder("void", methodName, stubContent)
return new MethodBuilder(methodName, stubContent)
}
void appendTo(BlockBuilder blockBuilder) {
blockBuilder.addLine('@Test')
blockBuilder.addLine("public $returnType $methodName() {")
blockBuilder.addLine("public void $methodName() {")
new MethodBodyBuilder(stubContent).appendTo(blockBuilder)
blockBuilder.addLine('}')
}