[RIC-433] Add DeleteEnb implementation

[ric-plt/e2mgr.git] / E2Manager / mocks / rnibWriterMock.go

diff --git a/E2Manager/mocks/rnibWriterMock.go b/E2Manager/mocks/rnibWriterMock.go
index 3503d10..292e455 100644 (file)
--- 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