From 884f1c87ae75ef0ef88c401e5095d022d87c1b6b Mon Sep 17 00:00:00 2001 From: Dan Bungert Date: Tue, 24 May 2022 13:31:04 -0600 Subject: [PATCH] pubsub: fix test fragility --- subiquitycore/tests/test_pubsub.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/subiquitycore/tests/test_pubsub.py b/subiquitycore/tests/test_pubsub.py index ecff7b61..3160a36a 100644 --- a/subiquitycore/tests/test_pubsub.py +++ b/subiquitycore/tests/test_pubsub.py @@ -13,38 +13,37 @@ # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . +import unittest from unittest.mock import MagicMock -from subiquitycore.tests import SubiTestCase from subiquitycore.pubsub import MessageHub -from subiquitycore.tests.util import run_coro -class TestMessageHub(SubiTestCase): +class TestMessageHub(unittest.IsolatedAsyncioTestCase): def setUp(self): self.hub = MessageHub() - def test_multicall(self): + async def test_multicall(self): cb = MagicMock() expected_calls = 3 channel_id = 1234 for _ in range(expected_calls): self.hub.subscribe(channel_id, cb) - run_coro(self.hub.abroadcast(channel_id)) + await self.hub.abroadcast(channel_id) self.assertEqual(expected_calls, cb.call_count) - def test_multisubscriber(self): + async def test_multisubscriber(self): cbs = [MagicMock() for _ in range(4)] channel_id = 2345 for cb in cbs: self.hub.subscribe(channel_id, cb) - run_coro(self.hub.abroadcast(channel_id)) + await self.hub.abroadcast(channel_id) for cb in cbs: cb.assert_called_once_with() - def test_message_arg(self): + async def test_message_arg(self): cb = MagicMock() channel_id = 'test-message-arg' self.hub.subscribe(channel_id, cb) - run_coro(self.hub.abroadcast(channel_id, '0', 1, 'two', [3], four=4)) + await self.hub.abroadcast(channel_id, '0', 1, 'two', [3], four=4) cb.assert_called_once_with('0', 1, 'two', [3], four=4)