E2Manager/managers/ran_status_change_manager_test.go

   1 package managers
   2
   3 import (
   4         "e2mgr/enums"
   5         "e2mgr/logger"
   6         "e2mgr/mocks"
   7         "e2mgr/rmrCgo"
   8         "e2mgr/services/rmrsender"
   9         "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
  10         "github.com/stretchr/testify/assert"
  11         "github.com/stretchr/testify/mock"
  12         "testing"
  13 )
  14
  15 func initRanStatusChangeManagerTest(t *testing.T) (*logger.Logger, *mocks.RmrMessengerMock, *rmrsender.RmrSender) {
  16         logger, err := logger.InitLogger(logger.DebugLevel)
  17         if err != nil {
  18                 t.Fatalf("#initStatusChangeManagerTest - failed to initialize logger, error: %s", err)
  19         }
  20
  21         rmrMessengerMock := &mocks.RmrMessengerMock{}
  22         rmrSender := initRmrSender(rmrMessengerMock, logger)
  23
  24         return logger, rmrMessengerMock, rmrSender
  25 }
  26
  27 func TestMarshalFailure(t *testing.T) {
  28         logger, _, rmrSender := initRanStatusChangeManagerTest(t)
  29         m := NewRanStatusChangeManager(logger, rmrSender)
  30
  31         nodebInfo := entities.NodebInfo{}
  32         err := m.Execute(123, 4, &nodebInfo)
  33
  34         assert.NotNil(t, err)
  35 }
  36
  37 func TestMarshalSuccess(t *testing.T) {
  38         logger, rmrMessengerMock, rmrSender := initRanStatusChangeManagerTest(t)
  39         m := NewRanStatusChangeManager(logger, rmrSender)
  40
  41         nodebInfo := entities.NodebInfo{NodeType: entities.Node_ENB}
  42         var err error
  43         rmrMessengerMock.On("SendMsg", mock.Anything).Return(&rmrCgo.MBuf{}, err)
  44         err  = m.Execute(rmrCgo.RAN_CONNECTED, enums.RIC_TO_RAN, &nodebInfo)
  45
  46         assert.Nil(t, err)
  47 }