-//func TestOneRanTryShuttingDownSucceedsClearSucceedsRmrSucceedsRanStatusIsAlreadyShutdown(t *testing.T) {
-// h, readerMock, writerMock, rmrMessengerMock, httpClientMock := setupDeleteAllRequestHandlerTest(t)
-// e2tAddresses := []string{E2TAddress}
-
-// readerMock.On("GetE2TAddresses").Return(e2tAddresses , nil)
-// mockHttpClientDissociateAllRans(httpClientMock, e2tAddresses ,true)
-// nbIdentityList := []*entities.NbIdentity{{InventoryName: "RanName_1"}}
-// readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-// nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: E2TAddress}
-// readerMock.On("GetNodeb", "RanName_1").Return(nb1, nil)
-// updatedNb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
-// writerMock.On("UpdateNodebInfo", updatedNb1).Return(nil)
-// readerMock.On("GetE2TAddresses").Return([]string{E2TAddress}, nil)
-// e2tInstance := entities.E2TInstance{Address: E2TAddress, AssociatedRanList: []string{"RanName_1"}}
-// readerMock.On("GetE2TInstances", []string{E2TAddress}).Return([]*entities.E2TInstance{&e2tInstance }, nil)
-// updatedE2tInstance := e2tInstance
-// updatedE2tInstance.AssociatedRanList = []string{}
-// writerMock.On("SaveE2TInstance", &updatedE2tInstance).Return(nil)
-//
-// rmrMessage := models.RmrMessage{MsgType: rmrCgo.RIC_SCTP_CLEAR_ALL}
-// mbuf := rmrCgo.NewMBuf(rmrMessage.MsgType, len(rmrMessage.Payload), rmrMessage.RanName, &rmrMessage.Payload, &rmrMessage.XAction)
-// rmrMessengerMock.On("SendMsg", mbuf, true).Return(mbuf, nil)
-//
-// readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-// nbAfterTimer := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN,}
-// readerMock.On("GetNodeb", /*"RanName_1"*/mock.Anything).Return(nbAfterTimer , nil) // Since this is a second call with same arguments we send mock.Anything due to mock limitations
-// _, err := h.Handle(nil)
-// assert.Nil(t, err)
-// rmrMessengerMock.AssertCalled(t, "SendMsg",mbuf, true)
-// readerMock.AssertExpectations(t)
-// writerMock.AssertExpectations(t)
-// writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
-//}
-
-//func TestOneRanTryShuttingDownSucceedsClearSucceedsRmrSucceedsRanStatusIsShuttingDownUpdateFailure(t *testing.T) {
-// h, readerMock, writerMock, rmrMessengerMock, httpClientMock := setupDeleteAllRequestHandlerTest(t)
-// e2tAddresses := []string{E2TAddress}
-// readerMock.On("GetE2TAddresses").Return(e2tAddresses, nil)
-// mockHttpClientDissociateAllRans(httpClientMock, e2tAddresses, true)
-// nbIdentityList := []*entities.NbIdentity{{InventoryName: "RanName_1"}}
-// readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-// //nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: E2TAddress}
-// //readerMock.On("GetNodeb", "RanName_1").Return(nb1, nil)
-// updatedNb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
-// writerMock.On("UpdateNodebInfo", updatedNb1).Return(nil)
-// readerMock.On("GetE2TAddresses").Return([]string{E2TAddress}, nil)
-// e2tInstance := entities.E2TInstance{Address: E2TAddress, AssociatedRanList: []string{"RanName_1"}}
-// readerMock.On("GetE2TInstances", []string{E2TAddress}).Return([]*entities.E2TInstance{&e2tInstance}, nil)
-// updatedE2tInstance := e2tInstance
-// updatedE2tInstance.AssociatedRanList = []string{}
-// writerMock.On("SaveE2TInstance", &updatedE2tInstance).Return(nil)
-//
-// rmrMessage := models.RmrMessage{MsgType: rmrCgo.RIC_SCTP_CLEAR_ALL}
-// mbuf := rmrCgo.NewMBuf(rmrMessage.MsgType, len(rmrMessage.Payload), rmrMessage.RanName, &rmrMessage.Payload, &rmrMessage.XAction)
-// rmrMessengerMock.On("SendMsg", mbuf, true).Return(mbuf, nil)
-//
-// readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-// readerMock.On("GetNodeb", "RanName_1").Return(updatedNb1, nil)
-// updatedNb2 := *updatedNb1
-// updatedNb2.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
-// writerMock.On("UpdateNodebInfo", &updatedNb2).Return(common.NewInternalError(errors.New("error")))
-// _, err := h.Handle(nil)
-// assert.IsType(t,&e2managererrors.RnibDbError{}, err)
-// rmrMessengerMock.AssertCalled(t, "SendMsg", mbuf, true)
-// readerMock.AssertExpectations(t)
-// writerMock.AssertExpectations(t)
-//}
+func TestOneRanTryShuttingDownSucceedsClearSucceedsRmrSucceedsRanStatusIsShuttingDownUpdateFailure(t *testing.T) {
+ h, readerMock, writerMock, rmrMessengerMock, httpClientMock, ranListManager := setupDeleteAllRequestHandlerTest(t)
+ e2tAddresses := []string{E2TAddress}
+ readerMock.On("GetE2TAddresses").Return(e2tAddresses, nil)
+ mockHttpClientDissociateAllRans(httpClientMock, e2tAddresses, true)
+ oldNbIdentity := &entities.NbIdentity{InventoryName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN, GlobalNbId: &entities.GlobalNbId{PlmnId: "plmnId1", NbId: "nbId1"}}
+ readerMock.On("GetListNodebIds").Return([]*entities.NbIdentity{oldNbIdentity}, nil)
+
+ err := ranListManager.InitNbIdentityMap()
+ if err != nil {
+ t.Errorf("Error cannot init identity")
+ }
+
+ nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN, AssociatedE2TInstanceAddress: E2TAddress, NodeType: entities.Node_GNB}
+ readerMock.On("GetNodeb", "RanName_1").Return(nb1, nil)
+
+ updatedNb1 := *nb1
+ writerMock.On("UpdateNodebInfo", &updatedNb1).Return(nil)
+
+ nodeb1NotAssociated := *nb1
+ nodeb1NotAssociated.AssociatedE2TInstanceAddress = ""
+ nodeb1NotAssociated.ConnectionStatus = entities.ConnectionStatus_SHUTTING_DOWN
+ writerMock.On("UpdateNodebInfo", &nodeb1NotAssociated).Return(nil)
+
+ readerMock.On("GetE2TAddresses").Return([]string{E2TAddress}, nil)
+ e2tInstance := entities.E2TInstance{Address: E2TAddress, AssociatedRanList: []string{"RanName_1"}}
+ readerMock.On("GetE2TInstances", []string{E2TAddress}).Return([]*entities.E2TInstance{&e2tInstance}, nil)
+ updatedE2tInstance := e2tInstance
+ updatedE2tInstance.AssociatedRanList = []string{}
+ writerMock.On("SaveE2TInstance", &updatedE2tInstance).Return(nil)
+
+ newNbIdentity := &entities.NbIdentity{InventoryName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN, GlobalNbId: &entities.GlobalNbId{PlmnId: "plmnId1", NbId: "nbId1"}}
+ writerMock.On("UpdateNbIdentities", updatedNb1.GetNodeType(), []*entities.NbIdentity{oldNbIdentity}, []*entities.NbIdentity{newNbIdentity}).Return(nil)
+
+ rmrMessage := models.RmrMessage{MsgType: rmrCgo.RIC_SCTP_CLEAR_ALL}
+ mbuf := rmrCgo.NewMBuf(rmrMessage.MsgType, len(rmrMessage.Payload), rmrMessage.RanName, &rmrMessage.Payload, &rmrMessage.XAction, rmrMessage.GetMsgSrc())
+ rmrMessengerMock.On("SendMsg", mbuf, true).Return(mbuf, nil)
+
+ readerMock.On("GetListNodebIds").Return([]*entities.NbIdentity{oldNbIdentity}, nil)
+ readerMock.On("GetNodeb", "RanName_1").Return(updatedNb1, nil)
+
+ updatedNb2 := *nb1 //&entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN,}
+ updatedNb2.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
+ updatedNb2.AssociatedE2TInstanceAddress = ""
+ writerMock.On("UpdateNodebInfo", &updatedNb2).Return(common.NewInternalError(errors.New("error")))
+
+ _, err = h.Handle(nil)
+
+ assert.IsType(t, &e2managererrors.RnibDbError{}, err)
+ rmrMessengerMock.AssertCalled(t, "SendMsg", mbuf, true)
+ readerMock.AssertExpectations(t)
+ writerMock.AssertExpectations(t)
+}