X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fresthooks%2Fresthooks_test.go;h=b18b1648ccb85f7b243b3bb337d12e6748b77f78;hb=refs%2Fchanges%2F30%2F9230%2F1;hp=517fb1111d2729927f478c260a19c0c3a07bc6bd;hpb=6e8d970e6ec582bd4473c3fbb6772906658608ca;p=ric-plt%2Fricdms.git

diff --git a/pkg/resthooks/resthooks_test.go b/pkg/resthooks/resthooks_test.go
index 517fb11..b18b164 100644
--- a/pkg/resthooks/resthooks_test.go
+++ b/pkg/resthooks/resthooks_test.go
@@ -22,11 +22,13 @@ package resthooks
 import (
 	"encoding/json"
 	"fmt"
+	"io"
 	"io/ioutil"
 	"net"
 	"net/http"
 	"net/http/httptest"
 	"os"
+	"strings"
 	"testing"
 
 	ch "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/charts"
@@ -124,6 +126,24 @@ func TestGetCharts(t *testing.T) {
 	assert.Equal(t, "SAMPLE_RESPONSE", successResp.Payload)
 }
 
+func TestDownloadChart(t *testing.T) {
+	svr := httptest.NewUnstartedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		ricdms.Logger.Debug("request received by mock to download chart")
+		reader := strings.NewReader("SAMPLE_RESPONSE")
+		data, _ := io.ReadAll(reader)
+		ricdms.Logger.Debug("writing : %+v", data)
+		w.Write(data)
+	}))
+	svr.Listener.Close()
+	svr.Listener, _ = net.Listen("tcp", ricdms.Config.MockServer)
+
+	svr.Start()
+	defer svr.Close()
+
+	resp := rh.DownloadChart("CHART_NAME", "VERSION")
+	assert.IsType(t, &charts.DownloadHelmChartOK{}, resp, "response did not match type")
+}
+
 type HealthCheckerMock struct {
 	mock.Mock
 }