Tried to add call count to test_create_order

This commit is contained in:
Sam Germain 2021-09-17 00:10:53 -06:00
parent ff5b402f58
commit 798a0c9827
1 changed files with 17 additions and 5 deletions

View File

@ -1094,10 +1094,7 @@ def test_create_dry_run_order_market_fill(default_conf, mocker, side, rate, amou
assert round(order["average"], 4) == round(endprice, 4)
@pytest.mark.parametrize("side", [
("buy"),
("sell")
])
@pytest.mark.parametrize("side", ["buy", "sell"])
@pytest.mark.parametrize("ordertype,rate,marketprice", [
("market", None, None),
("market", 200, True),
@ -1126,7 +1123,7 @@ def test_create_order(default_conf, mocker, side, ordertype, rate, marketprice,
side=side,
amount=1,
rate=200,
leverage=3.0
leverage=1.0
)
assert 'id' in order
@ -1138,6 +1135,21 @@ def test_create_order(default_conf, mocker, side, ordertype, rate, marketprice,
assert api_mock.create_order.call_args[0][3] == 1
assert api_mock.create_order.call_args[0][4] is rate
assert api_mock._set_leverage.call_count == 0 if side == "buy" else 1
assert api_mock.set_margin_mode.call_count == 0 if side == "buy" else 1
order = exchange.create_order(
pair='ETH/BTC',
ordertype=ordertype,
side=side,
amount=1,
rate=200,
leverage=3.0
)
assert api_mock._set_leverage.call_count == 1
assert api_mock.set_margin_mode.call_count == 1
def test_buy_dry_run(default_conf, mocker):
default_conf['dry_run'] = True