test: fix picker error NodeName issue

unit-mesh · Jan 4, 2024 · 15fe844 · 15fe844
1 parent abc0c96
commit 15fe844
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/rust/RustTestCodeService.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/rust/RustTestCodeService.kt
@@ -33,7 +33,7 @@ class RustTestCodeService(val job: JobContext) : UnitTestService {
 
         val result = testCode.map { codeFunction ->
             codeFunction.FunctionCalls.mapNotNull { codeCall ->
-                val canonicalName = buildCanonicalName(codeCall.NodeName, codeCall.FunctionName)
+                val canonicalName = buildCanonicalName(codeCall.Package, codeCall.FunctionName)
                 val underTestFunction = functionMap.firstNotNullOfOrNull { it[canonicalName] }
                 if (underTestFunction == null) {
                     null
@@ -65,9 +65,9 @@ class RustTestCodeService(val job: JobContext) : UnitTestService {
         return buildCanonicalName(function.Package, function.Name)
     }
 
-    private fun buildCanonicalName(nodeName: String, funcName: String): String {
-        val node = nodeName.ifEmpty { "main" }
-        return "$node::$funcName"
+    private fun buildCanonicalName(pkgName: String, funcName: String): String {
+        val pkg = pkgName.ifEmpty { "main" }
+        return "$pkg::$funcName"
     }
 
     override fun build(dataStruct: CodeDataStruct): List<TypedTestIns> {

diff --git a/...-picker/src/test/kotlin/cc/unitmesh/pick/builder/unittest/rust/RustTestCodeServiceTest.kt b/...-picker/src/test/kotlin/cc/unitmesh/pick/builder/unittest/rust/RustTestCodeServiceTest.kt
@@ -44,7 +44,7 @@ class RustTestCodeServiceTest {
                    assert_eq!(embedding.len(), 128);
                }
            }
-           """.trimIndent()
+       """.trimIndent()
 
         val container = RustAnalyser().analysis(testCode, "lib.rs")
         container.buildSourceCode(testCode.lines())
@@ -70,7 +70,8 @@ class RustTestCodeServiceTest {
         val build: List<BasicTestIns> = rustTestCodeService.build(container) as List<BasicTestIns>
 
         assertEquals(1, build.size)
-        assertEquals("""fn init_semantic(model: Vec<u8>, tokenizer_data: Vec<u8>) -> Result<Arc<Semantic>, SemanticError> {
+        assertEquals(
+            """fn init_semantic(model: Vec<u8>, tokenizer_data: Vec<u8>) -> Result<Arc<Semantic>, SemanticError> {
      let result = Semantic::init_semantic(model, tokenizer_data)?;
      Ok(Arc::new(result))
 }""", build[0].underTestCode)