fix: local module path matching (#202)

Signed-off-by: Fernandez Ludovic <ldez@users.noreply.github.com>
daixiang0 · Apr 3, 2024 · 869c320 · 869c320
1 parent 4f7ff3c
commit 869c320
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/pkg/section/local_module.go b/pkg/section/local_module.go
@@ -18,7 +18,7 @@ type LocalModule struct {
 }
 
 func (m *LocalModule) MatchSpecificity(spec *parse.GciImports) specificity.MatchSpecificity {
-	if strings.HasPrefix(spec.Path, m.Path) {
+	if spec.Path == m.Path || strings.HasPrefix(spec.Path, m.Path+"/") {
 		return specificity.LocalModule{}
 	}
 

diff --git a/pkg/section/local_module_test.go b/pkg/section/local_module_test.go
@@ -0,0 +1,18 @@
+package section
+
+import (
+	"testing"
+
+	"github.com/daixiang0/gci/pkg/specificity"
+)
+
+func TestLocalModule_specificity(t *testing.T) {
+	testCases := []specificityTestData{
+		{"example.com/hello", &LocalModule{Path: "example.com/hello"}, specificity.LocalModule{}},
+		{"example.com/hello/world", &LocalModule{Path: "example.com/hello"}, specificity.LocalModule{}},
+		{"example.com/hello-world", &LocalModule{Path: "example.com/hello"}, specificity.MisMatch{}},
+		{"example.com/helloworld", &LocalModule{Path: "example.com/hello"}, specificity.MisMatch{}},
+		{"example.com", &LocalModule{Path: "example.com/hello"}, specificity.MisMatch{}},
+	}
+	testSpecificity(t, testCases)
+}