// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
-//
+
+// This source code is part of the near-RT RIC (RAN Intelligent Controller)
+// platform project (RICP).
package httpmsghandlers
import (
+ "bytes"
+ "e2mgr/clients"
"e2mgr/configuration"
"e2mgr/e2managererrors"
"e2mgr/logger"
+ "e2mgr/managers"
"e2mgr/mocks"
"e2mgr/models"
- "e2mgr/rNibWriter"
"e2mgr/rmrCgo"
"e2mgr/services"
- "e2mgr/sessions"
+ "e2mgr/services/rmrsender"
"e2mgr/tests"
- "fmt"
+ "encoding/json"
"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
- "gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
+ "github.com/pkg/errors"
"github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/mock"
- "reflect"
+ "io/ioutil"
+ "net/http"
"testing"
)
-func TestHandleBeforeTimerGetListNodebIdsFailedFlow(t *testing.T){
+const E2TAddress = "10.0.2.15:8989"
+const BaseRMUrl = "http://10.10.2.15:12020/routingmanager"
+
+func setupDeleteAllRequestHandlerTest(t *testing.T) (*DeleteAllRequestHandler, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.RmrMessengerMock, *mocks.HttpClientMock) {
log := initLog(t)
+ config := configuration.ParseConfiguration()
+ config.BigRedButtonTimeoutSec = 1
+ config.RoutingManager.BaseUrl = BaseRMUrl
readerMock := &mocks.RnibReaderMock{}
- readerProvider := func() reader.RNibReader {
- return readerMock
- }
writerMock := &mocks.RnibWriterMock{}
- writerProvider := func() rNibWriter.RNibWriter {
- return writerMock
- }
- rmrMessengerMock := &mocks.RmrMessengerMock{}
-
- config := configuration.ParseConfiguration()
-
- handler := NewDeleteAllRequestHandler(getRmrService(rmrMessengerMock, log), config, writerProvider, readerProvider)
+ rnibDataService := services.NewRnibDataService(log, config, readerMock, writerMock)
- rnibErr := &common.ResourceNotFoundError{}
- var nbIdentityList []*entities.NbIdentity
- readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibErr)
+ rmrMessengerMock := &mocks.RmrMessengerMock{}
+ rmrSender := getRmrSender(rmrMessengerMock, log)
- expected := &e2managererrors.RnibDbError{}
- actual := handler.Handle(log, nil)
- if reflect.TypeOf(actual) != reflect.TypeOf(expected){
- t.Errorf("Error actual = %v, and Expected = %v.", actual, expected)
- }
+ e2tInstancesManager := managers.NewE2TInstancesManager(rnibDataService, log)
+ httpClientMock := &mocks.HttpClientMock{}
+ rmClient := clients.NewRoutingManagerClient(log, config, httpClientMock)
+ handler := NewDeleteAllRequestHandler(log, rmrSender, config, rnibDataService, e2tInstancesManager, rmClient)
+ return handler, readerMock, writerMock, rmrMessengerMock, httpClientMock
}
-func TestHandleAfterTimerGetListNodebIdsFailedFlow(t *testing.T){
- log := initLog(t)
+func mapE2TAddressesToE2DataList(e2tAddresses []string) models.RoutingManagerE2TDataList {
+ e2tDataList := make(models.RoutingManagerE2TDataList, len(e2tAddresses))
- readerMock := &mocks.RnibReaderMock{}
- readerProvider := func() reader.RNibReader {
- return readerMock
+ for i, v := range e2tAddresses {
+ e2tDataList[i] = models.NewRoutingManagerE2TData(v)
}
- writerMock := &mocks.RnibWriterMock{}
- writerProvider := func() rNibWriter.RNibWriter {
- return writerMock
- }
- rmrMessengerMock := &mocks.RmrMessengerMock{}
- config := configuration.ParseConfiguration()
- config.BigRedButtonTimeoutSec = 1
-
- handler := NewDeleteAllRequestHandler(getRmrService(rmrMessengerMock, log), config, writerProvider, readerProvider)
-
- rnibErr := &common.ResourceNotFoundError{}
- //Before timer: Disconnected->ShutDown, ShuttingDown->Ignore, Connected->ShuttingDown
- nbIdentityList := createIdentityList()
-
- readerMock.On("GetListNodebIds").Return(nbIdentityList, nil).Return(nbIdentityList, rnibErr)
- nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED,}
- nb2 := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- nb3 := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus:entities.ConnectionStatus_CONNECTED,}
- readerMock.On("GetNodeb", "RanName_1").Return(nb1, nil)
- readerMock.On("GetNodeb", "RanName_2").Return(nb2, nil)
- readerMock.On("GetNodeb", "RanName_3").Return(nb3, nil)
-
- updatedNb1 := &entities.NodebInfo{RanName:"RanName_1", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- updatedNb3 := &entities.NodebInfo{RanName:"RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb1).Return(nil)
- writerMock.On("SaveNodeb", mock.Anything, updatedNb3).Return(nil)
-
- expected := &e2managererrors.RnibDbError{}
- actual := handler.Handle(log, nil)
+ return e2tDataList
+}
- if reflect.TypeOf(actual) != reflect.TypeOf(expected){
- t.Errorf("Error actual = %v, and Expected = %v.", actual, expected)
+func mockHttpClientDissociateAllRans(httpClientMock *mocks.HttpClientMock, e2tAddresses []string, ok bool) {
+ data := mapE2TAddressesToE2DataList(e2tAddresses)
+ marshaled, _ := json.Marshal(data)
+ body := bytes.NewBuffer(marshaled)
+ url := BaseRMUrl + clients.DissociateRanE2TInstanceApiSuffix
+ respBody := ioutil.NopCloser(bytes.NewBufferString(""))
+
+ var status int
+ if ok {
+ status = http.StatusOK
+ } else {
+ status = http.StatusBadRequest
}
+ httpClientMock.On("Post", url, "application/json", body).Return(&http.Response{StatusCode: status, Body: respBody}, nil)
}
-func TestHandleSuccessFlow(t *testing.T){
- log := initLog(t)
-
- readerMock := &mocks.RnibReaderMock{}
- readerProvider := func() reader.RNibReader {
- return readerMock
- }
- writerMock := &mocks.RnibWriterMock{}
- writerProvider := func() rNibWriter.RNibWriter {
- return writerMock
- }
- rmrMessengerMock := &mocks.RmrMessengerMock{}
- config := configuration.ParseConfiguration()
- config.BigRedButtonTimeoutSec = 1
- handler := NewDeleteAllRequestHandler(getRmrService(rmrMessengerMock, log), config, writerProvider, readerProvider)
+func TestGetE2TAddressesFailure(t *testing.T) {
+ h, readerMock, _, _, _ := setupDeleteAllRequestHandlerTest(t)
+ readerMock.On("GetE2TAddresses").Return([]string{}, common.NewInternalError(errors.New("error")))
+ _, err := h.Handle(nil)
+ assert.IsType(t, &e2managererrors.RnibDbError{}, err)
+ readerMock.AssertExpectations(t)
+}
- //Before timer: Disconnected->ShutDown, ShuttingDown->Ignore, Connected->ShuttingDown
- nbIdentityList := createIdentityList()
+func TestOneRanGetE2TAddressesEmptyList(t *testing.T) {
+ h, readerMock, writerMock, _, _ := setupDeleteAllRequestHandlerTest(t)
+ readerMock.On("GetE2TAddresses").Return([]string{}, nil)
+ nbIdentityList := []*entities.NbIdentity{{InventoryName: "RanName_1"}}
readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-
nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED,}
- nb2 := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- nb3 := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus:entities.ConnectionStatus_CONNECTED,}
readerMock.On("GetNodeb", "RanName_1").Return(nb1, nil)
- readerMock.On("GetNodeb", "RanName_2").Return(nb2, nil)
- readerMock.On("GetNodeb", "RanName_3").Return(nb3, nil)
-
- updatedNb1 := &entities.NodebInfo{RanName:"RanName_1", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- updatedNb3 := &entities.NodebInfo{RanName:"RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb1).Return(nil)
- writerMock.On("SaveNodeb", mock.Anything, updatedNb3).Return(nil)
-
- //after timer: ShutDown->Ignore, ShuttingDown->ShutDown
- readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-
- nb1AfterTimer := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN,}
- nb2AfterTimer := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- nb3AfterTimer := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- readerMock.On("GetNodeb", "RanName_1").Return(nb1AfterTimer, nil)
- readerMock.On("GetNodeb", "RanName_2").Return(nb2AfterTimer, nil)
- readerMock.On("GetNodeb", "RanName_3").Return(nb3AfterTimer, nil)
-
- updatedNb2AfterTimer := &entities.NodebInfo{RanName:"RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- updatedNb3AfterTimer := &entities.NodebInfo{RanName:"RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb2AfterTimer).Return(nil)
- writerMock.On("SaveNodeb", mock.Anything, updatedNb3AfterTimer).Return(nil)
-
- mbuf := rmrCgo.NewMBuf(tests.MessageType, tests.MaxMsgSize,"RanName" , &tests.DummyPayload, &tests.DummyXAction)
- rmrMessengerMock.On("SendMsg", mock.AnythingOfType(fmt.Sprintf("%T", mbuf)), tests.MaxMsgSize).Return(mbuf, nil)
-
- actual := handler.Handle(log, nil)
-
- assert.Nil(t, actual)
+ updatedNb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb1).Return(nil)
+ _, err := h.Handle(nil)
+ assert.Nil(t, err)
+ readerMock.AssertExpectations(t)
+ writerMock.AssertExpectations(t)
}
-func TestHandleSuccessGetNextStatusFlow(t *testing.T){
- log := initLog(t)
-
- readerMock := &mocks.RnibReaderMock{}
- readerProvider := func() reader.RNibReader {
- return readerMock
- }
- writerMock := &mocks.RnibWriterMock{}
- writerProvider := func() rNibWriter.RNibWriter {
- return writerMock
- }
- rmrMessengerMock := &mocks.RmrMessengerMock{}
- config := configuration.ParseConfiguration()
- config.BigRedButtonTimeoutSec = 1
- handler := NewDeleteAllRequestHandler(getRmrService(rmrMessengerMock, log), config, writerProvider, readerProvider)
-
- nbIdentityList := []*entities.NbIdentity{{InventoryName: "RanName_1"}}
+func TestTwoRansGetE2TAddressesEmptyListOneGetNodebFailure(t *testing.T) {
+ h, readerMock, writerMock, _, _ := setupDeleteAllRequestHandlerTest(t)
+ readerMock.On("GetE2TAddresses").Return([]string{}, nil)
+ nbIdentityList := []*entities.NbIdentity{{InventoryName: "RanName_1"}, {InventoryName: "RanName_2"}}
readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-
- nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_CONNECTED,}
+ nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED,}
readerMock.On("GetNodeb", "RanName_1").Return(nb1, nil)
+ updatedNb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb1).Return(nil)
+ var nb2 *entities.NodebInfo
+ readerMock.On("GetNodeb", "RanName_2").Return(nb2, common.NewInternalError(errors.New("error")))
+ _, err := h.Handle(nil)
+ assert.IsType(t,&e2managererrors.RnibDbError{}, err)
+ writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 1)
+ readerMock.AssertExpectations(t)
+}
- updatedNb1 := &entities.NodebInfo{RanName:"RanName_1", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb1).Return(nil)
-
- //after timer: ShutDown->Ignore, ShuttingDown->ShutDown
+func TestTwoRansGetE2TAddressesEmptyListOneUpdateNodebInfoFailure(t *testing.T) {
+ h, readerMock, writerMock, _, _ := setupDeleteAllRequestHandlerTest(t)
+ readerMock.On("GetE2TAddresses").Return([]string{}, nil)
+ nbIdentityList := []*entities.NbIdentity{{InventoryName: "RanName_1"}, {InventoryName: "RanName_2"}}
readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
- nb1AfterTimer := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
- readerMock.On("GetNodeb", "RanName_1").Return(nb1AfterTimer, nil)
-
- updatedNb1AfterTimer := &entities.NodebInfo{RanName:"RanName_1", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb1AfterTimer).Return(nil)
-
- mbuf := rmrCgo.NewMBuf(tests.MessageType, tests.MaxMsgSize,"RanName" , &tests.DummyPayload, &tests.DummyXAction)
- rmrMessengerMock.On("SendMsg", mock.AnythingOfType(fmt.Sprintf("%T", mbuf)), tests.MaxMsgSize).Return(mbuf, nil)
+ nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED,}
+ readerMock.On("GetNodeb", "RanName_1").Return(nb1, nil)
+ updatedNb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb1).Return(nil)
- actual := handler.Handle(log, nil)
+ nb2 := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED,}
+ readerMock.On("GetNodeb", "RanName_2").Return(nb2, nil)
+ updatedNb2 := &entities.NodebInfo{RanName: "RanName_2", 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)
+ readerMock.AssertExpectations(t)
+ writerMock.AssertExpectations(t)
+}
- assert.Nil(t, actual)
+func TestOneRanDissociateSucceedsTryShuttingDownFailure(t *testing.T) {
+ h, readerMock, writerMock, _, httpClientMock := setupDeleteAllRequestHandlerTest(t)
+ e2tAddresses := []string{E2TAddress}
+ readerMock.On("GetE2TAddresses").Return(e2tAddresses, nil)
+ mockHttpClientDissociateAllRans(httpClientMock, e2tAddresses, true)
+ nbIdentityList := []*entities.NbIdentity{}
+ readerMock.On("GetListNodebIds").Return(nbIdentityList, common.NewInternalError(errors.New("error")))
+ _, err := h.Handle(nil)
+ assert.IsType(t, &e2managererrors.RnibDbError{}, err)
+ readerMock.AssertExpectations(t)
+ writerMock.AssertExpectations(t)
}
-func TestHandleShuttingDownStatusFlow(t *testing.T){
- log := initLog(t)
+func TestOneRanDissociateFailsTryShuttingDownFailure(t *testing.T) {
+ h, readerMock, writerMock, _, httpClientMock := setupDeleteAllRequestHandlerTest(t)
+ e2tAddresses := []string{E2TAddress}
- readerMock := &mocks.RnibReaderMock{}
- readerProvider := func() reader.RNibReader {
- return readerMock
- }
- writerMock := &mocks.RnibWriterMock{}
- writerProvider := func() rNibWriter.RNibWriter {
- return writerMock
- }
- rmrMessengerMock := &mocks.RmrMessengerMock{}
- config := configuration.ParseConfiguration()
- config.BigRedButtonTimeoutSec = 1
- handler := NewDeleteAllRequestHandler(getRmrService(rmrMessengerMock, log), config, writerProvider, readerProvider)
+ readerMock.On("GetE2TAddresses").Return(e2tAddresses, nil)
+ mockHttpClientDissociateAllRans(httpClientMock, e2tAddresses, false)
+ nbIdentityList := []*entities.NbIdentity{}
+ readerMock.On("GetListNodebIds").Return(nbIdentityList, common.NewInternalError(errors.New("error")))
+ _, err := h.Handle(nil)
+ assert.IsType(t, &e2managererrors.RnibDbError{}, err)
+ readerMock.AssertExpectations(t)
+ writerMock.AssertExpectations(t)
+}
+func TestOneRanTryShuttingDownSucceedsClearFails(t *testing.T) {
+ h, readerMock, writerMock, _, 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_SHUTTING_DOWN,}
+ nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: E2TAddress}
readerMock.On("GetNodeb", "RanName_1").Return(nb1, nil)
-
- //after timer: ShutDown->Ignore, ShuttingDown->ShutDown
- readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-
- nb1AfterTimer := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
- readerMock.On("GetNodeb", "RanName_1").Return(nb1AfterTimer, nil)
-
- updatedNb1AfterTimer := &entities.NodebInfo{RanName:"RanName_1", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb1AfterTimer).Return(nil)
-
- mbuf := rmrCgo.NewMBuf(tests.MessageType, tests.MaxMsgSize,"RanName" , &tests.DummyPayload, &tests.DummyXAction)
- rmrMessengerMock.On("SendMsg", mock.AnythingOfType(fmt.Sprintf("%T", mbuf)), tests.MaxMsgSize).Return(mbuf, nil)
-
- actual := handler.Handle(log, nil)
-
- assert.Nil(t, actual)
+ updatedNb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb1).Return(nil)
+ readerMock.On("GetE2TAddresses").Return([]string{E2TAddress}, nil)
+ readerMock.On("GetE2TInstances", []string{E2TAddress}).Return([]*entities.E2TInstance{}, common.NewInternalError(errors.New("error")))
+ _, err := h.Handle(nil)
+ assert.IsType(t, &e2managererrors.RnibDbError{}, err)
+ readerMock.AssertExpectations(t)
+ writerMock.AssertExpectations(t)
}
-func TestHandleGetNodebFailedFlow(t *testing.T){
- log := initLog(t)
-
- readerMock := &mocks.RnibReaderMock{}
- readerProvider := func() reader.RNibReader {
- return readerMock
- }
- writerMock := &mocks.RnibWriterMock{}
- writerProvider := func() rNibWriter.RNibWriter {
- return writerMock
- }
- rmrMessengerMock := &mocks.RmrMessengerMock{}
- config := configuration.ParseConfiguration()
- config.BigRedButtonTimeoutSec = 1
- handler := NewDeleteAllRequestHandler(getRmrService(rmrMessengerMock, log), config, writerProvider, readerProvider)
-
- //Before timer: Disconnected->ShutDown(will fail), ShuttingDown->Ignore, Connected->ShuttingDown
- nbIdentityList := createIdentityList()
- readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-
- errRnib := &common.ResourceNotFoundError{}
- nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED,}
- nb2 := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- nb3 := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus:entities.ConnectionStatus_CONNECTED,}
- readerMock.On("GetNodeb", "RanName_1").Return(nb1, errRnib)
- readerMock.On("GetNodeb", "RanName_2").Return(nb2, nil)
- readerMock.On("GetNodeb", "RanName_3").Return(nb3, nil)
-
- updatedNb1 := &entities.NodebInfo{RanName:"RanName_1", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- updatedNb3 := &entities.NodebInfo{RanName:"RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb1).Return(errRnib)
- writerMock.On("SaveNodeb", mock.Anything, updatedNb3).Return(nil)
-
- //after timer: ShutDown->Ignore, ShuttingDown->ShutDown
+func TestOneRanTryShuttingDownSucceedsClearSucceedsRmrSendFails(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)
-
- nb1AfterTimer := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN,}
- nb2AfterTimer := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- nb3AfterTimer := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- readerMock.On("GetNodeb", "RanName_1").Return(nb1AfterTimer, errRnib)
- readerMock.On("GetNodeb", "RanName_2").Return(nb2AfterTimer, nil)
- readerMock.On("GetNodeb", "RanName_3").Return(nb3AfterTimer, nil)
-
- updatedNb2AfterTimer := &entities.NodebInfo{RanName:"RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- updatedNb3AfterTimer := &entities.NodebInfo{RanName:"RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb2AfterTimer).Return(nil)
- writerMock.On("SaveNodeb", mock.Anything, updatedNb3AfterTimer).Return(nil)
-
- mbuf := rmrCgo.NewMBuf(tests.MessageType, tests.MaxMsgSize,"RanName" , &tests.DummyPayload, &tests.DummyXAction)
- rmrMessengerMock.On("SendMsg", mock.AnythingOfType(fmt.Sprintf("%T", mbuf)), tests.MaxMsgSize).Return(mbuf, nil)
-
- actual := handler.Handle(log, nil)
-
- assert.Nil(t, actual)
+ 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, rmrMessage.GetMsgSrc())
+ rmrMessengerMock.On("SendMsg", mbuf, true).Return(mbuf, e2managererrors.NewRmrError())
+ _, err := h.Handle(nil)
+ assert.IsType(t, &e2managererrors.RmrError{}, err)
+ rmrMessengerMock.AssertCalled(t, "SendMsg", mbuf, true)
+ readerMock.AssertExpectations(t)
+ writerMock.AssertExpectations(t)
}
-func TestHandleSaveFailedFlow(t *testing.T){
- log := initLog(t)
-
- readerMock := &mocks.RnibReaderMock{}
- readerProvider := func() reader.RNibReader {
- return readerMock
- }
- writerMock := &mocks.RnibWriterMock{}
- writerProvider := func() rNibWriter.RNibWriter {
- return writerMock
- }
- rmrMessengerMock := &mocks.RmrMessengerMock{}
- config := configuration.ParseConfiguration()
- config.BigRedButtonTimeoutSec = 1
- handler := NewDeleteAllRequestHandler(getRmrService(rmrMessengerMock, log), config, writerProvider, readerProvider)
-
- //Before timer: Disconnected->ShutDown, ShuttingDown->Ignore, Connected->ShuttingDown(will fail)
- nbIdentityList := createIdentityList()
+func testTwoRansTryShuttingDownSucceedsClearSucceedsRmrSucceedsAllRansAreShutdown(t *testing.T, partial bool) {
+ h, readerMock, writerMock, rmrMessengerMock, httpClientMock := setupDeleteAllRequestHandlerTest(t)
+ e2tAddresses := []string{E2TAddress}
+ readerMock.On("GetE2TAddresses").Return(e2tAddresses, nil)
+ mockHttpClientDissociateAllRans(httpClientMock, e2tAddresses, !partial)
+ nbIdentityList := []*entities.NbIdentity{{InventoryName: "RanName_1"}, {InventoryName: "RanName_2"}}
readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-
- nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED,}
- nb2 := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- nb3 := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus:entities.ConnectionStatus_CONNECTED,}
+ nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN}
+ nb2 := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN}
readerMock.On("GetNodeb", "RanName_1").Return(nb1, nil)
readerMock.On("GetNodeb", "RanName_2").Return(nb2, nil)
- readerMock.On("GetNodeb", "RanName_3").Return(nb3, nil)
-
- errRnib := &common.ResourceNotFoundError{}
- updatedNb1 := &entities.NodebInfo{RanName:"RanName_1", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- updatedNb3 := &entities.NodebInfo{RanName:"RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb1).Return(nil)
- writerMock.On("SaveNodeb", mock.Anything, updatedNb3).Return(errRnib)
-
- //after timer: ShutDown->Ignore, ShuttingDown->ShutDown(will fail)
- readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-
- nb1AfterTimer := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN,}
- nb2AfterTimer := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- nb3AfterTimer := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- readerMock.On("GetNodeb", "RanName_1").Return(nb1AfterTimer, nil)
- readerMock.On("GetNodeb", "RanName_2").Return(nb2AfterTimer, nil)
- readerMock.On("GetNodeb", "RanName_3").Return(nb3AfterTimer, nil)
-
- updatedNb2AfterTimer := &entities.NodebInfo{RanName:"RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- updatedNb3AfterTimer := &entities.NodebInfo{RanName:"RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb2AfterTimer).Return(nil)
- writerMock.On("SaveNodeb", mock.Anything, updatedNb3AfterTimer).Return(errRnib)
-
- mbuf := rmrCgo.NewMBuf(tests.MessageType, tests.MaxMsgSize,"RanName" , &tests.DummyPayload, &tests.DummyXAction)
- rmrMessengerMock.On("SendMsg", mock.AnythingOfType(fmt.Sprintf("%T", mbuf)), tests.MaxMsgSize).Return(mbuf, nil)
-
- actual := handler.Handle(log, nil)
+ readerMock.On("GetE2TAddresses").Return([]string{E2TAddress}, nil)
+ e2tInstance := entities.E2TInstance{Address: E2TAddress, AssociatedRanList: []string{"RanName_1", "RanName_2"}}
+ 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, rmrMessage.GetMsgSrc())
+ rmrMessengerMock.On("SendMsg", mbuf, true).Return(mbuf, nil)
+ resp, err := h.Handle(nil)
+ assert.Nil(t, err)
+
+ if partial {
+ assert.IsType(t, &models.RedButtonPartialSuccessResponseModel{}, resp)
+ } else {
+ assert.Nil(t, resp)
+ }
- assert.Nil(t, actual)
+ rmrMessengerMock.AssertCalled(t, "SendMsg", mbuf, true)
+ readerMock.AssertExpectations(t)
+ writerMock.AssertExpectations(t)
}
-func TestHandleSendRmrFailedFlow(t *testing.T){
- log := initLog(t)
+func TestTwoRansTryShuttingDownSucceedsClearSucceedsRmrSucceedsAllRansAreShutdownSuccess(t *testing.T) {
+ testTwoRansTryShuttingDownSucceedsClearSucceedsRmrSucceedsAllRansAreShutdown(t, false)
+}
- readerMock := &mocks.RnibReaderMock{}
- readerProvider := func() reader.RNibReader {
- return readerMock
- }
- writerMock := &mocks.RnibWriterMock{}
- writerProvider := func() rNibWriter.RNibWriter {
- return writerMock
- }
- rmrMessengerMock := &mocks.RmrMessengerMock{}
- config := configuration.ParseConfiguration()
- config.BigRedButtonTimeoutSec = 1
- handler := NewDeleteAllRequestHandler(getRmrService(rmrMessengerMock, log), config, writerProvider, readerProvider)
+func TestTwoRansTryShuttingDownSucceedsClearSucceedsRmrSucceedsAllRansAreShutdownPartialSuccess(t *testing.T) {
+ testTwoRansTryShuttingDownSucceedsClearSucceedsRmrSucceedsAllRansAreShutdown(t, true)
+}
- //Before timer: Disconnected->ShutDown, ShuttingDown->Ignore, Connected->ShuttingDown(will fail)
- nbIdentityList := createIdentityList()
+//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 testOneRanTryShuttingDownSucceedsClearSucceedsRmrSucceedsRanStatusIsShuttingDown(t *testing.T, partial bool) {
+ h, readerMock, writerMock, rmrMessengerMock, httpClientMock := setupDeleteAllRequestHandlerTest(t)
+ e2tAddresses := []string{E2TAddress}
+ readerMock.On("GetE2TAddresses").Return(e2tAddresses, nil)
+ mockHttpClientDissociateAllRans(httpClientMock, e2tAddresses, !partial)
+ 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, rmrMessage.GetMsgSrc())
+ rmrMessengerMock.On("SendMsg", mbuf, true).Return(mbuf, nil)
- nb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED,}
- nb2 := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- nb3 := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus:entities.ConnectionStatus_CONNECTED,}
- readerMock.On("GetNodeb", "RanName_1").Return(nb1, nil)
- readerMock.On("GetNodeb", "RanName_2").Return(nb2, nil)
- readerMock.On("GetNodeb", "RanName_3").Return(nb3, nil)
-
- updatedNb1 := &entities.NodebInfo{RanName:"RanName_1", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- updatedNb3 := &entities.NodebInfo{RanName:"RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb1).Return(nil)
- writerMock.On("SaveNodeb", mock.Anything, updatedNb3).Return(nil)
-
- //after timer: ShutDown->Ignore, ShuttingDown->ShutDown(will fail)
readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
-
- nb1AfterTimer := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN,}
- nb2AfterTimer := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- nb3AfterTimer := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUTTING_DOWN,}
- readerMock.On("GetNodeb", "RanName_1").Return(nb1AfterTimer, nil)
- readerMock.On("GetNodeb", "RanName_2").Return(nb2AfterTimer, nil)
- readerMock.On("GetNodeb", "RanName_3").Return(nb3AfterTimer, nil)
-
- updatedNb2AfterTimer := &entities.NodebInfo{RanName:"RanName_2", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- updatedNb3AfterTimer := &entities.NodebInfo{RanName:"RanName_3", ConnectionStatus:entities.ConnectionStatus_SHUT_DOWN,}
- writerMock.On("SaveNodeb", mock.Anything, updatedNb2AfterTimer).Return(nil)
- writerMock.On("SaveNodeb", mock.Anything, updatedNb3AfterTimer).Return(nil)
-
- expected := e2managererrors.NewRmrError()
- mbuf := rmrCgo.NewMBuf(tests.MessageType, tests.MaxMsgSize,"RanName" , &tests.DummyPayload, &tests.DummyXAction)
- rmrMessengerMock.On("SendMsg", mock.AnythingOfType(fmt.Sprintf("%T", mbuf)), tests.MaxMsgSize).Return(mbuf, expected)
-
- actual := handler.Handle(log, nil)
-
- if reflect.TypeOf(actual) != reflect.TypeOf(expected){
- t.Errorf("Error actual = %v, and Expected = %v.", actual, expected)
- }
+ readerMock.On("GetNodeb", "RanName_1").Return(updatedNb1, nil)
+ updatedNb2 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUT_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb2).Return(nil)
+ _, err := h.Handle(nil)
+ assert.Nil(t, err)
+ rmrMessengerMock.AssertCalled(t, "SendMsg", mbuf, true)
+ readerMock.AssertExpectations(t)
+ writerMock.AssertExpectations(t)
+ writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 2)
}
-func TestHandleGetListEnbIdsEmptyFlow(t *testing.T){
- log := initLog(t)
-
- readerMock := &mocks.RnibReaderMock{}
- readerProvider := func() reader.RNibReader {
- return readerMock
- }
- writerMock := &mocks.RnibWriterMock{}
- writerProvider := func() rNibWriter.RNibWriter {
- return writerMock
- }
- rmrMessengerMock := &mocks.RmrMessengerMock{}
- config := configuration.ParseConfiguration()
-
- handler := NewDeleteAllRequestHandler(getRmrService(rmrMessengerMock, log), config, writerProvider, readerProvider)
-
- var rnibError error
- nbIdentityList := []*entities.NbIdentity{}
-
- readerMock.On("GetListNodebIds").Return(nbIdentityList, rnibError)
-
- actual := handler.Handle(log, nil)
- readerMock.AssertNumberOfCalls(t, "GetNodeb", 0)
- assert.Nil(t, actual)
+func TestOneRanTryShuttingDownSucceedsClearSucceedsRmrSucceedsRanStatusIsShuttingDownSuccess (t *testing.T) {
+ testOneRanTryShuttingDownSucceedsClearSucceedsRmrSucceedsRanStatusIsShuttingDown(t, false)
}
-func createIdentityList() []*entities.NbIdentity {
- nbIdentity1 := entities.NbIdentity{InventoryName: "RanName_1"}
- nbIdentity2 := entities.NbIdentity{InventoryName: "RanName_2"}
- nbIdentity3 := entities.NbIdentity{InventoryName: "RanName_3"}
+func TestOneRanTryShuttingDownSucceedsClearSucceedsRmrSucceedsRanStatusIsShuttingDownPartialSuccess (t *testing.T) {
+ testOneRanTryShuttingDownSucceedsClearSucceedsRmrSucceedsRanStatusIsShuttingDown(t, true)
+}
- var nbIdentityList []*entities.NbIdentity
- nbIdentityList = append(nbIdentityList, &nbIdentity1)
- nbIdentityList = append(nbIdentityList, &nbIdentity2)
- nbIdentityList = append(nbIdentityList, &nbIdentity3)
+func TestSuccessTwoE2TInstancesSixRans(t *testing.T) {
+ h, readerMock, writerMock, rmrMessengerMock, httpClientMock := setupDeleteAllRequestHandlerTest(t)
+ e2tAddresses := []string{E2TAddress, E2TAddress2}
+ readerMock.On("GetE2TAddresses").Return(e2tAddresses, nil)
+ mockHttpClientDissociateAllRans(httpClientMock, e2tAddresses, true)
+ nbIdentityList := []*entities.NbIdentity{{InventoryName: "RanName_1"}, {InventoryName: "RanName_2"}, {InventoryName: "RanName_3"}, {InventoryName: "RanName_4"}, {InventoryName: "RanName_5"}, {InventoryName: "RanName_6"}}
+ 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)
+ //nb2 := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: E2TAddress}
+ //readerMock.On("GetNodeb", "RanName_2").Return(nb2, nil)
+ //nb3 := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: E2TAddress}
+ //readerMock.On("GetNodeb", "RanName_3").Return(nb3, nil)
+ //nb4 := &entities.NodebInfo{RanName: "RanName_4", ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: E2TAddress2}
+ //readerMock.On("GetNodeb", "RanName_4").Return(nb4, nil)
+ //nb5 := &entities.NodebInfo{RanName: "RanName_5", ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: E2TAddress2}
+ //readerMock.On("GetNodeb", "RanName_5").Return(nb5, nil)
+ //nb6 := &entities.NodebInfo{RanName: "RanName_6", ConnectionStatus: entities.ConnectionStatus_CONNECTED, AssociatedE2TInstanceAddress: E2TAddress2}
+ //readerMock.On("GetNodeb", "RanName_6").Return(nb6, nil)
+
+ updatedNb1 := &entities.NodebInfo{RanName: "RanName_1", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb1).Return(nil)
+ updatedNb2 := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb2).Return(nil)
+ updatedNb3 := &entities.NodebInfo{RanName: "RanName_3", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb3).Return(nil)
+ updatedNb4 := &entities.NodebInfo{RanName: "RanName_4", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb4).Return(nil)
+ updatedNb5 := &entities.NodebInfo{RanName: "RanName_5", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb5).Return(nil)
+ updatedNb6 := &entities.NodebInfo{RanName: "RanName_6", ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN,}
+ writerMock.On("UpdateNodebInfo", updatedNb6).Return(nil)
+
+ readerMock.On("GetE2TAddresses").Return(e2tAddresses, nil)
+ e2tInstance := entities.E2TInstance{Address: E2TAddress, AssociatedRanList: []string{"RanName_1", "RanName_2", "RanName_3"}}
+ e2tInstance2 := entities.E2TInstance{Address: E2TAddress2, AssociatedRanList: []string{"RanName_4", "RanName_5", "RanName_6"}}
+ readerMock.On("GetE2TInstances", e2tAddresses).Return([]*entities.E2TInstance{&e2tInstance, &e2tInstance2}, nil)
+ updatedE2tInstance := e2tInstance
+ updatedE2tInstance.AssociatedRanList = []string{}
+ updatedE2tInstance2 := e2tInstance2
+ updatedE2tInstance2.AssociatedRanList = []string{}
+ writerMock.On("SaveE2TInstance", &updatedE2tInstance).Return(nil)
+ writerMock.On("SaveE2TInstance", &updatedE2tInstance2).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)
- return nbIdentityList
+ readerMock.On("GetListNodebIds").Return(nbIdentityList, nil)
+ readerMock.On("GetNodeb", "RanName_1").Return(updatedNb1, nil)
+ readerMock.On("GetNodeb", "RanName_2").Return(updatedNb2, nil)
+ readerMock.On("GetNodeb", "RanName_3").Return(updatedNb3, nil)
+ readerMock.On("GetNodeb", "RanName_4").Return(updatedNb4, nil)
+ readerMock.On("GetNodeb", "RanName_5").Return(updatedNb5, nil)
+ readerMock.On("GetNodeb", "RanName_6").Return(updatedNb6, nil)
+
+ updatedNb1AfterTimer := *updatedNb1
+ updatedNb1AfterTimer.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
+ writerMock.On("UpdateNodebInfo", &updatedNb1AfterTimer).Return(nil)
+ updatedNb2AfterTimer := *updatedNb2
+ updatedNb2AfterTimer.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
+ writerMock.On("UpdateNodebInfo", &updatedNb2AfterTimer).Return(nil)
+ updatedNb3AfterTimer := *updatedNb3
+ updatedNb3AfterTimer.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
+ writerMock.On("UpdateNodebInfo", &updatedNb3AfterTimer).Return(nil)
+ updatedNb4AfterTimer := *updatedNb4
+ updatedNb4AfterTimer.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
+ writerMock.On("UpdateNodebInfo", &updatedNb4AfterTimer).Return(nil)
+ updatedNb5AfterTimer := *updatedNb5
+ updatedNb5AfterTimer.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
+ writerMock.On("UpdateNodebInfo", &updatedNb5AfterTimer).Return(nil)
+ updatedNb6AfterTimer := *updatedNb6
+ updatedNb6AfterTimer.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN
+ writerMock.On("UpdateNodebInfo", &updatedNb6AfterTimer).Return(nil)
+ _, err := h.Handle(nil)
+ assert.Nil(t, err)
+ rmrMessengerMock.AssertCalled(t, "SendMsg", mbuf, true)
+ readerMock.AssertExpectations(t)
+ writerMock.AssertExpectations(t)
+ writerMock.AssertNumberOfCalls(t, "UpdateNodebInfo", 12)
}
func initLog(t *testing.T) *logger.Logger {
- log, err := logger.InitLogger(logger.InfoLevel)
+ log, err := logger.InitLogger(logger.DebugLevel)
if err != nil {
t.Errorf("#initLog test - failed to initialize logger, error: %s", err)
}
return log
}
-
-func getRmrService(rmrMessengerMock *mocks.RmrMessengerMock, log *logger.Logger) *services.RmrService {
+func getRmrSender(rmrMessengerMock *mocks.RmrMessengerMock, log *logger.Logger) *rmrsender.RmrSender {
rmrMessenger := rmrCgo.RmrMessenger(rmrMessengerMock)
- messageChannel := make(chan *models.NotificationResponse)
rmrMessengerMock.On("Init", tests.GetPort(), tests.MaxMsgSize, tests.Flags, log).Return(&rmrMessenger)
- return services.NewRmrService(services.NewRmrConfig(tests.Port, tests.MaxMsgSize, tests.Flags, log), rmrMessenger, make(sessions.E2Sessions), messageChannel)
-}
\ No newline at end of file
+ return rmrsender.NewRmrSender(log, rmrMessenger)
+}
Code Review / ric-plt / e2mgr.git / blobdiff