+
+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)
+}