[RIC-249] [RIC-588] US RIC SERVICE UPDATE - Health Check Received | RIC SERVICE UPDAT...

[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..de2a2bc 100644 (file)
--- a/E2Manager/mocks/ran_list_manager_mock.go
+++ b/E2Manager/mocks/ran_list_manager_mock.go
@@ -1,6 +1,7 @@
 //
 // Copyright 2019 AT&T Intellectual Property
 // Copyright 2019 Nokia
+// Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -28,8 +29,50 @@ 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)
+}
+
+func (m *RanListManagerMock) UpdateHealthcheckTimeStampSent(oldRRanName string) (*entities.NbIdentity, *entities.NbIdentity){
+       args := m.Called(oldRRanName)
+       return args.Get(0).(*entities.NbIdentity), args.Get(1).(*entities.NbIdentity)
+}
+
+func (m *RanListManagerMock) UpdateHealthcheckTimeStampReceived(oldRRanName string) (*entities.NbIdentity, *entities.NbIdentity){
+       args := m.Called(oldRRanName)
+       return args.Get(0).(*entities.NbIdentity), args.Get(1).(*entities.NbIdentity)
+}
+
+func (m *RanListManagerMock) UpdateNbIdentities(nodeType entities.Node_Type, oldNbIdentities []*entities.NbIdentity, newNbIdentities []*entities.NbIdentity) error{
+       args:= m.Called(nodeType, oldNbIdentities, newNbIdentities)
+       return args.Error(0)
+}
+