Sync from Azure to LF

[ric-plt/resource-status-manager.git] / RSM / httpserver / http_server_test.go

diff --git a/RSM/httpserver/http_server_test.go b/RSM/httpserver/http_server_test.go
new file mode 100644 (file)
index 0000000..2185220
--- /dev/null
+++ b/RSM/httpserver/http_server_test.go
@@ -0,0 +1,85 @@
+//
+// Copyright 2019 AT&T Intellectual Property
+// Copyright 2019 Nokia
+//
+// 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
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// 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.
+//
+
+package httpserver
+
+import (
+       "github.com/gorilla/mux"
+       "github.com/stretchr/testify/assert"
+       "net/http"
+       "net/http/httptest"
+       "rsm/mocks"
+       "testing"
+       "time"
+)
+
+func setupRouterAndMocks() (*mux.Router, *mocks.RootControllerMock) {
+       rootControllerMock := &mocks.RootControllerMock{}
+       rootControllerMock.On("HandleHealthCheckRequest").Return(nil)
+
+       router := mux.NewRouter()
+       initializeRoutes(router, rootControllerMock)
+       return router, rootControllerMock
+}
+
+func TestRouteGetHealth(t *testing.T) {
+       router, rootControllerMock := setupRouterAndMocks()
+
+       req, err := http.NewRequest("GET", "/v1/health", nil)
+       if err != nil {
+               t.Fatal(err)
+       }
+       rr := httptest.NewRecorder()
+       router.ServeHTTP(rr, req)
+
+       rootControllerMock.AssertNumberOfCalls(t, "HandleHealthCheckRequest", 1)
+}
+
+func TestRouteNotFound(t *testing.T) {
+       router, _ := setupRouterAndMocks()
+
+       req, err := http.NewRequest("GET", "/v1/no/such/route", nil)
+       if err != nil {
+               t.Fatal(err)
+       }
+       rr := httptest.NewRecorder()
+       router.ServeHTTP(rr, req)
+
+       assert.Equal(t, http.StatusNotFound, rr.Code, "handler returned wrong status code")
+}
+
+func TestRunError(t *testing.T) {
+       rootControllerMock := &mocks.RootControllerMock{}
+
+       err := Run(111222333, rootControllerMock)
+
+       assert.NotNil(t, err)
+}
+
+func TestRun(t *testing.T) {
+       rootControllerMock := &mocks.RootControllerMock{}
+       rootControllerMock.On("HandleHealthCheckRequest").Return(nil)
+
+       go Run(11223, rootControllerMock)
+
+       time.Sleep(time.Millisecond * 100)
+       resp, err := http.Get("http://localhost:11223/v1/health")
+       if err != nil {
+               t.Fatalf("failed to perform GET to http://localhost:11223/v1/health")
+       }
+       assert.Equal(t, 200, resp.StatusCode)
+}
\ No newline at end of file