def test_process_get_action_response(sm: SubscriptionManager):
category = Category.MARKET
events = [MarketEvents.ORDERS, MarketEvents.CANCELS]
topics = ['ETH_AURA', 'ETH_ZRX']
payload = dict(events=events, topics=topics)
assert sm.subscriptions == {}
sm._process_get_action_response(category, payload)
assert sm.subscriptions == {}
sm._process_subscribe_action_response(category, payload)
payload['topics'] = ['ETH_SAN']
old_events = sm.subscriptions[category].events
sm._process_get_action_response(category, payload)
assert sm.subscriptions[category].events == old_events
assert sm.subscriptions[category].topics == ('ETH_SAN', )
def test_process_sub_response_get_handler(sm: SubscriptionManager):
message = dict(
result='success',
request='subscribeToMarkets',
payload=dict(action='get'),
)
sm._logger.error = Mock()
sm._process_get_action_response = Mock()
result = sm.process_sub_response(message)
sm._process_get_action_response.assert_called_once_with(
Category(message['request']), message['payload'])
assert result is None