X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=dmaap-mediator-producer%2Finternal%2Fserver%2Fserver_test.go;h=1db36446134b623b245838f0dc423850831c0269;hb=5feecd881172a3b22041d35443c1f946e7d5f63e;hp=d221c933e17c32ac432432a14d22afa1561f7ab6;hpb=63a42cacf9c52b7dff64431a3354f55c49bd6e4b;p=nonrtric.git diff --git a/dmaap-mediator-producer/internal/server/server_test.go b/dmaap-mediator-producer/internal/server/server_test.go index d221c933..1db36446 100644 --- a/dmaap-mediator-producer/internal/server/server_test.go +++ b/dmaap-mediator-producer/internal/server/server_test.go @@ -30,16 +30,83 @@ import ( "net/http/httptest" "testing" + "github.com/gorilla/mux" + "github.com/stretchr/testify/mock" "github.com/stretchr/testify/require" "oransc.org/nonrtric/dmaapmediatorproducer/internal/jobs" - "oransc.org/nonrtric/dmaapmediatorproducer/mocks" + "oransc.org/nonrtric/dmaapmediatorproducer/mocks/jobhandler" ) +func TestNewRouter(t *testing.T) { + assertions := require.New(t) + + r := NewRouter(nil) + statusRoute := r.Get("status") + assertions.NotNil(statusRoute) + supportedMethods, err := statusRoute.GetMethods() + assertions.Equal([]string{http.MethodGet}, supportedMethods) + assertions.Nil(err) + path, _ := statusRoute.GetPathTemplate() + assertions.Equal("/status", path) + + addJobRoute := r.Get("add") + assertions.NotNil(addJobRoute) + supportedMethods, err = addJobRoute.GetMethods() + assertions.Equal([]string{http.MethodPost}, supportedMethods) + assertions.Nil(err) + path, _ = addJobRoute.GetPathTemplate() + assertions.Equal("/jobs", path) + + deleteJobRoute := r.Get("delete") + assertions.NotNil(deleteJobRoute) + supportedMethods, err = deleteJobRoute.GetMethods() + assertions.Equal([]string{http.MethodDelete}, supportedMethods) + assertions.Nil(err) + path, _ = deleteJobRoute.GetPathTemplate() + assertions.Equal("/jobs/{infoJobId}", path) + + notFoundHandler := r.NotFoundHandler + handler := http.HandlerFunc(notFoundHandler.ServeHTTP) + responseRecorder := httptest.NewRecorder() + handler.ServeHTTP(responseRecorder, newRequest("GET", "/wrong", nil, t)) + assertions.Equal(http.StatusNotFound, responseRecorder.Code) + assertions.Contains(responseRecorder.Body.String(), "404 not found.") + + methodNotAllowedHandler := r.MethodNotAllowedHandler + handler = http.HandlerFunc(methodNotAllowedHandler.ServeHTTP) + responseRecorder = httptest.NewRecorder() + handler.ServeHTTP(responseRecorder, newRequest(http.MethodPut, "/status", nil, t)) + assertions.Equal(http.StatusMethodNotAllowed, responseRecorder.Code) + assertions.Contains(responseRecorder.Body.String(), "Method is not supported.") + + setLogLevelRoute := r.Get("setLogLevel") + assertions.NotNil(setLogLevelRoute) + supportedMethods, err = setLogLevelRoute.GetMethods() + assertions.Equal([]string{http.MethodPut}, supportedMethods) + assertions.Nil(err) + path, _ = setLogLevelRoute.GetPathTemplate() + assertions.Equal("/admin/log", path) +} + func TestStatusHandler(t *testing.T) { assertions := require.New(t) + + handler := http.HandlerFunc(statusHandler) + responseRecorder := httptest.NewRecorder() + r := newRequest(http.MethodGet, "/status", nil, t) + + handler.ServeHTTP(responseRecorder, r) + + assertions.Equal(http.StatusOK, responseRecorder.Code) + assertions.Equal("", responseRecorder.Body.String()) +} + +func TestAddInfoJobHandler(t *testing.T) { + assertions := require.New(t) + type args struct { - responseRecorder *httptest.ResponseRecorder - r *http.Request + job jobs.JobInfo + mockReturn error } tests := []struct { name string @@ -48,66 +115,74 @@ func TestStatusHandler(t *testing.T) { wantedBody string }{ { - name: "StatusHandler with correct path and method, should return OK", + name: "AddInfoJobHandler with correct job, should return OK", args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("GET", "/", nil, t), + job: jobs.JobInfo{ + Owner: "owner", + LastUpdated: "now", + InfoJobIdentity: "jobId", + TargetUri: "target", + InfoJobData: "{}", + InfoTypeIdentity: "type", + }, }, wantedStatus: http.StatusOK, - wantedBody: "All is well!", }, { - name: "StatusHandler with incorrect path, should return NotFound", + name: "AddInfoJobHandler with incorrect job info, should return BadRequest", args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("GET", "/wrong", nil, t), + job: jobs.JobInfo{ + Owner: "bad", + }, + mockReturn: errors.New("error"), }, - wantedStatus: http.StatusNotFound, - wantedBody: "404 not found.\n", - }, - { - name: "StatusHandler with incorrect method, should return MethodNotAllowed", - args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("PUT", "/", nil, t), - }, - wantedStatus: http.StatusMethodNotAllowed, - wantedBody: "Method is not supported.\n", + wantedStatus: http.StatusBadRequest, + wantedBody: "Invalid job info. Cause: error", }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { - handler := http.HandlerFunc(StatusHandler) - handler.ServeHTTP(tt.args.responseRecorder, tt.args.r) - assertions.Equal(tt.wantedStatus, tt.args.responseRecorder.Code) + jobHandlerMock := jobhandler.JobHandler{} + jobHandlerMock.On("AddJobFromRESTCall", tt.args.job).Return(tt.args.mockReturn) + + callbackHandlerUnderTest := NewProducerCallbackHandler(&jobHandlerMock) - assertions.Equal(tt.wantedBody, tt.args.responseRecorder.Body.String()) + handler := http.HandlerFunc(callbackHandlerUnderTest.addInfoJobHandler) + responseRecorder := httptest.NewRecorder() + r := newRequest(http.MethodPost, "/jobs", &tt.args.job, t) + + handler.ServeHTTP(responseRecorder, r) + + assertions.Equal(tt.wantedStatus, responseRecorder.Code, tt.name) + assertions.Contains(responseRecorder.Body.String(), tt.wantedBody, tt.name) + jobHandlerMock.AssertCalled(t, "AddJobFromRESTCall", tt.args.job) }) } } -func TestCreateInfoJobHandler(t *testing.T) { +func TestDeleteJob(t *testing.T) { + assertions := require.New(t) + jobHandlerMock := jobhandler.JobHandler{} + jobHandlerMock.On("DeleteJobFromRESTCall", mock.Anything).Return(nil) + + callbackHandlerUnderTest := NewProducerCallbackHandler(&jobHandlerMock) + + responseRecorder := httptest.NewRecorder() + r := mux.SetURLVars(newRequest(http.MethodDelete, "/jobs/", nil, t), map[string]string{"infoJobId": "job1"}) + handler := http.HandlerFunc(callbackHandlerUnderTest.deleteInfoJobHandler) + handler.ServeHTTP(responseRecorder, r) + assertions.Equal(http.StatusOK, responseRecorder.Result().StatusCode) + + assertions.Equal("", responseRecorder.Body.String()) + + jobHandlerMock.AssertCalled(t, "DeleteJobFromRESTCall", "job1") +} + +func TestSetLogLevel(t *testing.T) { assertions := require.New(t) - jobHandlerMock := mocks.JobHandler{} - - goodJobInfo := jobs.JobInfo{ - Owner: "owner", - LastUpdated: "now", - InfoJobIdentity: "jobId", - TargetUri: "target", - InfoJobData: "{}", - InfoTypeIdentity: "type", - } - badJobInfo := jobs.JobInfo{ - Owner: "bad", - } - jobHandlerMock.On("AddJob", goodJobInfo).Return(nil) - jobHandlerMock.On("AddJob", badJobInfo).Return(errors.New("error")) - jobs.Handler = &jobHandlerMock type args struct { - responseRecorder *httptest.ResponseRecorder - r *http.Request + logLevel string } tests := []struct { name string @@ -116,49 +191,33 @@ func TestCreateInfoJobHandler(t *testing.T) { wantedBody string }{ { - name: "CreateInfoJobHandler with correct path and method, should return OK", + name: "Set to valid log level, should return OK", args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("POST", "/producer_simulator/info_job", &goodJobInfo, t), + logLevel: "Debug", }, wantedStatus: http.StatusOK, - wantedBody: "", }, { - name: "CreateInfoJobHandler with incorrect job info, should return BadRequest", + name: "Set to invalid log level, should return BadRequest", args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("POST", "/producer_simulator/info_job", &badJobInfo, t), + logLevel: "bad", }, wantedStatus: http.StatusBadRequest, - wantedBody: "Invalid job info. Cause: error", - }, - { - name: "CreateInfoJobHandler with incorrect path, should return NotFound", - args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("GET", "/wrong", nil, t), - }, - wantedStatus: http.StatusNotFound, - wantedBody: "404 not found.", - }, - { - name: "CreateInfoJobHandler with incorrect method, should return MethodNotAllowed", - args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("PUT", "/producer_simulator/info_job", nil, t), - }, - wantedStatus: http.StatusMethodNotAllowed, - wantedBody: "Method is not supported.", + wantedBody: "Invalid log level: bad", }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { - handler := http.HandlerFunc(CreateInfoJobHandler) - handler.ServeHTTP(tt.args.responseRecorder, tt.args.r) - assertions.Equal(tt.wantedStatus, tt.args.responseRecorder.Code) + callbackHandlerUnderTest := NewProducerCallbackHandler(nil) + + handler := http.HandlerFunc(callbackHandlerUnderTest.setLogLevel) + responseRecorder := httptest.NewRecorder() + r, _ := http.NewRequest(http.MethodPut, "/admin/log?level="+tt.args.logLevel, nil) + + handler.ServeHTTP(responseRecorder, r) - assertions.Contains(tt.args.responseRecorder.Body.String(), tt.wantedBody) + assertions.Equal(tt.wantedStatus, responseRecorder.Code, tt.name) + assertions.Contains(responseRecorder.Body.String(), tt.wantedBody, tt.name) }) } }