X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=internal%2Fsdlgoredis%2Fsdlgoredis_test.go;h=4869d8fce8def3f4567e452baf21731b425d3a45;hb=0d5d1ef2006a6eb39b1a2b9f7591bd489cb21dff;hp=ce7260764d4f88d5741cd8c005f589909dd3b975;hpb=97445fe7e6737da5536c5f7a6909821a1612f66c;p=ric-plt%2Fsdlgo.git

diff --git a/internal/sdlgoredis/sdlgoredis_test.go b/internal/sdlgoredis/sdlgoredis_test.go
index ce72607..4869d8f 100644
--- a/internal/sdlgoredis/sdlgoredis_test.go
+++ b/internal/sdlgoredis/sdlgoredis_test.go
@@ -715,6 +715,7 @@ func TestSubscribeChannelDBSubscribeTwoUnsubscribeOne(t *testing.T) {
 		receivedChannel2 = channel
 	}, "{prefix}", "---", "{prefix}channel2")
 
+	time.Sleep(1 * time.Second)
 	db.UnsubscribeChannelDB("{prefix}channel1")
 	ch <- &msg2
 	db.UnsubscribeChannelDB("{prefix}channel2")
@@ -726,6 +727,42 @@ func TestSubscribeChannelDBSubscribeTwoUnsubscribeOne(t *testing.T) {
 	ps.AssertExpectations(t)
 }
 
+func TestSubscribeChannelReDBSubscribeAfterUnsubscribe(t *testing.T) {
+	ps, r, db := setup(true)
+	ch := make(chan *redis.Message)
+	msg := redis.Message{
+		Channel: "{prefix}channel",
+		Pattern: "pattern",
+		Payload: "event",
+	}
+	ps.On("Channel").Return(ch)
+	ps.On("Unsubscribe").Return(nil)
+	ps.On("Close").Return(nil)
+	count := 0
+	receivedChannel := ""
+
+	db.SubscribeChannelDB(func(channel string, payload ...string) {
+		count++
+		receivedChannel = channel
+	}, "{prefix}", "---", "{prefix}channel")
+	ch <- &msg
+	db.UnsubscribeChannelDB("{prefix}channel")
+	time.Sleep(1 * time.Second)
+
+	db.SubscribeChannelDB(func(channel string, payload ...string) {
+		count++
+		receivedChannel = channel
+	}, "{prefix}", "---", "{prefix}channel")
+	ch <- &msg
+	db.UnsubscribeChannelDB("{prefix}channel")
+
+	time.Sleep(1 * time.Second)
+	assert.Equal(t, 2, count)
+	assert.Equal(t, "channel", receivedChannel)
+	r.AssertExpectations(t)
+	ps.AssertExpectations(t)
+}
+
 func TestPTTLSuccessfully(t *testing.T) {
 	_, r, db := setup(true)
 	expectedKey := "key"