diff --git a/src/main/java/io/outfoxx/typescriptpoet/FunctionSpec.kt b/src/main/java/io/outfoxx/typescriptpoet/FunctionSpec.kt index f59c34df..cb5a8128 100644 --- a/src/main/java/io/outfoxx/typescriptpoet/FunctionSpec.kt +++ b/src/main/java/io/outfoxx/typescriptpoet/FunctionSpec.kt @@ -104,7 +104,7 @@ private constructor( codeWriter.emitCode("]") } - if (returnType != null && returnType != TypeName.VOID) { + if (returnType != null) { codeWriter.emitCode(CodeBlock.of(": %T", returnType)) } } diff --git a/src/test/java/io/outfoxx/typescriptpoet/test/FunctionSpecTests.kt b/src/test/java/io/outfoxx/typescriptpoet/test/FunctionSpecTests.kt index 1c8805eb..fd03e050 100644 --- a/src/test/java/io/outfoxx/typescriptpoet/test/FunctionSpecTests.kt +++ b/src/test/java/io/outfoxx/typescriptpoet/test/FunctionSpecTests.kt @@ -197,8 +197,8 @@ class FunctionSpecTests { } @Test - @DisplayName("Generates no return type when void") - fun testGenNoReturnTypeForVoid() { + @DisplayName("Generates return type when void") + fun testGenReturnTypeForVoid() { val testClass = FunctionSpec.builder("test") .returns(TypeName.VOID) .build() @@ -210,7 +210,7 @@ class FunctionSpecTests { out.toString(), equalTo( """ - function test() { + function test(): void { } """.trimIndent()