[RICPLT-1423] Big Red button API - Implementation - Disconnect status

[ric-plt/e2mgr.git] / E2Manager / controllers / controller_test.go

diff --git a/E2Manager/controllers/controller_test.go b/E2Manager/controllers/controller_test.go
index e753bcf..6bc1613 100644 (file)
--- a/E2Manager/controllers/controller_test.go
+++ b/E2Manager/controllers/controller_test.go
@@ -42,6 +42,7 @@ func TestShutdownHandlerRnibError(t *testing.T) {
        log := initLog(t)
        config := configuration.ParseConfiguration()
 
+       rmrMessengerMock := &mocks.RmrMessengerMock{}
        readerMock := &mocks.RnibReaderMock{}
        readerProvider := func() reader.RNibReader {
                return readerMock
@@ -50,14 +51,13 @@ func TestShutdownHandlerRnibError(t *testing.T) {
        writerProvider := func() rNibWriter.RNibWriter {
                return writerMock
        }
-       var messageChannel chan<- *models.NotificationResponse
 
        rnibErr := &common.RNibError{}
        var nbIdentityList []*entities.NbIdentity
        readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibErr)
 
        writer := httptest.NewRecorder()
-       controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
+       controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
        controller.ShutdownHandler(writer, tests.GetHttpRequest(), nil)
 
        var errorResponse = parseJsonRequest(t, writer.Body)
@@ -70,6 +70,7 @@ func TestHeaderValidationFailed(t *testing.T) {
        log := initLog(t)
        config := configuration.ParseConfiguration()
 
+       rmrMessengerMock := &mocks.RmrMessengerMock{}
        readerMock := &mocks.RnibReaderMock{}
        readerProvider := func() reader.RNibReader {
                return readerMock
@@ -78,11 +79,10 @@ func TestHeaderValidationFailed(t *testing.T) {
        writerProvider := func() rNibWriter.RNibWriter {
                return writerMock
        }
-       var messageChannel chan<- *models.NotificationResponse
 
        writer := httptest.NewRecorder()
 
-       controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
+       controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
 
        header := &http.Header{}
 
@@ -99,6 +99,7 @@ func TestHeaderValidationFailed(t *testing.T) {
 func TestShutdownStatusNoContent(t *testing.T){
        log := initLog(t)
 
+       rmrMessengerMock := &mocks.RmrMessengerMock{}
        readerMock := &mocks.RnibReaderMock{}
        readerProvider := func() reader.RNibReader {
                return readerMock
@@ -108,14 +109,13 @@ func TestShutdownStatusNoContent(t *testing.T){
                return writerMock
        }
        config := configuration.ParseConfiguration()
-       var messageChannel chan<- *models.NotificationResponse
 
        var rnibError common.IRNibError
        nbIdentityList := []*entities.NbIdentity{}
        readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibError)
 
        writer := httptest.NewRecorder()
-       controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
+       controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
        controller.ShutdownHandler(writer, tests.GetHttpRequest(), nil)
 
        assert.Equal(t, http.StatusNoContent, writer.Result().StatusCode)
@@ -125,6 +125,7 @@ func TestHandleInternalError(t *testing.T) {
        log := initLog(t)
        config := configuration.ParseConfiguration()
 
+       rmrMessengerMock := &mocks.RmrMessengerMock{}
        readerMock := &mocks.RnibReaderMock{}
        readerProvider := func() reader.RNibReader {
                return readerMock
@@ -133,10 +134,9 @@ func TestHandleInternalError(t *testing.T) {
        writerProvider := func() rNibWriter.RNibWriter {
                return writerMock
        }
-       var messageChannel chan<- *models.NotificationResponse
 
        writer := httptest.NewRecorder()
-       controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
+       controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
        err := e2managererrors.NewInternalError()
 
        controller.handleErrorResponse(err, writer)
@@ -151,6 +151,7 @@ func TestHandleCommandAlreadyInProgressError(t *testing.T) {
        log := initLog(t)
        config := configuration.ParseConfiguration()
 
+       rmrMessengerMock := &mocks.RmrMessengerMock{}
        readerMock := &mocks.RnibReaderMock{}
        readerProvider := func() reader.RNibReader {
                return readerMock
@@ -159,9 +160,8 @@ func TestHandleCommandAlreadyInProgressError(t *testing.T) {
        writerProvider := func() rNibWriter.RNibWriter {
                return writerMock
        }
-       var messageChannel chan<- *models.NotificationResponse
        writer := httptest.NewRecorder()
-       controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
+       controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
        err := e2managererrors.NewCommandAlreadyInProgressError()
 
        controller.handleErrorResponse(err, writer)
@@ -175,6 +175,7 @@ func TestHandleCommandAlreadyInProgressError(t *testing.T) {
 func TestValidateHeaders(t *testing.T){
        log := initLog(t)
 
+       rmrMessengerMock := &mocks.RmrMessengerMock{}
        readerMock := &mocks.RnibReaderMock{}
        readerProvider := func() reader.RNibReader {
                return readerMock
@@ -184,9 +185,8 @@ func TestValidateHeaders(t *testing.T){
                return writerMock
        }
        config := configuration.ParseConfiguration()
-       var messageChannel chan<- *models.NotificationResponse
 
-       controller := NewController(log, readerProvider, writerProvider, config, messageChannel)
+       controller := NewController(log, getRmrService(rmrMessengerMock, log), readerProvider, writerProvider, config)
 
        header := http.Header{}
        header.Set("Content-Type", "application/json")