X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2FrNibWriter%2FrNibWriter_test.go;h=257bb85d95f841844496373f706ffabb3489dc8f;hb=e96654dbb323864ef4d9f91cd703edb6183bf4b6;hp=9306b97f4242d4c95d8056ec24ed248702be5402;hpb=08bbf91d9f5afd6ecd6eeb9bcc8c160b8242d1ec;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/rNibWriter/rNibWriter_test.go b/E2Manager/rNibWriter/rNibWriter_test.go
index 9306b97..257bb85 100644
--- a/E2Manager/rNibWriter/rNibWriter_test.go
+++ b/E2Manager/rNibWriter/rNibWriter_test.go
@@ -33,6 +33,9 @@ import (
 )
 
 var namespace = "namespace"
+const (
+	RanName = "test"
+)
 
 func initSdlInstanceMock(namespace string) (w RNibWriter, sdlInstanceMock *mocks.MockSdlInstance) {
 	sdlInstanceMock = new(mocks.MockSdlInstance)
@@ -65,7 +68,7 @@ func generateServedNrCells(cellIds ...string) []*entities.ServedNRCell {
 
 	servedNrCells := []*entities.ServedNRCell{}
 
-	for _, v := range cellIds {
+	for i, v := range cellIds {
 		servedNrCells = append(servedNrCells, &entities.ServedNRCell{ServedNrCellInformation: &entities.ServedNRCellInformation{
 			CellId: v,
 			ChoiceNrMode: &entities.ServedNRCellInformation_ChoiceNRMode{
@@ -74,7 +77,7 @@ func generateServedNrCells(cellIds ...string) []*entities.ServedNRCell {
 				},
 			},
 			NrMode:      entities.Nr_FDD,
-			NrPci:       5,
+			NrPci:       uint32(i + 1),
 			ServedPlmns: []string{"whatever"},
 		}})
 	}
@@ -82,6 +85,22 @@ func generateServedNrCells(cellIds ...string) []*entities.ServedNRCell {
 	return servedNrCells
 }
 
+func TestRemoveServedNrCellsSuccess(t *testing.T) {
+	w, sdlInstanceMock := initSdlInstanceMock(namespace)
+	servedNrCellsToRemove := generateServedNrCells("whatever1", "whatever2")
+	sdlInstanceMock.On("Remove", buildCellKeysToRemove(RanName, servedNrCellsToRemove)).Return(nil)
+	err := w.RemoveServedNrCells(RanName, servedNrCellsToRemove)
+	assert.Nil(t, err)
+}
+
+func TestRemoveServedNrCellsFailure(t *testing.T) {
+	w, sdlInstanceMock := initSdlInstanceMock(namespace)
+	servedNrCellsToRemove := generateServedNrCells("whatever1", "whatever2")
+	sdlInstanceMock.On("Remove", buildCellKeysToRemove(RanName, servedNrCellsToRemove)).Return(errors.New("expected error"))
+	err := w.RemoveServedNrCells(RanName, servedNrCellsToRemove)
+	assert.IsType(t, &common.InternalError{}, err)
+}
+
 func TestUpdateGnbCellsInvalidNodebInfoFailure(t *testing.T) {
 	w, sdlInstanceMock := initSdlInstanceMock(namespace)
 	servedNrCells := generateServedNrCells("test1", "test2")