openstack/oslo.messaging
Code Issues Proposed changes

Wait for expected messages in listener pool test

Browse Source 
In the listener pool test, we must wait that all expected messages
have been receive before stopping the listener.

Closes bug: #1397338

Change-Id: I5e4b39b54e3b99c014f5f9c77926108b90f94192
This commit is contained in:
Mehdi Abaakouk 2014-11-29 10:04:48 +01:00
parent f58d03e0de
commit 30a5b12eb4
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/notify/test_listener.py
View File

@ -62,6 +62,10 @@ class ListenerSetupMixin(object):
if self._expect_messages == self._received_msgs: if self._expect_messages == self._received_msgs:
self.stop() self.stop()
def wait_for(self, expect_messages):
while expect_messages != self._received_msgs:
pass
def stop(self): def stop(self):
for listener in self.listeners: for listener in self.listeners:
# Check start() does nothing with a running listener # Check start() does nothing with a running listener
@ -351,6 +355,7 @@ class TestNotifyListener(test_utils.BaseTestCase, ListenerSetupMixin):
'test message%d' % i) 'test message%d' % i)
mocked_endpoint1_calls.append(mocked_endpoint_call(i)) mocked_endpoint1_calls.append(mocked_endpoint_call(i))
self.trackers['pool2'].wait_for(25)
listener2_thread.stop() listener2_thread.stop()
for i in range(0, 25): for i in range(0, 25):
@ -358,6 +363,7 @@ class TestNotifyListener(test_utils.BaseTestCase, ListenerSetupMixin):
'test message%d' % i) 'test message%d' % i)
mocked_endpoint1_calls.append(mocked_endpoint_call(i)) mocked_endpoint1_calls.append(mocked_endpoint_call(i))
self.trackers['pool2'].wait_for(50)
listener2_thread.start() listener2_thread.start()
listener3_thread.stop() listener3_thread.stop()
@ -366,6 +372,7 @@ class TestNotifyListener(test_utils.BaseTestCase, ListenerSetupMixin):
'test message%d' % i) 'test message%d' % i)
mocked_endpoint1_calls.append(mocked_endpoint_call(i)) mocked_endpoint1_calls.append(mocked_endpoint_call(i))
self.trackers['pool2'].wait_for(75)
listener3_thread.start() listener3_thread.start()
for i in range(0, 25): for i in range(0, 25):