diff options
Diffstat (limited to 'tests/test_store.py')
-rw-r--r-- | tests/test_store.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tests/test_store.py b/tests/test_store.py index 4232fc3..ffd2645 100644 --- a/tests/test_store.py +++ b/tests/test_store.py | |||
@@ -4,12 +4,14 @@ import datetime | |||
4 | import threading | 4 | import threading |
5 | import market, portfolio, store | 5 | import market, portfolio, store |
6 | 6 | ||
7 | @unittest.skipUnless("unit" in limits, "Unit skipped") | ||
7 | class NoopLockTest(unittest.TestCase): | 8 | class NoopLockTest(unittest.TestCase): |
8 | def test_with(self): | 9 | def test_with(self): |
9 | noop_lock = store.NoopLock() | 10 | noop_lock = store.NoopLock() |
10 | with noop_lock: | 11 | with noop_lock: |
11 | self.assertTrue(True) | 12 | self.assertTrue(True) |
12 | 13 | ||
14 | @unittest.skipUnless("unit" in limits, "Unit skipped") | ||
13 | class LockedVarTest(unittest.TestCase): | 15 | class LockedVarTest(unittest.TestCase): |
14 | 16 | ||
15 | def test_values(self): | 17 | def test_values(self): |
@@ -61,6 +63,7 @@ class LockedVarTest(unittest.TestCase): | |||
61 | thread3.join() | 63 | thread3.join() |
62 | self.assertEqual("Bar", locked_var.get()[0:3]) | 64 | self.assertEqual("Bar", locked_var.get()[0:3]) |
63 | 65 | ||
66 | @unittest.skipUnless("unit" in limits, "Unit skipped") | ||
64 | class TradeStoreTest(WebMockTestCase): | 67 | class TradeStoreTest(WebMockTestCase): |
65 | def test_compute_trades(self): | 68 | def test_compute_trades(self): |
66 | self.m.balances.currencies.return_value = ["XMR", "DASH", "XVG", "BTC", "ETH"] | 69 | self.m.balances.currencies.return_value = ["XMR", "DASH", "XVG", "BTC", "ETH"] |
@@ -285,6 +288,7 @@ class TradeStoreTest(WebMockTestCase): | |||
285 | 288 | ||
286 | self.assertEqual([trade_mock1, trade_mock2], trade_store.pending) | 289 | self.assertEqual([trade_mock1, trade_mock2], trade_store.pending) |
287 | 290 | ||
291 | @unittest.skipUnless("unit" in limits, "Unit skipped") | ||
288 | class BalanceStoreTest(WebMockTestCase): | 292 | class BalanceStoreTest(WebMockTestCase): |
289 | def setUp(self): | 293 | def setUp(self): |
290 | super().setUp() | 294 | super().setUp() |
@@ -437,6 +441,7 @@ class BalanceStoreTest(WebMockTestCase): | |||
437 | self.assertEqual(1, as_json["BTC"]) | 441 | self.assertEqual(1, as_json["BTC"]) |
438 | self.assertEqual(2, as_json["ETH"]) | 442 | self.assertEqual(2, as_json["ETH"]) |
439 | 443 | ||
444 | @unittest.skipUnless("unit" in limits, "Unit skipped") | ||
440 | class ReportStoreTest(WebMockTestCase): | 445 | class ReportStoreTest(WebMockTestCase): |
441 | def test_add_log(self): | 446 | def test_add_log(self): |
442 | with self.subTest(market=self.m): | 447 | with self.subTest(market=self.m): |
@@ -830,6 +835,7 @@ class ReportStoreTest(WebMockTestCase): | |||
830 | response = mock.Mock() | 835 | response = mock.Mock() |
831 | response.status_code = 200 | 836 | response.status_code = 200 |
832 | response.text = "Hey" | 837 | response.text = "Hey" |
838 | response.elapsed.total_seconds.return_value = 120 | ||
833 | 839 | ||
834 | report_store.log_http_request("method", "url", "body", | 840 | report_store.log_http_request("method", "url", "body", |
835 | "headers", response) | 841 | "headers", response) |
@@ -840,6 +846,7 @@ class ReportStoreTest(WebMockTestCase): | |||
840 | 'body': 'body', | 846 | 'body': 'body', |
841 | 'headers': 'headers', | 847 | 'headers': 'headers', |
842 | 'status': 200, | 848 | 'status': 200, |
849 | 'duration': 120, | ||
843 | 'response': 'Hey', | 850 | 'response': 'Hey', |
844 | 'response_same_as': None, | 851 | 'response_same_as': None, |
845 | }) | 852 | }) |
@@ -995,6 +1002,7 @@ class ReportStoreTest(WebMockTestCase): | |||
995 | 'action': 'Hey' | 1002 | 'action': 'Hey' |
996 | }) | 1003 | }) |
997 | 1004 | ||
1005 | @unittest.skipUnless("unit" in limits, "Unit skipped") | ||
998 | class PortfolioTest(WebMockTestCase): | 1006 | class PortfolioTest(WebMockTestCase): |
999 | def setUp(self): | 1007 | def setUp(self): |
1000 | super().setUp() | 1008 | super().setUp() |