import unittest
from framework import autowired, parse_date, get_logger
from api import PortfoliosBuilder, PortfoliosType


class PortfoliosTest(unittest.TestCase):

    logger = get_logger(__name__)

    @autowired(names={'builder': 'poem'})
    def test_portfolio_builder(self, builder: PortfoliosBuilder = None):
        result, detail = builder.build_portfolio(parse_date('2011-11-07'), PortfoliosType.NORMAL)
        self.logger.info("portfolios: ")
        for risk, portfolio in result.items():
            self.logger.info(risk.name)
            self.logger.info(portfolio)
            self.logger.info(detail[risk])


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