add unit test for VCSRemoteForImport

dependabot · Feb 11, 2025 · c3f83d8 · c3f83d8
1 parent 5ad920f
commit c3f83d8
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/go_modules/helpers/importresolver/main.go b/go_modules/helpers/importresolver/main.go
@@ -29,6 +29,8 @@ func VCSRemoteForImport(args *Args) (interface{}, error) {
 	if err != nil {
 		return nil, err
 	}
-
+	defer func() {
+		os.RemoveAll(repo.LocalPath())
+	}()
 	return repo.Remote(), nil
 }
diff --git a/go_modules/helpers/importresolver/main_test.go b/go_modules/helpers/importresolver/main_test.go
@@ -0,0 +1,15 @@
+package importresolver
+
+import (
+	"testing"
+)
+
+func TestVCSRemoteForImport(t *testing.T) {
+	args := &Args{
+		Import: "https://github.com/dependabot/dependabot-core",
+	}
+	_, err := VCSRemoteForImport(args)
+	if err != nil {
+		t.Fatalf("failed to get VCS remote for import %s: %v", args.Import, err)
+	}
+}