from .helper import *
@unittest.skipUnless("acceptance" in limits, "Acceptance skipped")
class AcceptanceTest(WebMockTestCase):
@unittest.expectedFailure
def test_success_sell_only_necessary(self):
# FIXME: catch stdout
self.m.report.verbose_print = False
fetch_balance = {
"ETH": {
"exchange_free": D("1.0"),
"exchange_used": D("0.0"),
"exchange_total": D("1.0"),
"total": D("1.0"),
},
"ETC": {
"exchange_free": D("4.0"),
"exchange_used": D("0.0"),
"exchange_total": D("4.0"),
"total": D("4.0"),
},
"XVG": {
"exchange_free": D("1000.0"),
"exchange_used": D("0.0"),
"exchange_total": D("1000.0"),
"total": D("1000.0"),
},
}
repartition = {
"ETH": (D("0.25"), "long"),
"ETC": (D("0.25"), "long"),
"BTC": (D("0.4"), "long"),
"BTD": (D("0.01"), "short"),
"B2X": (D("0.04"), "long"),
"USDT": (D("0.05"), "long"),
}
def fetch_ticker(symbol):
if symbol == "ETH/BTC":
return {
"symbol": "ETH/BTC",
"bid": D("0.14"),
"ask": D("0.16")
}
if symbol == "ETC/BTC":
return {
"symbol": "ETC/BTC",
"bid": D("0.002"),
"ask": D("0.003")
}
if symbol == "XVG/BTC":
return {
"symbol": "XVG/BTC",
"bid": D("0.00003"),
"ask": D("0.00005")
}
if symbol == "BTD/BTC":
return {
"symbol": "BTD/BTC",
"bid": D("0.0008"),
"ask": D("0.0012")
}
if symbol == "B2X/BTC":
return {
"symbol": "B2X/BTC",
"bid": D("0.0008"),
"ask": D("0.0012")
}
if symbol == "USDT/BTC":
raise helper.ExchangeError
if symbol == "BTC/USDT":
return {
"symbol": "BTC/USDT",
"bid": D("14000"),
"ask": D("16000")
}
self.fail("Shouldn't have been called with {}".format(symbol))
market = mock.Mock()
market.fetch_all_balances.return_value = fetch_balance
market.fetch_ticker.side_effect = fetch_ticker
with mock.patch.object(market.Portfolio, "repartition", return_value=repartition):
# Action 1
helper.prepare_trades(market)
balances = portfolio.BalanceStore.all
self.assertEqual(portfolio.Amount("ETH", 1), balances["ETH"].total)
self.assertEqual(portfolio.Amount("ETC", 4), balances["ETC"].total)
self.assertEqual(portfolio.Amount("XVG", 1000), balances["XVG"].total)
trades = portfolio.TradeStore.all
self.assertEqual(portfolio.Amount("BTC", D("0.15")), trades[0].value_from)
self.assertEqual(portfolio.Amount("BTC", D("0.05")), trades[0].value_to)
self.assertEqual("dispose", trades[0].action)
self.assertEqual(portfolio.Amount("BTC", D("0.01")), trades[1].value_from)
self.assertEqual(portfolio.Amount("BTC", D("0.05")), trades[1].value_to)
self.assertEqual("acquire", trades[1].action)
self.assertEqual(portfolio.Amount("BTC", D("0.04")), trades[2].value_from)
self.assertEqual(portfolio.Amount("BTC", D("0.00")), trades[2].value_to)
self.assertEqual("dispose", trades[2].action)
self.assertEqual(portfolio.Amount("BTC", D("0.00")), trades[3].value_from)
self.assertEqual(portfolio.Amount("BTC", D("-0.002")), trades[3].value_to)
self.assertEqual("acquire", trades[3].action)
self.assertEqual(portfolio.Amount("BTC", D("0.00")), trades[4].value_from)
self.assertEqual(portfolio.Amount("BTC", D("0.008")), trades[4].value_to)
self.assertEqual("acquire", trades[4].action)
self.assertEqual(portfolio.Amount("BTC", D("0.00")), trades[5].value_from)
self.assertEqual(portfolio.Amount("BTC", D("0.01")), trades[5].value_to)
self.assertEqual("acquire", trades[5].action)
# Action 2
portfolio.TradeStore.prepare_orders(only="dispose", compute_value=lambda x, y: x["bid"] * D("1.001"))
all_orders = portfolio.TradeStore.all_orders(state="pending")
self.assertEqual(2, len(all_orders))
self.assertEqual(2, 3*all_orders[0].amount.value)
self.assertEqual(D("0.14014"), all_orders[0].rate)
self.assertEqual(1000, all_orders[1].amount.value)
self.assertEqual(D("0.00003003"), all_orders[1].rate)
def create_order(symbol, type, action, amount, price=None, account="exchange"):
self.assertEqual("limit", type)
if symbol == "ETH/BTC":
self.assertEqual("sell", action)
self.assertEqual(D('0.66666666'), amount)
self.assertEqual(D("0.14014"), price)
elif symbol == "XVG/BTC":
self.assertEqual("sell", action)
self.assertEqual(1000, amount)
self.assertEqual(D("0.00003003"), price)
else:
self.fail("I shouldn't have been called")
return {
"id": symbol,
}
market.create_order.side_effect = create_order
market.order_precision.return_value = 8
# Action 3
portfolio.TradeStore.run_orders()
self.assertEqual("open", all_orders[0].status)
self.assertEqual("open", all_orders[1].status)
market.fetch_order.return_value = { "status": "closed", "datetime": "2018-01-20 13:40:00" }
market.privatePostReturnOrderTrades.return_value = [
{
"tradeID": 42, "type": "buy", "fee": "0.0015",
"date": "2017-12-30 12:00:12", "rate": "0.1",
"amount": "10", "total": "1"
}
]
with mock.patch.object(market.time, "sleep") as sleep:
# Action 4
helper.follow_orders(verbose=False)
sleep.assert_called_with(30)
for order in all_orders:
self.assertEqual("closed", order.status)
fetch_balance = {
"ETH": {
"exchange_free": D("1.0") / 3,
"exchange_used": D("0.0"),
"exchange_total": D("1.0") / 3,
"margin_total": 0,
"total": D("1.0") / 3,
},
"BTC": {
"exchange_free": D("0.134"),
"exchange_used": D("0.0"),
"exchange_total": D("0.134"),
"margin_total": 0,
"total": D("0.134"),
},
"ETC": {
"exchange_free": D("4.0"),
"exchange_used": D("0.0"),
"exchange_total": D("4.0"),
"margin_total": 0,
"total": D("4.0"),
},
"XVG": {
"exchange_free": D("0.0"),
"exchange_used": D("0.0"),
"exchange_total": D("0.0"),
"margin_total": 0,
"total": D("0.0"),
},
}
market.fetch_all_balances.return_value = fetch_balance
with mock.patch.object(market.Portfolio, "repartition", return_value=repartition):
# Action 5
helper.prepare_trades(market, only="acquire", compute_value="average")
balances = portfolio.BalanceStore.all
self.assertEqual(portfolio.Amount("ETH", 1 / D("3")), balances["ETH"].total)
self.assertEqual(portfolio.Amount("ETC", 4), balances["ETC"].total)
self.assertEqual(portfolio.Amount("BTC", D("0.134")), balances["BTC"].total)
self.assertEqual(portfolio.Amount("XVG", 0), balances["XVG"].total)
trades = portfolio.TradeStore.all
self.assertEqual(portfolio.Amount("BTC", D("0.15")), trades[0].value_from)
self.assertEqual(portfolio.Amount("BTC", D("0.05")), trades[0].value_to)
self.assertEqual("dispose", trades[0].action)
self.assertEqual(portfolio.Amount("BTC", D("0.01")), trades[1].value_from)
self.assertEqual(portfolio.Amount("BTC", D("0.05")), trades[1].value_to)
self.assertEqual("acquire", trades[1].action)
self.assertNotIn("BTC", trades)
self.assertEqual(portfolio.Amount("BTC", D("0.04")), trades[2].value_from)
self.assertEqual(portfolio.Amount("BTC", D("0.00")), trades[2].value_to)
self.assertEqual("dispose", trades[2].action)
self.assertEqual(portfolio.Amount("BTC", D("0.00")), trades[3].value_from)
self.assertEqual(portfolio.Amount("BTC", D("-0.002")), trades[3].value_to)
self.assertEqual("acquire", trades[3].action)
self.assertEqual(portfolio.Amount("BTC", D("0.00")), trades[4].value_from)
self.assertEqual(portfolio.Amount("BTC", D("0.008")), trades[4].value_to)
self.assertEqual("acquire", trades[4].action)
self.assertEqual(portfolio.Amount("BTC", D("0.00")), trades[5].value_from)
self.assertEqual(portfolio.Amount("BTC", D("0.01")), trades[5].value_to)
self.assertEqual("acquire", trades[5].action)
# Action 6
portfolio.TradeStore.prepare_orders(only="acquire", compute_value=lambda x, y: x["ask"])
all_orders = portfolio.TradeStore.all_orders(state="pending")
self.assertEqual(4, len(all_orders))
self.assertEqual(portfolio.Amount("ETC", D("12.83333333")), round(all_orders[0].amount))
self.assertEqual(D("0.003"), all_orders[0].rate)
self.assertEqual("buy", all_orders[0].action)
self.assertEqual("long", all_orders[0].trade_type)
self.assertEqual(portfolio.Amount("BTD", D("1.61666666")), round(all_orders[1].amount))
self.assertEqual(D("0.0012"), all_orders[1].rate)
self.assertEqual("sell", all_orders[1].action)
self.assertEqual("short", all_orders[1].trade_type)
diff = portfolio.Amount("B2X", D("19.4")/3) - all_orders[2].amount
self.assertAlmostEqual(0, diff.value)
self.assertEqual(D("0.0012"), all_orders[2].rate)
self.assertEqual("buy", all_orders[2].action)
self.assertEqual("long", all_orders[2].trade_type)
self.assertEqual(portfolio.Amount("BTC", D("0.0097")), all_orders[3].amount)
self.assertEqual(D("16000"), all_orders[3].rate)
self.assertEqual("sell", all_orders[3].action)
self.assertEqual("long", all_orders[3].trade_type)
# Action 6b
# TODO:
# Move balances to margin
# Action 7
# TODO
# portfolio.TradeStore.run_orders()
with mock.patch.object(market.time, "sleep") as sleep:
# Action 8
helper.follow_orders(verbose=False)
sleep.assert_called_with(30)