class SupervisorEntryTests(unittest.TestCase): def setUp(self) -> None: self.exchange_mock = Mock() self.exchange_mock.get_open_orders_ws.return_value = [] self.exchange_mock.get_position_size_ws.return_value = 0 self.supervisor = Supervisor(interface=self.exchange_mock) def tearDown(self) -> None: self.supervisor.exit_cycle() def test_market_entry(self): self.supervisor.enter_by_market_order(228) self.exchange_mock.place_market_order.assert_called_once_with(qty=228) def test_fb_entry_last_price(self): self.exchange_mock.get_last_price_ws.return_value = 1000 self.exchange_mock.get_order_status_ws.return_value = 'Filled' self.exchange_mock.conn.get_tick_size.return_value = 0.5 self.supervisor.enter_fb_method(qty=228, price_type='last', max_retry=5, timeout=3) order = Order(order_type='Limit', qty=228, price=1000, side='Buy', passive=True) self.exchange_mock.place_order.assert_called_once_with(order) def test_fb_entry_last_price_timedout(self): self.exchange_mock.get_last_price_ws.return_value = 1000 self.exchange_mock.get_order_status_ws.return_value = 'New' self.exchange_mock.conn.get_tick_size.return_value = 0.5 self.supervisor.enter_fb_method(qty=228, price_type='last', max_retry=3, timeout=1) order = Order(order_type='Limit', qty=228, price=1000, side='Buy', passive=True) expected_calls = [call(order)] * 3 self.exchange_mock.place_order.assert_has_calls(expected_calls) self.exchange_mock.place_market_order.assert_called_once_with(qty=228) def test_fb_entry_first_orderbook_price(self): self.exchange_mock.get_first_orderbook_price_ws.return_value = 1000 self.exchange_mock.get_order_status_ws.return_value = 'Filled' self.exchange_mock.conn.get_tick_size.return_value = 0.5 self.supervisor.enter_fb_method(qty=228, price_type='first_ob', max_retry=5, timeout=1) order = Order(order_type='Limit', qty=228, price=1000, side='Buy', passive=True) self.exchange_mock.place_order.assert_called_once_with(order) def test_fb_entry_third_orderbook_price(self): self.exchange_mock.get_third_orderbook_price_ws.return_value = 1000 self.exchange_mock.get_order_status_ws.return_value = 'Filled' self.exchange_mock.conn.get_tick_size.return_value = 0.5 self.supervisor.enter_fb_method(qty=228, price_type='third_ob', max_retry=5, timeout=1) order = Order(order_type='Limit', qty=228, price=1000, side='Buy', passive=True) self.exchange_mock.place_order.assert_called_once_with(order) def test_fb_entry_with_deviation(self): self.exchange_mock.get_last_price_ws.return_value = 1000 self.exchange_mock.get_order_status_ws.return_value = 'Filled' self.exchange_mock.conn.get_tick_size.return_value = 0.5 self.supervisor.enter_fb_method(qty=228, price_type='deviant', max_retry=5, timeout=1, deviation=-10) order = Order(order_type='Limit', qty=228, price=900, side='Buy', passive=True) self.exchange_mock.place_order.assert_called_once_with(order)
class SupervisorEntryTests(unittest.TestCase): def setUp(self) -> None: self.exchange_mock = Mock() self.exchange_mock.get_open_orders_ws.return_value = [] self.exchange_mock.get_position_size_ws.return_value = 0 self.supervisor = Supervisor(interface=self.exchange_mock) def tearDown(self) -> None: self.supervisor.exit_cycle() def test_market_position_enter_while_running_cycle(self): self.supervisor.run_cycle() self.supervisor.stop_cycle() self.supervisor.enter_by_market_order(qty=20) self.exchange_mock.get_position_size_ws.return_value = 20 self.supervisor.run_cycle() sleep(1) # assert that Supervisor change position only once self.exchange_mock.place_market_order.assert_called_once_with(qty=20) def test_fb_last_position_enter_while_running_cycle(self): self.exchange_mock.get_last_price_ws.return_value = 1000 self.exchange_mock.get_order_status_ws.return_value = 'Filled' self.exchange_mock.conn.get_tick_size.return_value = 0.5 self.supervisor.run_cycle() self.supervisor.stop_cycle() self.supervisor.enter_fb_method(qty=20, price_type='last', timeout=5, max_retry=3) self.exchange_mock.get_position_size_ws.return_value = 20 self.supervisor.run_cycle() sleep(1) order = Order(order_type='Limit', qty=20, price=1000, side='Buy', passive=True) # assert that Supervisor change position only once with limit order self.exchange_mock.place_order.assert_called_once_with(order) def test_fb_first_ob_position_enter_while_running_cycle(self): self.exchange_mock.get_first_orderbook_price_ws.return_value = 1000 self.exchange_mock.get_order_status_ws.return_value = 'Filled' self.exchange_mock.conn.get_tick_size.return_value = 0.5 self.supervisor.run_cycle() self.supervisor.stop_cycle() self.supervisor.enter_fb_method(qty=20, price_type='first_ob', timeout=5, max_retry=3) self.exchange_mock.get_position_size_ws.return_value = 20 self.supervisor.run_cycle() sleep(1) order = Order(order_type='Limit', qty=20, price=1000, side='Buy', passive=True) # assert that Supervisor change position only once with limit order self.exchange_mock.place_order.assert_called_once_with(order) def test_fb_third_ob_position_enter_while_running_cycle(self): self.exchange_mock.get_third_orderbook_price_ws.return_value = 1000 self.exchange_mock.get_order_status_ws.return_value = 'Filled' self.exchange_mock.conn.get_tick_size.return_value = 0.5 self.supervisor.run_cycle() self.supervisor.stop_cycle() self.supervisor.enter_fb_method(qty=20, price_type='third_ob', timeout=5, max_retry=3) self.exchange_mock.get_position_size_ws.return_value = 20 self.supervisor.run_cycle() sleep(1) order = Order(order_type='Limit', qty=20, price=1000, side='Buy', passive=True) # assert that Supervisor change position only once with limit order self.exchange_mock.place_order.assert_called_once_with(order) def test_fb_positive_deviant_position_enter_while_running_cycle(self): self.exchange_mock.get_last_price_ws.return_value = 1000 self.exchange_mock.get_order_status_ws.return_value = 'Filled' self.exchange_mock.conn.get_tick_size.return_value = 0.5 self.supervisor.run_cycle() self.supervisor.stop_cycle() self.supervisor.enter_fb_method(qty=20, price_type='deviant', timeout=5, max_retry=3, deviation=15) self.exchange_mock.get_position_size_ws.return_value = 20 self.supervisor.run_cycle() sleep(1) order = Order(order_type='Limit', qty=20, price=1150, side='Buy', passive=True) # assert that Supervisor change position only once with limit order self.exchange_mock.place_order.assert_called_once_with(order) def test_fb_negative_deviant_position_enter_while_running_cycle(self): self.exchange_mock.get_last_price_ws.return_value = 1000 self.exchange_mock.get_order_status_ws.return_value = 'Filled' self.exchange_mock.conn.get_tick_size.return_value = 0.5 self.supervisor.run_cycle() self.supervisor.stop_cycle() self.supervisor.enter_fb_method(qty=20, price_type='deviant', timeout=5, max_retry=3, deviation=-15) self.exchange_mock.get_position_size_ws.return_value = 20 self.supervisor.run_cycle() sleep(1) order = Order(order_type='Limit', qty=20, price=850, side='Buy', passive=True) # assert that Supervisor change position only once with limit order self.exchange_mock.place_order.assert_called_once_with(order)