X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fcontrol%2Fe2if_state_test.go;h=09ee212d9f7ca3c5865dc57fb7ba324c131a673e;hb=84662c4f3e2273152fc849f74880108b6346bcad;hp=e4ee848d4877e5cb697ad0b58b9f5fab41af4ee6;hpb=c7da4ee19c2b2a96c1eeccd8da944902fcd9323f;p=ric-plt%2Fsubmgr.git

diff --git a/pkg/control/e2if_state_test.go b/pkg/control/e2if_state_test.go
index e4ee848..09ee212 100644
--- a/pkg/control/e2if_state_test.go
+++ b/pkg/control/e2if_state_test.go
@@ -71,6 +71,7 @@ func TestMock(t *testing.T) {
 	xappRnibMock.CreateGnb("gnb_369_11105_aaaaa3", entities.ConnectionStatus_SHUTTING_DOWN)
 	xappRnibMock.CreateGnb("gnb_369_11105_aaaaa3", entities.ConnectionStatus_SHUT_DOWN)
 	xappRnibMock.CreateGnb("gnb_369_11105_aaaaa3", entities.ConnectionStatus_DISCONNECTED)
+	xappRnibMock.CreateGnb("gnb_369_11105_aaaaa3", entities.ConnectionStatus_UNDER_RESET)
 
 	mainCtrl.c.e2IfState.ReadE2ConfigurationFromRnib()
 	mainCtrl.c.e2IfState.SubscribeChannels()
@@ -92,6 +93,9 @@ func TestMock(t *testing.T) {
 	if err := xappRnibMock.XappRnibStoreAndPublish("RAN_CONNECTION_STATUS_CHANGE", "gnb_369_11105_aaaaa3_DISCONNECTED", "key1", "data1"); err != nil {
 		t.Errorf("XappRnibStoreAndPublish failed: %v", err)
 	}
+	if err := xappRnibMock.XappRnibStoreAndPublish("RAN_CONNECTION_STATUS_CHANGE", "gnb_369_11105_aaaaa3_UNDER_RESET", "key1", "data1"); err != nil {
+		t.Errorf("XappRnibStoreAndPublish failed: %v", err)
+	}
 }
 
 func (x *XappRnibMock) CreateGnb(gnbId string, connectionStatus entities.ConnectionStatus) {
@@ -237,6 +241,10 @@ func ExtratNbIdAndConnectionStatus(s string) (string, entities.ConnectionStatus,
 		connectionStatus = entities.ConnectionStatus_SHUT_DOWN
 		splitStringTbl := strings.Split(s, "_SHUT_DOWN")
 		nbId = splitStringTbl[0]
+	} else if strings.Contains(s, "_UNDER_RESET") {
+		connectionStatus = entities.ConnectionStatus_UNDER_RESET
+		splitStringTbl := strings.Split(s, "_UNDER_RESET")
+		nbId = splitStringTbl[0]
 	} else {
 		return "", 0, fmt.Errorf("XappRnibMock: Invalid connection status. %s", s)
 	}