Delete all subscriptions when Ran is under reset.

[ric-plt/submgr.git] / pkg / control / e2if_state_test.go

diff --git a/pkg/control/e2if_state_test.go b/pkg/control/e2if_state_test.go
index e4ee848..09ee212 100644 (file)
--- 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)
        }