feat: <core> add function test

phodal · Feb 2, 2020 · efb00ca · efb00ca
1 parent b1be5fb
commit efb00ca
Show file tree

Hide file tree

Showing 3 changed files with 61 additions and 1 deletion.
diff --git a/src/main/kotlin/domain/core/CodeAnnotation.kt b/src/main/kotlin/domain/core/CodeAnnotation.kt
@@ -2,7 +2,10 @@ package domain.core
 
 data class AnnotationKeyValue(var Key: String = "", var Value: String = "") {}
 
-open class CodeAnnotation(var Name: String = "", var KeyValues: Array<AnnotationKeyValue>) {
+open class CodeAnnotation(
+    var Name: String = "",
+    var KeyValues: Array<AnnotationKeyValue> = arrayOf()
+) {
     open fun isComponentOrRepository(): Boolean {
         return this.Name == "Component" || this.Name == "Repository"
     }

diff --git a/src/main/kotlin/domain/core/CodeFunction.kt b/src/main/kotlin/domain/core/CodeFunction.kt
@@ -31,4 +31,30 @@ open class CodeFunction(
     fun isGetterSetter(): Boolean {
         return this.Name.startsWith("set") || this.Name.startsWith("get")
     }
+
+    fun buildFullMethodName(node: CodeDataStruct): String {
+        return node.Package + "." + node.NodeName + "." + this.Name
+    }
+
+
+    fun getAllCallString(): Array<String> {
+        var calls = arrayOf<String>()
+        for (codeCall in this.FunctionCalls) {
+            if (codeCall.NodeName != "") {
+                calls += codeCall.buildClassFullName()
+            }
+        }
+
+        return calls
+    }
+
+    fun isJUnitTest(): Boolean {
+        var isTest = false
+        for (annotation in this.Annotations) {
+            if (annotation.isIgnoreOrTest()) {
+                return true
+            }
+        }
+        return false
+    }
 }
diff --git a/src/test/kotlin/domain/core/CodeFunctionTest.kt b/src/test/kotlin/domain/core/CodeFunctionTest.kt
@@ -22,4 +22,35 @@ internal class CodeFunctionTest {
         assertEquals(CodeFunction(Name = "getFunc").isGetterSetter(), true)
         assertEquals(CodeFunction(Name = "setFunc").isGetterSetter(), true)
     }
+
+    @Test
+    internal fun shouldEnableBuildFullName() {
+        val dataStruct = CodeDataStruct(Package = "package", NodeName = "nodeName")
+
+        val fullMethodName = CodeFunction(Name = "getFunc").buildFullMethodName(dataStruct)
+
+        assertEquals(fullMethodName, "package.nodeName.getFunc")
+    }
+
+    @Test
+    internal fun shouldGetAllCallStrings() {
+        var calls = arrayOf<CodeCall>()
+        calls += CodeCall(Package = "package", NodeName = "nodeName")
+
+        val allCallString = CodeFunction(Name = "getFunc", FunctionCalls = calls).getAllCallString()
+
+        assertEquals(allCallString.size, 1)
+        assertEquals(allCallString[0], "package.nodeName")
+    }
+
+
+    @Test
+    internal fun shouldHandleJunitTest() {
+        var annotations = arrayOf<CodeAnnotation>()
+        annotations += CodeAnnotation(Name = "Test")
+
+        val isTest = CodeFunction(Name = "getFunc", Annotations = annotations).isJUnitTest()
+
+        assertEquals(isTest, true)
+    }
 }