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=5c2027aa9d237a6c92a07ed67572f5b587345f0d;hpb=cc74808f1f5835a0b42c77ced5a585648d6d434c;p=nonrtric.git

diff --git a/dmaap-mediator-producer/internal/server/server_test.go b/dmaap-mediator-producer/internal/server/server_test.go
index 5c2027aa..1d458c98 100644
--- a/dmaap-mediator-producer/internal/server/server_test.go
+++ b/dmaap-mediator-producer/internal/server/server_test.go
@@ -136,7 +136,7 @@ func TestAddInfoJobHandler(t *testing.T) {
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			jobHandlerMock := jobhandler.JobHandler{}
-			jobHandlerMock.On("AddJob", tt.args.job).Return(tt.args.mockReturn)
+			jobHandlerMock.On("AddJobFromRESTCall", tt.args.job).Return(tt.args.mockReturn)
 
 			callbackHandlerUnderTest := NewProducerCallbackHandler(&jobHandlerMock)
 
@@ -148,7 +148,7 @@ func TestAddInfoJobHandler(t *testing.T) {
 
 			assertions.Equal(tt.wantedStatus, responseRecorder.Code, tt.name)
 			assertions.Contains(responseRecorder.Body.String(), tt.wantedBody, tt.name)
-			jobHandlerMock.AssertCalled(t, "AddJob", tt.args.job)
+			jobHandlerMock.AssertCalled(t, "AddJobFromRESTCall", tt.args.job)
 		})
 	}
 }
@@ -156,7 +156,7 @@ func TestAddInfoJobHandler(t *testing.T) {
 func TestDeleteJob(t *testing.T) {
 	assertions := require.New(t)
 	jobHandlerMock := jobhandler.JobHandler{}
-	jobHandlerMock.On("DeleteJob", mock.Anything).Return(nil)
+	jobHandlerMock.On("DeleteJobFromRESTCall", mock.Anything).Return(nil)
 
 	callbackHandlerUnderTest := NewProducerCallbackHandler(&jobHandlerMock)
 
@@ -168,7 +168,7 @@ func TestDeleteJob(t *testing.T) {
 
 	assertions.Equal("", responseRecorder.Body.String())
 
-	jobHandlerMock.AssertCalled(t, "DeleteJob", "job1")
+	jobHandlerMock.AssertCalled(t, "DeleteJobFromRESTCall", "job1")
 }
 
 func newRequest(method string, url string, jobInfo *jobs.JobInfo, t *testing.T) *http.Request {