def test_unsubscribe_custom_data_with_client_id(self): # Arrange actor = MockActor() actor.register_base( trader_id=self.trader_id, msgbus=self.msgbus, cache=self.cache, clock=self.clock, logger=self.logger, ) data_type = DataType(str, {"type": "NEWS_WIRE", "topic": "Earthquake"}) actor.subscribe_data(data_type, ClientId("QUANDL")) # Act actor.unsubscribe_data(data_type, ClientId("QUANDL")) # Assert assert self.data_engine.command_count == 2 assert actor.msgbus.subscriptions() == []
def test_subscribe_custom_data(self): # Arrange actor = MockActor() actor.register_base( trader_id=self.trader_id, msgbus=self.msgbus, cache=self.cache, clock=self.clock, logger=self.logger, ) data_type = DataType(str, {"type": "NEWS_WIRE", "topic": "Earthquake"}) # Act actor.subscribe_data(data_type) # Assert assert self.data_engine.command_count == 0 assert actor.msgbus.subscriptions( )[0].topic == "data.str.type=NEWS_WIRE.topic=Earthquake"