class DataClientTests(unittest.TestCase):
def setUp(self):
# Fixture Setup
self.clock = TestClock()
self.uuid_factory = UUIDFactory()
self.logger = TestLogger(self.clock)
self.portfolio = Portfolio(
clock=self.clock,
logger=self.logger,
)
self.data_engine = DataEngine(
portfolio=self.portfolio,
clock=self.clock,
logger=self.logger,
)
self.venue = Venue("SIM")
self.client = DataClient(
name="TEST_PROVIDER",
engine=self.data_engine,
clock=self.clock,
logger=self.logger,
)
def test_connect_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.connect)
def test_disconnect_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.disconnect)
def test_reset_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.reset)
def test_dispose_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.dispose)
def test_subscribe_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.subscribe,
DataType(str))
def test_unsubscribe_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.unsubscribe,
DataType(str))
def test_request_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.request,
DataType(str), self.uuid_factory.generate())
def test_handle_data_sends_to_data_engine(self):
# Arrange
data_type = DataType(str, {"Type": "NEWS_WIRE"})
data = GenericData(data_type, "Some news headline", UNIX_EPOCH)
# Act
self.client._handle_data_py(data)
# Assert
self.assertEqual(1, self.data_engine.data_count)
def test_handle_data_response_sends_to_data_engine(self):
# Arrange
data_type = DataType(str, {
"Type": "ECONOMIC_DATA",
"topic": "unemployment"
})
data = GenericData(data_type, "may 2020, 6.9%", UNIX_EPOCH)
# Act
self.client._handle_data_response_py(data,
self.uuid_factory.generate())
# Assert
self.assertEqual(1, self.data_engine.response_count)
class TestDataClient:
def setup(self):
# Fixture Setup
self.clock = TestClock()
self.uuid_factory = UUIDFactory()
self.logger = Logger(self.clock)
self.trader_id = TestStubs.trader_id()
self.msgbus = MessageBus(
trader_id=self.trader_id,
clock=self.clock,
logger=self.logger,
)
self.cache = TestStubs.cache()
self.portfolio = Portfolio(
msgbus=self.msgbus,
cache=self.cache,
clock=self.clock,
logger=self.logger,
)
self.data_engine = DataEngine(
msgbus=self.msgbus,
cache=self.cache,
clock=self.clock,
logger=self.logger,
)
self.venue = Venue("SIM")
self.client = DataClient(
client_id=ClientId("TEST_PROVIDER"),
msgbus=self.msgbus,
cache=self.cache,
clock=self.clock,
logger=self.logger,
)
def test_handle_data_sends_to_data_engine(self):
# Arrange
data_type = DataType(NewsEvent, {"Type": "NEWS_WIRE"})
data = NewsEvent(
impact=NewsImpact.HIGH,
name="Unemployment Rate",
currency=USD,
ts_event=0,
ts_init=0,
)
generic_data = GenericData(data_type, data)
# Act
self.client._handle_data_py(generic_data)
# Assert
assert self.data_engine.data_count == 1
def test_handle_data_response_sends_to_data_engine(self):
# Arrange
data_type = DataType(NewsEvent, {"Type": "NEWS_WIRE"})
data = NewsEvent(
impact=NewsImpact.HIGH,
name="Unemployment Rate",
currency=USD,
ts_event=0,
ts_init=0,
)
# Act
self.client._handle_data_response_py(data_type, data,
self.uuid_factory.generate())
# Assert
assert self.data_engine.response_count == 1
class DataClientTests(unittest.TestCase):
def setUp(self):
# Fixture Setup
self.clock = TestClock()
self.uuid_factory = UUIDFactory()
self.logger = Logger(self.clock)
self.cache = TestStubs.cache()
self.portfolio = Portfolio(
cache=self.cache,
clock=self.clock,
logger=self.logger,
)
self.data_engine = DataEngine(
portfolio=self.portfolio,
cache=self.cache,
clock=self.clock,
logger=self.logger,
)
self.venue = Venue("SIM")
self.client = DataClient(
client_id=ClientId("TEST_PROVIDER"),
engine=self.data_engine,
clock=self.clock,
logger=self.logger,
)
def test_connect_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.connect)
def test_disconnect_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.disconnect)
def test_reset_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.reset)
def test_dispose_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.dispose)
def test_subscribe_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.subscribe, DataType(str))
def test_unsubscribe_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(NotImplementedError, self.client.unsubscribe, DataType(str))
def test_request_when_not_implemented_raises_exception(self):
# Arrange
# Act
# Assert
self.assertRaises(
NotImplementedError,
self.client.request,
DataType(str),
self.uuid_factory.generate(),
)
def test_handle_data_sends_to_data_engine(self):
# Arrange
data_type = DataType(NewsEvent, {"Type": "NEWS_WIRE"})
data = NewsEvent(
impact=NewsImpact.HIGH,
name="Unemployment Rate",
currency=USD,
ts_event_ns=0,
ts_recv_ns=0,
)
generic_data = GenericData(data_type, data)
# Act
self.client._handle_data_py(generic_data)
# Assert
self.assertEqual(1, self.data_engine.data_count)
def test_handle_data_response_sends_to_data_engine(self):
# Arrange
data_type = DataType(NewsEvent, {"Type": "NEWS_WIRE"})
data = NewsEvent(
impact=NewsImpact.HIGH,
name="Unemployment Rate",
currency=USD,
ts_event_ns=0,
ts_recv_ns=0,
)
# Act
self.client._handle_data_response_py(
data_type, data, self.uuid_factory.generate()
)
# Assert
self.assertEqual(1, self.data_engine.response_count)