diff options
Diffstat (limited to 'tests/test_store.py')
-rw-r--r-- | tests/test_store.py | 47 |
1 files changed, 45 insertions, 2 deletions
diff --git a/tests/test_store.py b/tests/test_store.py index ffd2645..df113b7 100644 --- a/tests/test_store.py +++ b/tests/test_store.py | |||
@@ -459,6 +459,13 @@ class ReportStoreTest(WebMockTestCase): | |||
459 | 459 | ||
460 | self.assertEqual({"foo": "bar", "date": mock.ANY, "user_id": None, "market_id": None}, result) | 460 | self.assertEqual({"foo": "bar", "date": mock.ANY, "user_id": None, "market_id": None}, result) |
461 | 461 | ||
462 | def test_add_redis_status(self): | ||
463 | report_store = market.ReportStore(self.m) | ||
464 | result = report_store.add_redis_status({"foo": "bar"}) | ||
465 | |||
466 | self.assertEqual({"foo": "bar"}, result) | ||
467 | self.assertEqual(result, report_store.redis_status[0]) | ||
468 | |||
462 | def test_set_verbose(self): | 469 | def test_set_verbose(self): |
463 | report_store = market.ReportStore(self.m) | 470 | report_store = market.ReportStore(self.m) |
464 | with self.subTest(verbose=True): | 471 | with self.subTest(verbose=True): |
@@ -534,6 +541,20 @@ class ReportStoreTest(WebMockTestCase): | |||
534 | self.assertEqual(("date1", "type1", '{\n "foo": "bar",\n "bla": "bla"\n}'), logs[0]) | 541 | self.assertEqual(("date1", "type1", '{\n "foo": "bar",\n "bla": "bla"\n}'), logs[0]) |
535 | self.assertEqual(("date2", "type2", '{\n "foo": "bar",\n "bla": "bla"\n}'), logs[1]) | 542 | self.assertEqual(("date2", "type2", '{\n "foo": "bar",\n "bla": "bla"\n}'), logs[1]) |
536 | 543 | ||
544 | def test_to_json_redis(self): | ||
545 | report_store = market.ReportStore(self.m) | ||
546 | report_store.redis_status.append({ | ||
547 | "type": "type1", "foo": "bar", "bla": "bla" | ||
548 | }) | ||
549 | report_store.redis_status.append({ | ||
550 | "type": "type2", "foo": "bar", "bla": "bla" | ||
551 | }) | ||
552 | logs = list(report_store.to_json_redis()) | ||
553 | |||
554 | self.assertEqual(2, len(logs)) | ||
555 | self.assertEqual(("type1", '{"foo": "bar", "bla": "bla"}'), logs[0]) | ||
556 | self.assertEqual(("type2", '{"foo": "bar", "bla": "bla"}'), logs[1]) | ||
557 | |||
537 | @mock.patch.object(market.ReportStore, "print_log") | 558 | @mock.patch.object(market.ReportStore, "print_log") |
538 | @mock.patch.object(market.ReportStore, "add_log") | 559 | @mock.patch.object(market.ReportStore, "add_log") |
539 | def test_log_stage(self, add_log, print_log): | 560 | def test_log_stage(self, add_log, print_log): |
@@ -559,7 +580,8 @@ class ReportStoreTest(WebMockTestCase): | |||
559 | 580 | ||
560 | @mock.patch.object(market.ReportStore, "print_log") | 581 | @mock.patch.object(market.ReportStore, "print_log") |
561 | @mock.patch.object(market.ReportStore, "add_log") | 582 | @mock.patch.object(market.ReportStore, "add_log") |
562 | def test_log_balances(self, add_log, print_log): | 583 | @mock.patch.object(market.ReportStore, "add_redis_status") |
584 | def test_log_balances(self, add_redis_status, add_log, print_log): | ||
563 | report_store = market.ReportStore(self.m) | 585 | report_store = market.ReportStore(self.m) |
564 | self.m.balances.as_json.return_value = "json" | 586 | self.m.balances.as_json.return_value = "json" |
565 | self.m.balances.all = { "FOO": "bar", "BAR": "baz" } | 587 | self.m.balances.all = { "FOO": "bar", "BAR": "baz" } |
@@ -575,10 +597,16 @@ class ReportStoreTest(WebMockTestCase): | |||
575 | 'balances': 'json', | 597 | 'balances': 'json', |
576 | 'tag': 'tag' | 598 | 'tag': 'tag' |
577 | }) | 599 | }) |
600 | add_redis_status.assert_called_once_with({ | ||
601 | 'type': 'balance', | ||
602 | 'balances': 'json', | ||
603 | 'tag': 'tag' | ||
604 | }) | ||
578 | 605 | ||
579 | @mock.patch.object(market.ReportStore, "print_log") | 606 | @mock.patch.object(market.ReportStore, "print_log") |
580 | @mock.patch.object(market.ReportStore, "add_log") | 607 | @mock.patch.object(market.ReportStore, "add_log") |
581 | def test_log_tickers(self, add_log, print_log): | 608 | @mock.patch.object(market.ReportStore, "add_redis_status") |
609 | def test_log_tickers(self, add_redis_status, add_log, print_log): | ||
582 | report_store = market.ReportStore(self.m) | 610 | report_store = market.ReportStore(self.m) |
583 | amounts = { | 611 | amounts = { |
584 | "BTC": portfolio.Amount("BTC", 10), | 612 | "BTC": portfolio.Amount("BTC", 10), |
@@ -603,6 +631,21 @@ class ReportStoreTest(WebMockTestCase): | |||
603 | }, | 631 | }, |
604 | 'total': D('10.3') | 632 | 'total': D('10.3') |
605 | }) | 633 | }) |
634 | add_redis_status.assert_called_once_with({ | ||
635 | 'type': 'tickers', | ||
636 | 'compute_value': 'default', | ||
637 | 'balance_type': 'total', | ||
638 | 'currency': 'BTC', | ||
639 | 'balances': { | ||
640 | 'BTC': D('10'), | ||
641 | 'ETH': D('0.3') | ||
642 | }, | ||
643 | 'rates': { | ||
644 | 'BTC': None, | ||
645 | 'ETH': D('0.1') | ||
646 | }, | ||
647 | 'total': D('10.3') | ||
648 | }) | ||
606 | 649 | ||
607 | add_log.reset_mock() | 650 | add_log.reset_mock() |
608 | compute_value = lambda x: x["bid"] | 651 | compute_value = lambda x: x["bid"] |