async def test_resubscribe(sm: SubscriptionManager): sub = Subscription(category=Category.MARKET, events=[MarketEvents.CANCELS, MarketEvents.ORDERS], topics=['ETH_AURA', 'ETH_ZRX']) sm.subscriptions = {Category.MARKET: sub} sm._init_subscriptions = Mock() sm.subscribe = CoroutineMock() await sm.resubscribe() sm._init_subscriptions.assert_called_once() sm.subscribe.assert_awaited_once_with(sub)
def test_init_subscriptions(sm: SubscriptionManager): sm._init_subscriptions() assert isinstance(sm.subscriptions, dict) assert sm.subscriptions == {}