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=586f396f28ccbc8d6bf74f3006163321fbfafc43;hpb=697bea2f66010c1c9bb4c3bafa41b3d0ca961291;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/mocks/e2t_instances_manager_mock.go b/E2Manager/mocks/e2t_instances_manager_mock.go
index 586f396..0c6cefe 100644
--- a/E2Manager/mocks/e2t_instances_manager_mock.go
+++ b/E2Manager/mocks/e2t_instances_manager_mock.go
@@ -40,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)
 }
 
@@ -50,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)
@@ -79,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)
+}
+
+func (m *E2TInstancesManagerMock) GetE2TAddresses() ([]string, error) {
+	args := m.Called()
+	return args.Get(0).([]string), args.Error(1)
+}
 
-}
\ No newline at end of file
+func (m *E2TInstancesManagerMock) ClearRansOfAllE2TInstances() error {
+	args := m.Called()
+	return args.Error(0)
+}