X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fresthooks%2Fresthooks_test.go;h=a40f7d7dd74fe2015f63e3cec2339a4883059433;hb=9dd4aee7e55eccd45714c045a05a7d0d0f140e30;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..a40f7d7 100644
--- a/pkg/resthooks/resthooks_test.go
+++ b/pkg/resthooks/resthooks_test.go
@@ -1,22 +1,23 @@
-//==================================================================================
-//  Copyright (c) 2022 Samsung
+// ==================================================================================
 //
-//   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
+//	Copyright (c) 2022 Samsung
 //
-//       http://www.apache.org/licenses/LICENSE-2.0
+//	 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
 //
-//   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.
+//	     http://www.apache.org/licenses/LICENSE-2.0
 //
-//   This source code is part of the near-RT RIC (RAN Intelligent Controller)
-//   platform project (RICP).
-//==================================================================================
+//	 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.
 //
+//	 This source code is part of the near-RT RIC (RAN Intelligent Controller)
+//	 platform project (RICP).
+//
+// ==================================================================================
 package resthooks
 
 import (
@@ -39,10 +40,10 @@ import (
 	"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"
+	"gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/experiment"
 	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 +56,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 +79,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",
@@ -220,10 +269,24 @@ func TestUninstallxApp(t *testing.T) {
 	}
 }
 
-type HealthCheckerMock struct {
-	mock.Mock
-}
+func TestCustomOnaboardEmptyPkg(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")
+	}))
 
-func (h HealthCheckerMock) GetStatus() *models.Status {
-	return successStatus
+	svr.Listener.Close()
+	svr.Listener, _ = net.Listen("tcp", ricdms.Config.MockServer)
+
+	svr.Start()
+	defer svr.Close()
+
+	resp := rh.Onboarder.CustomOnboard(nil)
+	ricdms.Logger.Debug("resp type:%T", resp)
+
+	switch resp.(type) {
+	case *experiment.PostCustomOnboardInternalServerError:
+		break
+	default:
+		assert.Failf(t, "case 1 response type did not match", "[actual: %T]", resp)
+	}
 }