async def asyncSetUp(self) -> None: """Set up test fixtures, if any.""" init_test_env() await clear_cache(cfg.redis.dsn) self.omega = await start_omega() await omicron.init() await cache.clear_bars_range("000001.XSHE", FrameType.MIN30)
async def asyncSetUp(self) -> None: init_test_env() await emit.start(engine=emit.Engine.REDIS, dsn=cfg.redis.dsn, start_server=True) await self.create_quotes_fetcher() await omicron.init(aq)
async def asyncSetUp(self) -> None: init_test_env() await emit.start(engine=emit.Engine.REDIS, dsn=cfg.redis.dsn, start_server=True) await self.create_quotes_fetcher() await omicron.init(aq) home = Path(cfg.omega.home).expanduser() os.makedirs(str(home / "data/chksum"), exist_ok=True) self.omega = await start_omega()
async def asyncSetUp(self) -> None: self.cfg = init_test_env() self.omega = await start_omega() # hack: enable postgres during unittest self.cfg.postgres.enabled = True await omicron.init()
async def asyncSetUp(self) -> None: self.cfg = init_test_env() # disable catch up sync, since test_cli, jobs will be launched many times and cache might by empty redis = await aioredis.create_redis(self.cfg.redis.dsn) last_sync = arrow.now(self.cfg.tz).format("YYYY-MM-DD HH:mm:SS") await redis.set("jobs.bars_sync.stop", last_sync)
async def asyncSetUp(self) -> None: self.cfg = init_test_env() # check if omega is running self.omega = await start_omega() await omicron.init() self.securities = Securities() await self.securities.load()
async def test_start_logging(self): init_test_env() # remove handlers set by config file, if there is. root = logging.getLogger() root.handlers.clear() fmt = "%(asctime)s %(levelname)-1.1s %(process)d %(name)s:%(funcName)s:%(lineno)s | %(message)s" channel = "test_start_logging" redis_logger = logging.getLogger("test_redis") handler = rlog.RedisHandler( channel=channel, level=logging.DEBUG, host="localhost", port="6379", formatter=logging.Formatter(fmt), ) redis_logger.addHandler(handler) _dir = "/tmp/omega/test_jobs" shutil.rmtree(_dir, ignore_errors=True) cfg4py.update_config({ "logreceiver": { "klass": "omega.logging.receiver.redis.RedisLogReceiver", "dsn": "redis://localhost:6379", "channel": channel, "filename": "/tmp/omega/test_jobs/omega.log", "backup_count": 2, "max_bytes": "0.08K", } }) await start_logging() for i in range(5): redis_logger.info("this is %sth test log", i) await asyncio.sleep(0.5) self.assertEqual(3, len(os.listdir(_dir))) with open(f"{_dir}/omega.log.2", "r", encoding="utf-8") as f: content = f.readlines()[0] msg = content.split("|")[1] self.assertEqual(" this is 2th test log\n", msg)
async def test_start_sync(self): init_test_env() port = find_free_port() url = f"http://localhost:{port}/jobs/sync_bars" job_server = None try: job_server = await start_job_server(port) sync_params = { "include": "000001.XSHE", "frame": FrameType.MIN60.value, "start": "2020-04-30", "stop": "2020-05-07", } async with aiohttp.ClientSession() as client: async with client.get(url, json=sync_params) as resp: self.assertEqual(200, resp.status) result = await resp.text() print(result) finally: if job_server: job_server.kill()
async def test_init(self): init_test_env() await omicron.init() await cache.sys.delete("jobs.bars_sync.stop") try: cfg.omega.sync.bars = [{ "frame": "1d", "start": "2020-01-02", "stop": "2020-01-03", "delay": 3, "cat": [], "include": "000001.XSHE", "exclude": "000001.XSHG", }] # disable init, just use cfg here with mock.patch("cfg4py.init"): await init(None, None) # won't trigger sync this time await init(None, None) finally: # cfg.omega.sync.bars = origin pass
async def asyncSetUp(self) -> None: init_test_env() self.cfg = cfg4py.get_instance() self.omega = await start_omega() self.archive = await start_archive_server() await omicron.init()
import unittest import arrow import numpy as np import omicron from omicron import cache from omicron.core.timeframe import tf from omicron.core.types import FrameType, SecurityType from omicron.models.securities import Securities from omicron.models.security import Security from tests import init_test_env, start_omega logger = logging.getLogger(__name__) cfg = init_test_env() class SecurityTest(unittest.IsolatedAsyncioTestCase): async def asyncSetUp(self) -> None: # check if omega is running self.omega = await start_omega() await omicron.init() self.securities = Securities() await self.securities.load() async def asyncTearDown(self) -> None: await omicron.shutdown() if self.omega is not None:
async def asyncSetUp(self) -> None: """Set up test fixtures, if any.""" init_test_env() self.omega = await start_omega() await omicron.init()
async def asyncSetUp(self) -> None: self.cfg = init_test_env()
async def asyncSetUp(self) -> None: init_test_env() self.cfg = cfg4py.get_instance() self.omega = await start_omega()
async def asyncSetUp(self) -> None: init_test_env() await self.create_quotes_fetcher() await omicron.init(aq)
def setUp(self) -> None: self.cfg = init_test_env()