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"
