from .helper import *
import main, market
-@unittest.skipUnless("unit" in limits, "Unit skipped")
class MainTest(WebMockTestCase):
def test_make_order(self):
self.m.get_ticker.return_value = {
mock.patch("main.parse_config") as parse_config,\
mock.patch("main.fetch_markets") as fetch_markets,\
mock.patch("main.process") as process,\
- mock.patch("store.Portfolio.start_worker") as start:
+ mock.patch("store.Portfolio.start_worker") as start,\
+ mock.patch("store.Portfolio.stop_worker") as stop:
args_mock = mock.Mock()
args_mock.parallel = True
parse_config.assert_called_with(args_mock)
fetch_markets.assert_called_with("pg_config", "user")
+ stop.assert_called_once_with()
start.assert_called_once_with()
self.assertEqual(2, process.call_count)
process.assert_has_calls([