X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fservices%2Frnib_data_service_test.go;h=07e2c4548ce6412216b6a33af56aab25ffc462cb;hb=7e2e493acba882641329f867cccb73b3d2a2d2d8;hp=ddc65408086aef1115e800dac525b4956c4f25c3;hpb=14eb5132c3dca8cdad98e7e5420701b04f452046;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/services/rnib_data_service_test.go b/E2Manager/services/rnib_data_service_test.go
index ddc6540..07e2c45 100644
--- a/E2Manager/services/rnib_data_service_test.go
+++ b/E2Manager/services/rnib_data_service_test.go
@@ -262,7 +262,7 @@ func TestSuccessfulUpdateNodebInfoOnConnectionStatusInversion(t *testing.T) {
 	event := "event"
 
 	nodebInfo := &entities.NodebInfo{}
-	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", nodebInfo, "RAN_CONNECTION_STATUS_CHANGE", event).Return(nil)
+	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", nodebInfo, event).Return(nil)
 
 	rnibDataService.UpdateNodebInfoOnConnectionStatusInversion(nodebInfo, event)
 	writerMock.AssertNumberOfCalls(t, "UpdateNodebInfoOnConnectionStatusInversion", 1)
@@ -274,7 +274,7 @@ func TestConnFailureUpdateNodebInfoOnConnectionStatusInversion(t *testing.T) {
 
 	nodebInfo := &entities.NodebInfo{}
 	mockErr := &common.InternalError{Err: &net.OpError{Err: fmt.Errorf("connection error")}}
-	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", nodebInfo, "RAN_CONNECTION_STATUS_CHANGE", event).Return(mockErr)
+	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", nodebInfo, event).Return(mockErr)
 
 	rnibDataService.UpdateNodebInfoOnConnectionStatusInversion(nodebInfo, event)
 	writerMock.AssertNumberOfCalls(t, "UpdateNodebInfoOnConnectionStatusInversion", 3)
@@ -335,3 +335,38 @@ func TestGetE2TInstanceOkOtherError(t *testing.T) {
 	assert.Nil(t, res)
 	assert.NotNil(t, err)
 }
+
+func TestRemoveEnbConnFailure(t *testing.T) {
+	rnibDataService, _, writerMock := setupRnibDataServiceTest(t)
+
+	mockErr := &common.InternalError{Err: &net.OpError{Err: fmt.Errorf("connection error")}}
+	nodebInfo := &entities.NodebInfo{}
+	writerMock.On("RemoveEnb", nodebInfo).Return(mockErr)
+
+	err := rnibDataService.RemoveEnb(nodebInfo)
+	writerMock.AssertNumberOfCalls(t, "RemoveEnb", 3)
+	assert.NotNil(t, err)
+}
+
+func TestRemoveEnbOkNoError(t *testing.T) {
+	rnibDataService, _, writerMock := setupRnibDataServiceTest(t)
+
+	nodebInfo := &entities.NodebInfo{}
+	writerMock.On("RemoveEnb", nodebInfo).Return(nil)
+
+	err := rnibDataService.RemoveEnb(nodebInfo)
+	writerMock.AssertNumberOfCalls(t, "RemoveEnb", 1)
+	assert.Nil(t, err)
+}
+
+func TestRemoveEnbOtherError(t *testing.T) {
+	rnibDataService, _, writerMock := setupRnibDataServiceTest(t)
+
+	mockErr := &common.InternalError{Err: fmt.Errorf("non connection error")}
+	nodebInfo := &entities.NodebInfo{}
+	writerMock.On("RemoveEnb", nodebInfo).Return(mockErr)
+
+	err := rnibDataService.RemoveEnb(nodebInfo)
+	writerMock.AssertNumberOfCalls(t, "RemoveEnb", 1)
+	assert.NotNil(t, err)
+}
\ No newline at end of file