X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fmocks%2FrnibWriterMock.go;h=292e4558efe8d77aeb142de1a2ab4e2a2b8979ee;hb=611eba27c24f595a57f7c55068605a57788ead95;hp=3503d104b9efa8c918f8e295477faaae7e006ba9;hpb=197657e0e814c871b903ef779ad654b13347e430;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/mocks/rnibWriterMock.go b/E2Manager/mocks/rnibWriterMock.go
index 3503d10..292e455 100644
--- a/E2Manager/mocks/rnibWriterMock.go
+++ b/E2Manager/mocks/rnibWriterMock.go
@@ -92,8 +92,8 @@ func (rnibWriterMock *RnibWriterMock) RemoveServedNrCells(inventoryName string,
 	return args.Error(0)
 }
 
-func (rnibWriterMock *RnibWriterMock) UpdateNodebInfoOnConnectionStatusInversion(nodebInfo *entities.NodebInfo, stateChangeMessageChannel string, event string) error {
-	args := rnibWriterMock.Called(nodebInfo, stateChangeMessageChannel, event)
+func (rnibWriterMock *RnibWriterMock) UpdateNodebInfoOnConnectionStatusInversion(nodebInfo *entities.NodebInfo, event string) error {
+	args := rnibWriterMock.Called(nodebInfo, event)
 
 	return args.Error(0)
 }
@@ -101,5 +101,11 @@ func (rnibWriterMock *RnibWriterMock) UpdateNodebInfoOnConnectionStatusInversion
 func (rnibWriterMock *RnibWriterMock) SaveGeneralConfiguration(config *entities.GeneralConfiguration) error {
 	args := rnibWriterMock.Called(config)
 
+	return args.Error(0)
+}
+
+func (rnibWriterMock *RnibWriterMock) RemoveEnb(nodebInfo *entities.NodebInfo) error {
+	args := rnibWriterMock.Called(nodebInfo)
+
 	return args.Error(0)
 }
\ No newline at end of file