X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2FrNibWriter%2FrNibWriter_test.go;h=d38d0fe4cc41874ee028e61f64a5abe792f2cb4c;hb=refs%2Fchanges%2F20%2F3320%2F1;hp=34ef4c87b969a00e435f136fbfa7f1c73223dd49;hpb=efcb4528362460aa2249d319c9752b63bb720fe2;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/rNibWriter/rNibWriter_test.go b/E2Manager/rNibWriter/rNibWriter_test.go
index 34ef4c8..d38d0fe 100644
--- a/E2Manager/rNibWriter/rNibWriter_test.go
+++ b/E2Manager/rNibWriter/rNibWriter_test.go
@@ -495,7 +495,7 @@ func TestSaveE2TInstanceSuccess(t *testing.T) {
 func TestSaveE2TInstanceNullE2tInstanceFailure(t *testing.T) {
 	w, _ := initSdlInstanceMock(namespace)
 	var address string
-	e2tInstance := entities.NewE2TInstance(address)
+	e2tInstance := entities.NewE2TInstance(address, "test")
 	err := w.SaveE2TInstance(e2tInstance)
 	assert.NotNil(t, err)
 	assert.IsType(t, &common.ValidationError{}, err)
@@ -529,7 +529,7 @@ func TestSaveE2TInstanceSdlFailure(t *testing.T) {
 }
 
 func generateE2tInstance(address string) *entities.E2TInstance {
-	e2tInstance := entities.NewE2TInstance(address)
+	e2tInstance := entities.NewE2TInstance(address, "pod test")
 
 	e2tInstance.AssociatedRanList = []string{"test1", "test2"}
 
@@ -577,6 +577,39 @@ func TestSaveE2TAddressesSdlFailure(t *testing.T) {
 	assert.IsType(t, &common.InternalError{}, rNibErr)
 }
 
+func TestRemoveE2TInstanceSuccess(t *testing.T) {
+	address := "10.10.2.15:9800"
+	w, sdlInstanceMock := initSdlInstanceMock(namespace)
+
+	e2tAddresses := []string{fmt.Sprintf("E2TInstance:%s", address)}
+	var e error
+	sdlInstanceMock.On("Remove", e2tAddresses).Return(e)
+
+	rNibErr := w.RemoveE2TInstance(address)
+	assert.Nil(t, rNibErr)
+	sdlInstanceMock.AssertExpectations(t)
+}
+
+func TestRemoveE2TInstanceSdlFailure(t *testing.T) {
+	address := "10.10.2.15:9800"
+	w, sdlInstanceMock := initSdlInstanceMock(namespace)
+
+	e2tAddresses := []string{fmt.Sprintf("E2TInstance:%s", address)}
+	expectedErr := errors.New("expected error")
+	sdlInstanceMock.On("Remove", e2tAddresses).Return(expectedErr)
+
+	rNibErr := w.RemoveE2TInstance(address)
+	assert.IsType(t, &common.InternalError{}, rNibErr)
+}
+
+func TestRemoveE2TInstanceEmptyAddressFailure(t *testing.T) {
+	w, sdlInstanceMock := initSdlInstanceMock(namespace)
+
+	rNibErr := w.RemoveE2TInstance("")
+	assert.IsType(t, &common.ValidationError{}, rNibErr)
+	sdlInstanceMock.AssertExpectations(t)
+}
+
 //Integration tests
 //
 //func TestSaveEnbGnbInteg(t *testing.T){