X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fmain%2Fhttp_server_test.go;h=760274d71fc7ce2280ef82152d2be29e1e977af7;hb=e3623cf1310f8c8d2fd9b5842102516b9be3b441;hp=aaac4db4d157ecf1a88cd5986dcbf828a27d5c75;hpb=0e58b60a82f5912f815be9292fe0c072a80b79a7;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/main/http_server_test.go b/E2Manager/main/http_server_test.go
index aaac4db..760274d 100644
--- a/E2Manager/main/http_server_test.go
+++ b/E2Manager/main/http_server_test.go
@@ -34,29 +34,43 @@ func setupRouterAndMocks() (*mux.Router, *mocks.ControllerMock, *mocks.NodebCont
 	controllerMock := &mocks.ControllerMock{}
 	controllerMock.On("ShutdownHandler").Return(nil)
 	controllerMock.On("X2ResetHandler").Return(nil)
+	controllerMock.On("X2SetupHandler").Return(nil)
+	controllerMock.On("EndcSetupHandler").Return(nil)
 
 	nodebControllerMock := &mocks.NodebControllerMock{}
-	nodebControllerMock.On("HandleRequest").Return(nil)
 	nodebControllerMock.On("GetNodebIdList").Return(nil)
 	nodebControllerMock.On("GetNodeb").Return(nil)
 	nodebControllerMock.On("HandleHealthCheckRequest").Return(nil)
 
-	router := mux.NewRouter();
+	router := mux.NewRouter()
 	initializeRoutes(router, nodebControllerMock, controllerMock)
 	return router, controllerMock, nodebControllerMock
 }
 
-func TestRoutePostNodebMessageType(t *testing.T) {
-	router, _, nodebControllerMock := setupRouterAndMocks()
+func TestRoutePostEndcSetup(t *testing.T) {
+	router, controllerMock, _ := setupRouterAndMocks()
+
+	req, err := http.NewRequest("POST", "/v1/nodeb/endc-setup", nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	rr := httptest.NewRecorder()
+	router.ServeHTTP(rr, req)
+
+	controllerMock.AssertNumberOfCalls(t,"EndcSetupHandler", 1)
+}
+
+func TestRoutePostX2Setup(t *testing.T) {
+	router, controllerMock, _ := setupRouterAndMocks()
 
-	req, err := http.NewRequest("POST", "/v1/nodeb/messageType", nil)
+	req, err := http.NewRequest("POST", "/v1/nodeb/x2-setup", nil)
 	if err != nil {
 		t.Fatal(err)
 	}
 	rr := httptest.NewRecorder()
 	router.ServeHTTP(rr, req)
 
-	nodebControllerMock.AssertNumberOfCalls(t, "HandleRequest", 1)
+	controllerMock.AssertNumberOfCalls(t,"X2SetupHandler", 1)
 }
 
 func TestRouteGetNodebIds(t *testing.T) {