[RIC-430] - Update Connection Status on RAN List

[ric-plt/e2mgr.git] / E2Manager / mocks / ran_list_manager_mock.go

diff --git a/E2Manager/mocks/ran_list_manager_mock.go b/E2Manager/mocks/ran_list_manager_mock.go
index 6df5c4f..9606070 100644 (file)
--- a/E2Manager/mocks/ran_list_manager_mock.go
+++ b/E2Manager/mocks/ran_list_manager_mock.go
@@ -28,8 +28,34 @@ type RanListManagerMock struct {
        mock.Mock
 }
 
+// TODO: remove after replaced with UpdateNbIdentityConnectionStatus
 func (m *RanListManagerMock) UpdateRanState(nodebInfo *entities.NodebInfo) error {
 
        args := m.Called(nodebInfo)
        return args.Error(0)
-}
\ No newline at end of file
+}
+
+func (m *RanListManagerMock) InitNbIdentityMap() error {
+       args := m.Called()
+       return args.Error(0)
+}
+
+func (m *RanListManagerMock) AddNbIdentity(nodeType entities.Node_Type, nbIdentity *entities.NbIdentity) error {
+       args := m.Called(nodeType, nbIdentity)
+       return args.Error(0)
+}
+
+func (m *RanListManagerMock) UpdateNbIdentityConnectionStatus(nodeType entities.Node_Type, ranName string, connectionStatus entities.ConnectionStatus) error {
+       args := m.Called(nodeType, ranName, connectionStatus)
+       return args.Error(0)
+}
+
+func (m *RanListManagerMock) RemoveNbIdentity(nodeType entities.Node_Type, ranName string) error {
+       args := m.Called(nodeType, ranName)
+       return args.Error(0)
+}
+
+func (m *RanListManagerMock) GetNbIdentityList() []*entities.NbIdentity {
+       args := m.Called()
+       return args.Get(0).([]*entities.NbIdentity)
+}