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=1d458c9838fb8f063585f440f81bedb267127e45;hpb=9c6bf9fab7b6a5a446bb445278b60cbd4312d3b6;p=nonrtric.git

diff --git a/dmaap-mediator-producer/internal/server/server_test.go b/dmaap-mediator-producer/internal/server/server_test.go
index 1d458c98..1db36446 100644
--- a/dmaap-mediator-producer/internal/server/server_test.go
+++ b/dmaap-mediator-producer/internal/server/server_test.go
@@ -78,6 +78,14 @@ func TestNewRouter(t *testing.T) {
 	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) {
@@ -119,7 +127,6 @@ func TestAddInfoJobHandler(t *testing.T) {
 				},
 			},
 			wantedStatus: http.StatusOK,
-			wantedBody:   "",
 		},
 		{
 			name: "AddInfoJobHandler with incorrect job info, should return BadRequest",
@@ -171,6 +178,50 @@ func TestDeleteJob(t *testing.T) {
 	jobHandlerMock.AssertCalled(t, "DeleteJobFromRESTCall", "job1")
 }
 
+func TestSetLogLevel(t *testing.T) {
+	assertions := require.New(t)
+
+	type args struct {
+		logLevel string
+	}
+	tests := []struct {
+		name         string
+		args         args
+		wantedStatus int
+		wantedBody   string
+	}{
+		{
+			name: "Set to valid log level, should return OK",
+			args: args{
+				logLevel: "Debug",
+			},
+			wantedStatus: http.StatusOK,
+		},
+		{
+			name: "Set to invalid log level, should return BadRequest",
+			args: args{
+				logLevel: "bad",
+			},
+			wantedStatus: http.StatusBadRequest,
+			wantedBody:   "Invalid log level: bad",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			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.Equal(tt.wantedStatus, responseRecorder.Code, tt.name)
+			assertions.Contains(responseRecorder.Body.String(), tt.wantedBody, tt.name)
+		})
+	}
+}
+
 func newRequest(method string, url string, jobInfo *jobs.JobInfo, t *testing.T) *http.Request {
 	var body io.Reader
 	if jobInfo != nil {