import unittest

from api import RebalanceSignal, PortfoliosRisk, RebalanceRuler
from framework import autowired, parse_date, get_logger


class RebalanceTest(unittest.TestCase):

    logger = get_logger(__name__)

    @autowired(names={'builder': 'crisis_one'})
    def test_crisis_one(self, builder: RebalanceSignal = None):
        signal = builder.get_signal(parse_date('2022-10-13'), PortfoliosRisk.FT9)
        self.logger.info(signal)

    @autowired(names={'builder': 'market-right'})
    def test_market_right(self, builder: RebalanceSignal = None):
        signal = builder.get_signal(parse_date('2022-10-13'), PortfoliosRisk.FT9)
        self.logger.info(signal)

    @autowired(names={'builder': 'curve-drift'})
    def test_curve_drift(self, builder: RebalanceSignal = None):
        signal = builder.get_signal(parse_date('2022-11-07'), PortfoliosRisk.FT3)
        self.logger.info(signal)

    @autowired(names={'builder': 'high-buy'})
    def test_high_buy(self, builder: RebalanceSignal = None):
        builder.get_signal(parse_date('2022-09-10'), PortfoliosRisk.FT3)

    @autowired
    def test_rebalance_builder(self, builder: RebalanceRuler = None):
        builder.take_next_signal(parse_date('2022-09-01'), PortfoliosRisk.FT3)


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