import unittest

from api import Navs, Datum, PortfoliosRisk
from framework import autowired, parse_date, get_logger


class BasicTest(unittest.TestCase):
    logger = get_logger(__name__)

    @autowired
    def test_index_close(self, navs: Navs = None):
        closes = navs.get_index_close(ticker='SPX Index', min_date=parse_date('2022-11-01'), datum_ids=67)
        self.logger.info(closes)

    @autowired
    def test_get_high_risk_datums(self, datum: Datum = None):
        datums = datum.get_high_risk_datums(PortfoliosRisk.FT9)
        self.logger.info(datums)


if __name__ == '__main__':
    unittest.main()