X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=dmaap-mediator-producer%2Finternal%2Fserver%2Fserver_test.go;h=1d458c9838fb8f063585f440f81bedb267127e45;hb=46a0fd717e5f49ebae6cb2c4fbcf54f0e329dc86;hp=444deba49ea5168654bee369fbccef1e304240b8;hpb=280385634f160bb78a8944998e9106a2f6549eb0;p=nonrtric.git diff --git a/dmaap-mediator-producer/internal/server/server_test.go b/dmaap-mediator-producer/internal/server/server_test.go index 444deba4..1d458c98 100644 --- a/dmaap-mediator-producer/internal/server/server_test.go +++ b/dmaap-mediator-producer/internal/server/server_test.go @@ -30,84 +30,75 @@ 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/jobhandler" ) -func TestStatusHandler(t *testing.T) { +func TestNewRouter(t *testing.T) { assertions := require.New(t) - type args struct { - responseRecorder *httptest.ResponseRecorder - r *http.Request - } - tests := []struct { - name string - args args - wantedStatus int - wantedBody string - }{ - { - name: "StatusHandler with correct path and method, should return OK", - args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("GET", "/", nil, t), - }, - wantedStatus: http.StatusOK, - wantedBody: "All is well!", - }, - { - name: "StatusHandler with incorrect path, should return NotFound", - args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("GET", "/wrong", nil, t), - }, - 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", - }, - } - 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) - assertions.Equal(tt.wantedBody, tt.args.responseRecorder.Body.String()) - }) - } + 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.") } -func TestCreateInfoJobHandler(t *testing.T) { +func TestStatusHandler(t *testing.T) { assertions := require.New(t) - jobHandlerMock := jobhandler.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 + 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 @@ -116,53 +107,70 @@ func TestCreateInfoJobHandler(t *testing.T) { wantedBody string }{ { - name: "CreateInfoJobHandler with correct path and method, should return OK", + name: "AddInfoJobHandler with correct job, should return OK", args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("POST", "/producer_simulator/info_job", &goodJobInfo, t), + job: jobs.JobInfo{ + Owner: "owner", + LastUpdated: "now", + InfoJobIdentity: "jobId", + TargetUri: "target", + InfoJobData: "{}", + InfoTypeIdentity: "type", + }, }, wantedStatus: http.StatusOK, wantedBody: "", }, { - name: "CreateInfoJobHandler with incorrect job info, should return BadRequest", + name: "AddInfoJobHandler with incorrect job info, should return BadRequest", args: args{ - responseRecorder: httptest.NewRecorder(), - r: newRequest("POST", "/producer_simulator/info_job", &badJobInfo, t), + job: jobs.JobInfo{ + Owner: "bad", + }, + mockReturn: errors.New("error"), }, 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.", - }, } 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) + jobHandlerMock := jobhandler.JobHandler{} + jobHandlerMock.On("AddJobFromRESTCall", tt.args.job).Return(tt.args.mockReturn) + + callbackHandlerUnderTest := NewProducerCallbackHandler(&jobHandlerMock) - assertions.Contains(tt.args.responseRecorder.Body.String(), tt.wantedBody) + 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 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 newRequest(method string, url string, jobInfo *jobs.JobInfo, t *testing.T) *http.Request { var body io.Reader if jobInfo != nil {