-//func TestConnFailureThenSuccessGetNodebIdList(t *testing.T) {
-// rnibDataService, readerMock, _ := setupRnibDataServiceTest(t)
-//
-// var nilNodeIds []*entities.NbIdentity = nil
-// nodeIds := []*entities.NbIdentity{}
-// mockErr := &common.InternalError{Err: &net.OpError{Err:fmt.Errorf("connection error")}}
-// //readerMock.On("GetListNodebIds").Return(nilNodeIds, mockErr)
-// //readerMock.On("GetListNodebIds").Return(nodeIds, nil)
-//
-// res, err := rnibDataService.GetListNodebIds()
-// readerMock.AssertNumberOfCalls(t, "GetListNodebIds", 2)
-// assert.True(t, strings.Contains(err.Error(),"connection failure", ))
-// assert.Equal(t, nodeIds, res)
-//}
+func TestSuccessfulUpdateNodebInfoOnConnectionStatusInversion(t *testing.T) {
+ rnibDataService, _, writerMock := setupRnibDataServiceTest(t)
+ event := "event"
+
+ nodebInfo := &entities.NodebInfo{}
+ writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", nodebInfo, event).Return(nil)
+
+ rnibDataService.UpdateNodebInfoOnConnectionStatusInversion(nodebInfo, event)
+ writerMock.AssertNumberOfCalls(t, "UpdateNodebInfoOnConnectionStatusInversion", 1)
+}
+
+func TestConnFailureUpdateNodebInfoOnConnectionStatusInversion(t *testing.T) {
+ rnibDataService, _, writerMock := setupRnibDataServiceTest(t)
+ event := "event"
+
+ nodebInfo := &entities.NodebInfo{}
+ mockErr := &common.InternalError{Err: &net.OpError{Err: fmt.Errorf("connection error")}}
+ writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", nodebInfo, event).Return(mockErr)
+
+ rnibDataService.UpdateNodebInfoOnConnectionStatusInversion(nodebInfo, event)
+ writerMock.AssertNumberOfCalls(t, "UpdateNodebInfoOnConnectionStatusInversion", 3)
+}
+
+func TestGetE2TInstanceConnFailure(t *testing.T) {
+ rnibDataService, readerMock, _ := setupRnibDataServiceTest(t)
+
+ address := "10.10.5.20:3200"
+ var e2tInstance *entities.E2TInstance = nil
+ mockErr := &common.InternalError{Err: &net.OpError{Err: fmt.Errorf("connection error")}}
+ readerMock.On("GetE2TInstance", address).Return(e2tInstance, mockErr)
+
+ res, err := rnibDataService.GetE2TInstance(address)
+ readerMock.AssertNumberOfCalls(t, "GetE2TInstance", 3)
+ assert.Nil(t, res)
+ assert.NotNil(t, err)
+}
+
+func TestGetE2TInstanceOkNoError(t *testing.T) {
+ rnibDataService, readerMock, _ := setupRnibDataServiceTest(t)
+
+ address := "10.10.5.20:3200"
+ e2tInstance := &entities.E2TInstance{}
+ readerMock.On("GetE2TInstance", address).Return(e2tInstance, nil)
+
+ res, err := rnibDataService.GetE2TInstance(address)
+ readerMock.AssertNumberOfCalls(t, "GetE2TInstance", 1)
+ assert.Nil(t, err)
+ assert.Equal(t, e2tInstance, res)
+}
+
+func TestGetE2TInstanceOkOtherError(t *testing.T) {
+ rnibDataService, readerMock, _ := setupRnibDataServiceTest(t)
+
+ address := "10.10.5.20:3200"
+ var e2tInstance *entities.E2TInstance = nil
+ mockErr := &common.InternalError{Err: fmt.Errorf("non connection error")}
+ readerMock.On("GetE2TInstance", address).Return(e2tInstance, mockErr)
+
+ res, err := rnibDataService.GetE2TInstance(address)
+ readerMock.AssertNumberOfCalls(t, "GetE2TInstance", 1)
+ 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)
+}