def test_datetime(self, mock_trade_close_init): trade = TradeFactory() trade.close(close_value=100, datetime="test") assert mock_trade_close_init.call_count == 1 assert mock_trade_close_init.call_args_list[0][1][ "datetime"] == "test"
def test_close_value(self, mock_trade_close_init): trade = TradeFactory(quantity=2) trade.close(close_value=100, datetime=arrow.utcnow()) assert mock_trade_close_init.call_count == 1 assert mock_trade_close_init.call_args_list[0][1][ "close_value"] == 100
def test_kwargs(self, mock_trade_close_init): trade = TradeFactory(quantity=2) trade.close(close_value=100, datetime=arrow.utcnow(), random_arg="test") assert mock_trade_close_init.call_args_list[0][1][ "random_arg"] == "test"
def test_append_to_closes(self, mocker): mocker.patch(f"{CLASS_TRADE_CLOSE_DEFINITION_PATH}", return_value="new_close") trade = TradeFactory() trade.close(close_value=100, datetime="test") assert trade.closes == ["new_close"]
def test_quantities(self, mock_trade_close_init, quantity, expected_quantity): trade = TradeFactory(quantity=5) trade.close(close_value=100, datetime=arrow.utcnow(), quantity=quantity) assert mock_trade_close_init.call_count == 1 assert (mock_trade_close_init.call_args_list[0][1]["quantity"] == expected_quantity)
def test_return_close(self, mocker): mocker.patch(f"{CLASS_TRADE_CLOSE_DEFINITION_PATH}", return_value="new_close") trade = TradeFactory() response = trade.close(close_value=100, datetime="test") assert response == "new_close"