fix: fix tests

unit-mesh · Dec 29, 2023 · 0c2d62e · 0c2d62e
1 parent 0513f79
commit 0c2d62e
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 10 deletions.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/strategy/ins/SimilarChunkIns.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/strategy/ins/SimilarChunkIns.kt
@@ -31,7 +31,7 @@ data class SimilarChunkIns(
             } else {
                 similarChunks
             }
-            "\n// Similar chunk:\n ```${language}\n${similarChunksLines.joinToString("\n")}\n```"
+            "\n// Similar chunk:\n```${language}\n${similarChunksLines.joinToString("\n")}\n```"
         } else {
             ""
         }

diff --git a/...ker/src/test/kotlin/cc/unitmesh/pick/builder/strategy/SimilarChunksStrategyBuilderTest.kt b/...ker/src/test/kotlin/cc/unitmesh/pick/builder/strategy/SimilarChunksStrategyBuilderTest.kt
@@ -102,14 +102,33 @@ public class HelloController {
 
         result.size shouldBe 3
 
-        result.first().similarChunks shouldBe """
-            // Compare this snippet from cc.unitmesh.testng.repository.BlogRepository
-            // 
-            // @Repository
-            // public interface BlogRepository extends CrudRepository<BlogPost, Long> {
-            // 
-            // }
-            //
-            """.trimIndent()
+        result.first().unique().input shouldBe """
+// Similar chunk:
+```
+// Compare this snippet from cc.unitmesh.testng.repository.BlogRepository
+// 
+// @Repository
+// public interface BlogRepository extends CrudRepository<BlogPost, Long> {
+// 
+// }
+//
+```
+
+Code:
+```
+package cc.unitmesh.testng.service;
+
+import cc.unitmesh.testng.entity.BlogPost;
+import cc.unitmesh.testng.repository.BlogRepository;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+@Service
+public class BlogService {
+    @Autowired
+    BlogRepository blogRepository;
+
+    public BlogPost createBlog(BlogPost blogDto) {
+```"""
     }
 }