Test for adding repository

pkg/dashboard/objects/repos_test.go

@@ -8,6 +8,8 @@ import (
 	"helm.sh/helm/v3/pkg/repo"
 )
 
+var filePath = "./testdata/repositories.yaml"
+
 func initRepository(t *testing.T, filePath string) *Repositories {
 	t.Helper()
 
@@ -30,8 +32,6 @@ func initRepository(t *testing.T, filePath string) *Repositories {
 
 func TestLoadRepo(t *testing.T) {
 
-	filePath := "./testdata/repositories.yaml"
-
 	res, err := repo.LoadFile(filePath)
 	if err != nil {
 		t.Fatal(err)
@@ -48,8 +48,6 @@ func TestLoadRepo(t *testing.T) {
 }
 
 func TestList(t *testing.T) {
-	filePath := "./testdata/repositories.yaml"
-
 	res, err := repo.LoadFile(filePath)
 	if err != nil {
 		t.Fatal(err)
@@ -65,3 +63,32 @@ func TestList(t *testing.T) {
 
 	assert.Equal(t, len(repos), len(res.Repositories))
 }
+
+func TestAdd(t *testing.T) {
+	testRepoName := "TEST"
+	testRepoUrl := "https://helm.github.io/examples"
+
+	res, err := repo.LoadFile(filePath)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// Delete the repository if already exist
+	res.Remove(testRepoName)
+
+	testRepository := initRepository(t, filePath)
+
+	err = testRepository.Add(testRepoName, testRepoUrl)
+
+	if err != nil {
+		t.Fatal(err, "Failed to add repo")
+	}
+
+	// Reload the file
+	res, err = repo.LoadFile(filePath)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	assert.Equal(t, res.Has(testRepoName), true)
+}

pkg/dashboard/objects/testdata/repositories.yaml

@@ -1,9 +1,30 @@
 apiVersion: ""
 generated: "0001-01-01T00:00:00Z"
 repositories:
-  - name: charts
-    url: "https://charts.helm.sh/stable"
-  - name: firstexample
-    url: "http://firstexample.com"
-  - name: secondexample
-    url: "http://secondexample.com"
+- caFile: ""
+  certFile: ""
+  insecure_skip_tls_verify: false
+  keyFile: ""
+  name: charts
+  pass_credentials_all: false
+  password: ""
+  url: https://charts.helm.sh/stable
+  username: ""
+- caFile: ""
+  certFile: ""
+  insecure_skip_tls_verify: false
+  keyFile: ""
+  name: firstexample
+  pass_credentials_all: false
+  password: ""
+  url: http://firstexample.com
+  username: ""
+- caFile: ""
+  certFile: ""
+  insecure_skip_tls_verify: false
+  keyFile: ""
+  name: secondexample
+  pass_credentials_all: false
+  password: ""
+  url: http://secondexample.com
+  username: ""