X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=RSM%2Fhttpserver%2Fhttp_server_test.go;fp=RSM%2Fhttpserver%2Fhttp_server_test.go;h=813ac89df40093a3e64cc68c3b6d2073157b10ff;hb=60652d98d51ee23c1eaca2e8bc2bf19c74c57658;hp=1eb0c5a1d18b0dd2fc106704eeb7befffaf1ab10;hpb=b8d3ff3abf409da49ecab244cd6d2c2124dbce7c;p=ric-plt%2Fresource-status-manager.git

diff --git a/RSM/httpserver/http_server_test.go b/RSM/httpserver/http_server_test.go
index 1eb0c5a..813ac89 100644
--- a/RSM/httpserver/http_server_test.go
+++ b/RSM/httpserver/http_server_test.go
@@ -30,17 +30,32 @@ import (
 	"time"
 )
 
-func setupRouterAndMocks() (*mux.Router, *mocks.RootControllerMock) {
+func setupRouterAndMocks() (*mux.Router, *mocks.RootControllerMock, *mocks.ControllerMock) {
 	rootControllerMock := &mocks.RootControllerMock{}
 	rootControllerMock.On("HandleHealthCheckRequest").Return(nil)
 
+	controllerMock := &mocks.ControllerMock{}
+	controllerMock.On("ResourceStatus").Return(nil)
+
 	router := mux.NewRouter()
-	initializeRoutes(router, rootControllerMock)
-	return router, rootControllerMock
+	initializeRoutes(router, rootControllerMock, controllerMock)
+	return router, rootControllerMock, controllerMock
+}
+func TestResourceStatus(t *testing.T) {
+	router, _, controllerMock := setupRouterAndMocks()
+
+	req, err := http.NewRequest("PUT", "/v1/general/resourcestatus", nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	rr := httptest.NewRecorder()
+	router.ServeHTTP(rr, req)
+
+	controllerMock.AssertNumberOfCalls(t, "ResourceStatus", 1)
 }
 
 func TestRouteGetHealth(t *testing.T) {
-	router, rootControllerMock := setupRouterAndMocks()
+	router, rootControllerMock, _ := setupRouterAndMocks()
 
 	req, err := http.NewRequest("GET", "/v1/health", nil)
 	if err != nil {
@@ -53,7 +68,7 @@ func TestRouteGetHealth(t *testing.T) {
 }
 
 func TestRouteNotFound(t *testing.T) {
-	router, _ := setupRouterAndMocks()
+	router, _, _ := setupRouterAndMocks()
 
 	req, err := http.NewRequest("GET", "/v1/no/such/route", nil)
 	if err != nil {
@@ -66,18 +81,17 @@ func TestRouteNotFound(t *testing.T) {
 }
 
 func TestRunError(t *testing.T) {
-	rootControllerMock := &mocks.RootControllerMock{}
+	_, rootControllerMock, controllerMock := setupRouterAndMocks()
 
-	err := Run(111222333, rootControllerMock)
+	err := Run(111222333, rootControllerMock, controllerMock)
 
 	assert.NotNil(t, err)
 }
 
 func TestRun(t *testing.T) {
-	rootControllerMock := &mocks.RootControllerMock{}
-	rootControllerMock.On("HandleHealthCheckRequest").Return(nil)
+	_, rootControllerMock, controllerMock := setupRouterAndMocks()
 
-	go Run(11223, rootControllerMock)
+	go Run(11223, rootControllerMock, controllerMock)
 
 	time.Sleep(time.Millisecond * 100)
 	resp, err := http.Get("http://localhost:11223/v1/health")