commit 170630c73d05bb295b6a527868d46e1de749426a
author Chip Boling <chip.boling@adtran.com> Mon Jun 11 12:03:55 2018 -0500
committer khenaidoo Nursimulu <knursimu@ciena.com> Thu Jun 14 19:34:10 2018 +0000
tree 3a73b745e8d937c5365f13357d8f4f17de2659e2
parent 33f1a8272f711f03ff0e8ff8ecd6eb700a7c81fb

VOL-943: Fix for event publish loop dropping subscribers

Change-Id: I96ca0a5bf0b06ee12140966dbfb0a15290e130df

tests/utests/common/test_event_bus.py

diff --git a/tests/utests/common/test_event_bus.py b/tests/utests/common/test_event_bus.py
index 532befb..67d8c8f 100644
--- a/tests/utests/common/test_event_bus.py
+++ b/tests/utests/common/test_event_bus.py
@@ -195,3 +195,26 @@
             msg = yield queue.get()
             self.assertEqual(msg, i)
         self.assertEqual(len(queue.pending), 0)
+
+    def test_subscribers_that_unsubscribe_when_called(self):
+        # VOL-943 bug fix check
+        ebc = EventBusClient(EventBus())
+
+        class UnsubscribeWhenCalled(object):
+            def __init__(self):
+                self.subscription = ebc.subscribe('news', self.unsubscribe)
+                self.called = False
+
+            def unsubscribe(self, _topic, _msg):
+                self.called = True
+                ebc.unsubscribe(self.subscription)
+
+        ebc1 = UnsubscribeWhenCalled()
+        ebc2 = UnsubscribeWhenCalled()
+        ebc3 = UnsubscribeWhenCalled()
+
+        ebc.publish('news', 'msg1')
+
+        self.assertTrue(ebc1.called)
+        self.assertTrue(ebc2.called)
+        self.assertTrue(ebc3.called)