ccxt.poloniexE.return_value = self.ccxt
self.ccxt.session.request.return_value = "response"
- m = market.Market.from_config("config")
+ m = market.Market.from_config({"key": "key", "secred": "secret"})
self.assertEqual(self.ccxt, m.ccxt)
m.report.log_http_request.assert_called_with('GET', 'URL', 'data',
'headers', 'response')
- m = market.Market.from_config("config", debug=True)
+ m = market.Market.from_config({"key": "key", "secred": "secret"}, debug=True)
self.assertEqual(True, m.debug)
def test_get_ticker(self):
mock.patch.object(portfolio.Order, "mark_finished_order") as mark_finished_order:
order = portfolio.Order("buy", portfolio.Amount("ETH", 0.001),
D("0.1"), "BTC", "long", self.m, "trade")
- self.m.ccxt.create_order.side_effect = portfolio.ExchangeNotAvailable
+ self.m.ccxt.create_order.side_effect = portfolio.InvalidOrder
order.run()
self.m.ccxt.create_order.assert_called_once()
self.assertEqual(0, len(order.results))
self.assertEqual("closed", order.status)
mark_finished_order.assert_called_once()
+ self.m.ccxt.order_precision.return_value = 8
+ self.m.ccxt.create_order.reset_mock()
+ with self.subTest(insufficient_funds=True),\
+ mock.patch.object(portfolio.Order, "mark_finished_order") as mark_finished_order:
+ order = portfolio.Order("buy", portfolio.Amount("ETH", "0.001"),
+ D("0.1"), "BTC", "long", self.m, "trade")
+ self.m.ccxt.create_order.side_effect = [
+ portfolio.InsufficientFunds,
+ portfolio.InsufficientFunds,
+ portfolio.InsufficientFunds,
+ { "id": 123 },
+ ]
+ order.run()
+ self.m.ccxt.create_order.assert_has_calls([
+ mock.call('ETH/BTC', 'limit', 'buy', D('0.0010'), account='exchange', price=D('0.1')),
+ mock.call('ETH/BTC', 'limit', 'buy', D('0.00099'), account='exchange', price=D('0.1')),
+ mock.call('ETH/BTC', 'limit', 'buy', D('0.0009801'), account='exchange', price=D('0.1')),
+ mock.call('ETH/BTC', 'limit', 'buy', D('0.00097029'), account='exchange', price=D('0.1')),
+ ])
+ self.assertEqual(4, self.m.ccxt.create_order.call_count)
+ self.assertEqual(1, len(order.results))
+ self.assertEqual("open", order.status)
+ self.assertEqual(4, order.tries)
+ self.m.report.log_error.assert_called()
+ self.assertEqual(4, self.m.report.log_error.call_count)
+
+ self.m.ccxt.order_precision.return_value = 8
+ self.m.ccxt.create_order.reset_mock()
+ self.m.report.log_error.reset_mock()
+ with self.subTest(insufficient_funds=True),\
+ mock.patch.object(portfolio.Order, "mark_finished_order") as mark_finished_order:
+ order = portfolio.Order("buy", portfolio.Amount("ETH", "0.001"),
+ D("0.1"), "BTC", "long", self.m, "trade")
+ self.m.ccxt.create_order.side_effect = [
+ portfolio.InsufficientFunds,
+ portfolio.InsufficientFunds,
+ portfolio.InsufficientFunds,
+ portfolio.InsufficientFunds,
+ portfolio.InsufficientFunds,
+ ]
+ order.run()
+ self.m.ccxt.create_order.assert_has_calls([
+ mock.call('ETH/BTC', 'limit', 'buy', D('0.0010'), account='exchange', price=D('0.1')),
+ mock.call('ETH/BTC', 'limit', 'buy', D('0.00099'), account='exchange', price=D('0.1')),
+ mock.call('ETH/BTC', 'limit', 'buy', D('0.0009801'), account='exchange', price=D('0.1')),
+ mock.call('ETH/BTC', 'limit', 'buy', D('0.00097029'), account='exchange', price=D('0.1')),
+ mock.call('ETH/BTC', 'limit', 'buy', D('0.00096059'), account='exchange', price=D('0.1')),
+ ])
+ self.assertEqual(5, self.m.ccxt.create_order.call_count)
+ self.assertEqual(0, len(order.results))
+ self.assertEqual("error", order.status)
+ self.assertEqual(5, order.tries)
+ self.m.report.log_error.assert_called()
+ self.assertEqual(5, self.m.report.log_error.call_count)
+ self.m.report.log_error.assert_called_with(mock.ANY, message="Giving up Order(buy long 0.00096060 ETH at 0.1 BTC [pending])", exception=mock.ANY)
+
@unittest.skipUnless("unit" in limits, "Unit skipped")
class MouvementTest(WebMockTestCase):
report_store.logs.append({"date": portfolio.datetime(2018, 2, 24)})
self.assertEqual('[{"foo": "bar"}, {"date": "2018-02-24T00:00:00"}]', report_store.to_json())
report_store.logs.append({"amount": portfolio.Amount("BTC", 1)})
- with self.assertRaises(TypeError):
- report_store.to_json()
+ self.assertEqual('[{"foo": "bar"}, {"date": "2018-02-24T00:00:00"}, {"amount": "1.00000000 BTC"}]', report_store.to_json())
@mock.patch.object(market.ReportStore, "print_log")
@mock.patch.object(market.ReportStore, "add_log")
@mock.patch("portfolio.Portfolio.wait_for_recent")
def test_main_process_market(self, wait, buy, sell):
with self.subTest(before=False, after=False):
- helper.main_process_market("user")
+ helper.main_process_market("user", None)
wait.assert_not_called()
buy.assert_not_called()
wait.reset_mock()
sell.reset_mock()
with self.subTest(before=True, after=False):
- helper.main_process_market("user", before=True)
+ helper.main_process_market("user", None, before=True)
wait.assert_not_called()
buy.assert_not_called()
wait.reset_mock()
sell.reset_mock()
with self.subTest(before=False, after=True):
- helper.main_process_market("user", after=True)
+ helper.main_process_market("user", None, after=True)
wait.assert_called_once_with("user")
buy.assert_called_once_with("user")
wait.reset_mock()
sell.reset_mock()
with self.subTest(before=True, after=True):
- helper.main_process_market("user", before=True, after=True)
+ helper.main_process_market("user", None, before=True, after=True)
wait.assert_called_once_with("user")
buy.assert_called_once_with("user")
sell.assert_called_once_with("user")
+ buy.reset_mock()
+ wait.reset_mock()
+ sell.reset_mock()
+ with self.subTest(action="print_balances"),\
+ mock.patch("helper.print_balances") as print_balances:
+ helper.main_process_market("user", "print_balances")
+
+ buy.assert_not_called()
+ wait.assert_not_called()
+ sell.assert_not_called()
+ print_balances.assert_called_once_with("user")
+
+ with self.subTest(action="print_orders"),\
+ mock.patch("helper.print_orders") as print_orders:
+ helper.main_process_market("user", "print_orders")
+
+ buy.assert_not_called()
+ wait.assert_not_called()
+ sell.assert_not_called()
+ print_orders.assert_called_once_with("user")
+
+ with self.subTest(action="unknown"),\
+ self.assertRaises(NotImplementedError):
+ helper.main_process_market("user", "unknown")
+
@mock.patch.object(helper, "psycopg2")
def test_fetch_markets(self, psycopg2):
connect_mock = mock.Mock()
connect_mock.cursor.return_value = cursor_mock
psycopg2.connect.return_value = connect_mock
- rows = list(helper.main_fetch_markets({"foo": "bar"}))
+ with self.subTest(user=None):
+ rows = list(helper.main_fetch_markets({"foo": "bar"}, None))
+
+ psycopg2.connect.assert_called_once_with(foo="bar")
+ cursor_mock.execute.assert_called_once_with("SELECT config,user_id FROM market_configs")
+
+ self.assertEqual(["row_1", "row_2"], rows)
+
+ psycopg2.connect.reset_mock()
+ cursor_mock.execute.reset_mock()
+ with self.subTest(user=1):
+ rows = list(helper.main_fetch_markets({"foo": "bar"}, 1))
- psycopg2.connect.assert_called_once_with(foo="bar")
- cursor_mock.execute.assert_called_once_with("SELECT config,user_id FROM market_configs")
+ psycopg2.connect.assert_called_once_with(foo="bar")
+ cursor_mock.execute.assert_called_once_with("SELECT config,user_id FROM market_configs WHERE user_id = %s", 1)
- self.assertEqual(["row_1", "row_2"], rows)
+ self.assertEqual(["row_1", "row_2"], rows)
@mock.patch.object(helper.sys, "exit")
def test_main_parse_args(self, exit):
helper.print_balances(self.m)
+ self.m.report.log_stage.assert_called_once_with("print_balances")
self.m.balances.fetch_balances.assert_called_with()
self.m.report.print_log.assert_has_calls([
mock.call("total:"),