X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fresthooks%2Fresthooks_test.go;h=991fbec309239ac9826093ccc3cb0d15df59cdf4;hb=d6fcfc287d573d94e54a8116a0fc414fe5349a1a;hp=b18b1648ccb85f7b243b3bb337d12e6748b77f78;hpb=f8dc81bc7dd198da9fcb17bef2e164dc71c39e49;p=ric-plt%2Fricdms.git

diff --git a/pkg/resthooks/resthooks_test.go b/pkg/resthooks/resthooks_test.go
index b18b164..991fbec 100644
--- a/pkg/resthooks/resthooks_test.go
+++ b/pkg/resthooks/resthooks_test.go
@@ -28,14 +28,17 @@ import (
 	"net/http"
 	"net/http/httptest"
 	"os"
+	"path/filepath"
 	"strings"
 	"testing"
 
 	ch "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/charts"
+	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/deploy"
 	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/health"
 	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/models"
 	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/onboard"
 	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/charts"
+	d "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/deploy"
 	h "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/health"
 	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/ricdms"
 	"github.com/stretchr/testify/assert"
@@ -55,6 +58,7 @@ func TestMain(m *testing.M) {
 		HealthChecker: HealthCheckerMock{},
 		Onboarder:     onboard.NewOnboarder(),
 		ChartMgr:      ch.NewChartmgr(),
+		DeployMgr:     deploy.NewDeploymentManager(),
 	}
 	code := m.Run()
 	os.Exit(code)
@@ -122,8 +126,9 @@ func TestGetCharts(t *testing.T) {
 	resp := rh.GetCharts()
 	assert.NotEqual(t, nil, resp)
 
-	successResp := resp.(*charts.GetChartsListOK)
-	assert.Equal(t, "SAMPLE_RESPONSE", successResp.Payload)
+	if _, ok := resp.(*charts.GetChartsListOK); !ok {
+		assert.Fail(t, "response type did not match : %t", resp)
+	}
 }
 
 func TestDownloadChart(t *testing.T) {
@@ -144,6 +149,70 @@ func TestDownloadChart(t *testing.T) {
 	assert.IsType(t, &charts.DownloadHelmChartOK{}, resp, "response did not match type")
 }
 
+func TestGetChartsByName(t *testing.T) {
+	svr := httptest.NewUnstartedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		ricdms.Logger.Debug("request received by mock to get chart by name")
+		path, _ := filepath.Abs("./mocks/resp-get-charts-by-name.json")
+		file, err := os.Open(path)
+
+		if err != nil {
+			ricdms.Logger.Error("error in reading file: %v", err)
+		}
+
+		jsonData, err := io.ReadAll(file)
+		if err != nil {
+			ricdms.Logger.Error("Error in rading file: %v", err)
+		}
+		w.Write(jsonData)
+	}))
+
+	svr.Listener.Close()
+	svr.Listener, _ = net.Listen("tcp", ricdms.Config.MockServer)
+
+	svr.Start()
+	defer svr.Close()
+
+	resp := rh.GetChartsByName("TEST_NAME")
+	ricdms.Logger.Debug("resp Data: %s", resp.(*charts.GetChartOK).Payload...)
+	assert.IsType(t, &charts.GetChartOK{}, resp, "response did not match type")
+}
+
+func TestGetChartsByNameAndVersion(t *testing.T) {
+	svr := httptest.NewUnstartedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		ricdms.Logger.Debug("request received by mock to get chart by name and version")
+		path, _ := filepath.Abs("./mocks/resp-get-charts-by-name-and-ver.json")
+		file, err := os.Open(path)
+
+		if err != nil {
+			ricdms.Logger.Error("error in reading file: %v", err)
+		}
+
+		jsonData, err := io.ReadAll(file)
+		if err != nil {
+			ricdms.Logger.Error("Error in rading file: %v", err)
+		}
+		w.Write(jsonData)
+	}))
+
+	svr.Listener.Close()
+	svr.Listener, _ = net.Listen("tcp", ricdms.Config.MockServer)
+
+	svr.Start()
+	defer svr.Close()
+
+	resp := rh.GetChartByNameAndVersion("Test", "1.0.0")
+	ricdms.Logger.Debug("resp data: %s", resp.(*charts.GetChartsFetcherOK).Payload)
+	assert.IsType(t, &charts.GetChartsFetcherOK{}, resp, "response did not match type")
+}
+
+func TestDownloadAndInstall(t *testing.T) {
+	response := rh.DownloadAndInstallChart("sample app", "1.0.0", "test")
+	if _, ok := response.(*d.PostDeployInternalServerError); !ok {
+		assert.Fail(t, "response type did not match (actual) %T", response)
+	}
+
+}
+
 type HealthCheckerMock struct {
 	mock.Mock
 }