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

diff --git a/E2Manager/mocks/e2t_instances_manager_mock.go b/E2Manager/mocks/e2t_instances_manager_mock.go
index 5919f9c..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,8 @@ 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)
 }