X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fresthooks%2Fresthooks_test.go;h=40b32ac393c725400b65294b7a7813274ac299a4;hb=c1e4374fb522f55e9263dd2deca3d244e709ebdf;hp=7b65f15959426cf213324891bdb308ff2f60b4c0;hpb=c39ef838fcfa6fd30e0c70a6f4a2abcc4a25643d;p=ric-plt%2Fricdms.git

diff --git a/pkg/resthooks/resthooks_test.go b/pkg/resthooks/resthooks_test.go
index 7b65f15..40b32ac 100644
--- a/pkg/resthooks/resthooks_test.go
+++ b/pkg/resthooks/resthooks_test.go
@@ -42,7 +42,6 @@ import (
 	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"
-	"github.com/stretchr/testify/mock"
 )
 
 var rh *Resthook
@@ -55,7 +54,7 @@ func TestMain(m *testing.M) {
 	}
 	ricdms.Init()
 	rh = &Resthook{
-		HealthChecker: HealthCheckerMock{},
+		HealthChecker: health.NewHealthChecker(),
 		Onboarder:     onboard.NewOnboarder(),
 		ChartMgr:      ch.NewChartmgr(),
 		DeployMgr:     deploy.NewDeploymentManager(),
@@ -78,6 +77,54 @@ func TestHealth(t *testing.T) {
 	}
 }
 
+func TestHealthxAppFail(t *testing.T) {
+	svr := httptest.NewUnstartedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(501)
+	}))
+
+	svr.Listener.Close()
+	svr.Listener, _ = net.Listen("tcp", ricdms.Config.MockServer)
+
+	svr.Start()
+	defer svr.Close()
+
+	resp := rh.GetxAppHealth("test", "test")
+	switch resp.(type) {
+	case *h.GetHealthCheckOK:
+		assert.Fail(t, "Health check should not be okay: %v", resp)
+
+	case *h.GetHealthCheckInternalServerError:
+		break
+
+	default:
+		assert.Fail(t, "Unknown type of resp : %v", resp)
+	}
+}
+
+func TestHealthxApp(t *testing.T) {
+	svr := httptest.NewUnstartedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+	}))
+
+	svr.Listener.Close()
+	svr.Listener, _ = net.Listen("tcp", ricdms.Config.MockServer)
+
+	svr.Start()
+	defer svr.Close()
+
+	resp := rh.GetxAppHealth("test", "test")
+	switch resp.(type) {
+	case *h.GetHealthCheckOK:
+		assert.Equal(t, successStatus, resp.(*h.GetHealthCheckOK).Payload)
+
+	case *h.GetHealthCheckInternalServerError:
+		assert.Fail(t, "Internal Server generated: %v", resp)
+
+	default:
+		assert.Fail(t, "Unknown type of resp : %v", resp)
+	}
+}
+
 func TestOnboard(t *testing.T) {
 	xApp := &models.Descriptor{
 		Config: "SAMPLE_CONFIGURATION",
@@ -219,11 +266,3 @@ func TestUninstallxApp(t *testing.T) {
 		assert.Fail(t, "response type did not match actual: %T", response)
 	}
 }
-
-type HealthCheckerMock struct {
-	mock.Mock
-}
-
-func (h HealthCheckerMock) GetStatus() *models.Status {
-	return successStatus
-}