Add test for helmmanager

[nonrtric/plt/sme.git] / capifcore / internal / helmmanagement / helm_test.go

diff --git a/capifcore/internal/helmmanagement/helm_test.go b/capifcore/internal/helmmanagement/helm_test.go
new file mode 100644 (file)
index 0000000..6f89c53
--- /dev/null
+++ b/capifcore/internal/helmmanagement/helm_test.go
@@ -0,0 +1,121 @@
+// -
+//   ========================LICENSE_START=================================
+//   O-RAN-SC
+//   %%
+//   Copyright (C) 2022: Nordix Foundation
+//   %%
+//   Licensed under the Apache License, Version 2.0 (the "License");
+//   you may not use this file except in compliance with the License.
+//   You may obtain a copy of the License at
+//
+//        http://www.apache.org/licenses/LICENSE-2.0
+//
+//   Unless required by applicable law or agreed to in writing, software
+//   distributed under the License is distributed on an "AS IS" BASIS,
+//   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+//   See the License for the specific language governing permissions and
+//   limitations under the License.
+//   ========================LICENSE_END===================================
+//
+
+package helmmanagement
+
+import (
+       "bufio"
+       "bytes"
+       "log"
+       "os"
+       "strings"
+       "testing"
+
+       "github.com/stretchr/testify/assert"
+       "github.com/stretchr/testify/mock"
+       "gopkg.in/yaml.v2"
+       "helm.sh/helm/v3/pkg/cli"
+       "helm.sh/helm/v3/pkg/getter"
+       "helm.sh/helm/v3/pkg/repo"
+       "helm.sh/helm/v3/pkg/time"
+       "oransc.org/nonrtric/capifcore/internal/helmmanagement/mocks"
+)
+
+func TestSetUpRepo_repoShouldBeAddedToReposFile(t *testing.T) {
+       settings := createReposFile(t)
+
+       managerUnderTest := NewHelmManager(settings)
+
+       repoName := "repoName"
+       repoURL := "http://url"
+       managerUnderTest.repo = getChartRepo(settings)
+
+       res := managerUnderTest.SetUpRepo(repoName, repoURL)
+
+       assert.Nil(t, res)
+       assert.True(t, containsRepo(settings.RepositoryConfig, repoName))
+}
+
+func createReposFile(t *testing.T) *cli.EnvSettings {
+       reposDir, err := os.MkdirTemp("", "helm")
+       if err != nil {
+               t.Errorf("Unable to create temporary directory for repos due to: %v", err)
+       }
+       t.Cleanup(func() {
+               os.RemoveAll(reposDir)
+       })
+
+       reposFile := reposDir + "/repositories.yaml"
+       settings := &cli.EnvSettings{
+               RepositoryConfig: reposFile,
+       }
+
+       repoData := repo.File{
+               Generated:    time.Now().Time,
+               Repositories: []*repo.Entry{},
+       }
+       data, err := yaml.Marshal(&repoData)
+       if err != nil {
+               assert.Fail(t, "Unable to marshal repo config yaml")
+       }
+       err2 := os.WriteFile(settings.RepositoryConfig, data, 0666)
+       if err2 != nil {
+               assert.Fail(t, "Unable to write repo config file")
+       }
+       return settings
+}
+
+func getChartRepo(settings *cli.EnvSettings) *repo.ChartRepository {
+       repoURL := "http://repoURL"
+       c := repo.Entry{
+               Name: "",
+               URL:  repoURL,
+       }
+       r, _ := repo.NewChartRepository(&c, getter.All(settings))
+       r.Client = getCLientMock(repoURL)
+       return r
+}
+
+func getCLientMock(repoURL string) *mocks.Getter {
+       clientMock := &mocks.Getter{}
+       b := []byte("apiVersion: v1\nentries: {}\ngenerated: \"2022-10-28T09:14:08+02:00\"\nserverInfo: {}\n")
+       z := bytes.NewBuffer(b)
+       clientMock.On("Get", repoURL+"/index.yaml", mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(z, nil)
+       return clientMock
+}
+
+func containsRepo(repoFile, repoName string) bool {
+       file, err := os.Open(repoFile)
+       if err != nil {
+               log.Fatalf("Error opening repository file: %v", err)
+       }
+       defer func(file *os.File) {
+               _ = file.Close()
+       }(file)
+
+       scanner := bufio.NewScanner(file)
+       for scanner.Scan() {
+               row := scanner.Text()
+               if strings.Contains(row, repoName) {
+                       return true
+               }
+       }
+       return false
+}