X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fmocks%2Fe2t_instances_manager_mock.go;h=0c6cefeb04c487a46d178802c3082c18472d8611;hb=a5f90ced0f8e4e9aeb89de647ca78089a91c1995;hp=8bf6e75417a8bdedb317bed4bed7a3df77c765fc;hpb=e597612ca39c11bb5b562b9e492e5340a618dba9;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/mocks/e2t_instances_manager_mock.go b/E2Manager/mocks/e2t_instances_manager_mock.go
index 8bf6e75..0c6cefe 100644
--- a/E2Manager/mocks/e2t_instances_manager_mock.go
+++ b/E2Manager/mocks/e2t_instances_manager_mock.go
@@ -14,6 +14,10 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 //
+
+//  This source code is part of the near-RT RIC (RAN Intelligent Controller)
+//  platform project (RICP).
+
 package mocks
 
 import (
@@ -36,8 +40,8 @@ func (m *E2TInstancesManagerMock) AddE2TInstance(e2tInstanceAddress string) erro
 	return args.Error(0)
 }
 
-func (m *E2TInstancesManagerMock) RemoveE2TInstance(e2tInstance *entities.E2TInstance) error {
-	args := m.Called(e2tInstance)
+func (m *E2TInstancesManagerMock) RemoveE2TInstance(e2tAddress string) error {
+	args := m.Called(e2tAddress)
 	return args.Error(0)
 }
 
@@ -46,11 +50,12 @@ func (m *E2TInstancesManagerMock) SelectE2TInstance() (string, error) {
 	return args.String(0), args.Error(1)
 }
 
-func (m *E2TInstancesManagerMock) AddRanToInstance(ranName string, e2tAddress string) error {
-	args := m.Called(ranName, e2tAddress)
+func (m *E2TInstancesManagerMock) AddRansToInstance(e2tAddress string, ranNames []string) error {
+	args := m.Called(e2tAddress, ranNames)
 	return args.Error(0)
 
 }
+
 func (m *E2TInstancesManagerMock) RemoveRanFromInstance(ranName string, e2tAddress string) error {
 	args := m.Called(ranName, e2tAddress)
 	return args.Error(0)
@@ -75,8 +80,17 @@ func (m *E2TInstancesManagerMock) ResetKeepAliveTimestamp(e2tAddress string) err
 
 }
 
-func (m *E2TInstancesManagerMock) ActivateE2TInstance(e2tInstance *entities.E2TInstance) error {
-	args := m.Called(e2tInstance)
+func (m *E2TInstancesManagerMock) SetE2tInstanceState(e2tAddress string, currentState entities.E2TInstanceState, newState entities.E2TInstanceState) error {
+	args := m.Called(e2tAddress, currentState, newState)
 	return args.Error(0)
+}
 
-}
\ No newline at end of file
+func (m *E2TInstancesManagerMock) GetE2TAddresses() ([]string, error) {
+	args := m.Called()
+	return args.Get(0).([]string), args.Error(1)
+}
+
+func (m *E2TInstancesManagerMock) ClearRansOfAllE2TInstances() error {
+	args := m.Called()
+	return args.Error(0)
+}