class TestMyViewSuccessCondition(BaseTest):

    def setUp(self):
        super(TestMyViewSuccessCondition, self).setUp()
        self.init_database()

        from .models import MyModel

        model = MyModel(name='one', value=55)
        self.session.add(model)

        # Setup Vistes
        worker = Employee(
            email="psanlorenzo@vinissimus.com",
            start_day=datetime.date(2018, 7, 6)
        )
        worker2 = Employee(
            email="jordic@vinissimus.com",
            start_day=datetime.date(2018, 8, 21)
        )
        worker3 = Employee(
            email="ccarrillo@vinissimus.com",
            start_day=datetime.date(2014, 9, 1)
        )
        apply1 = HolidayApply(
            first_day=datetime.date(2019, 7, 12),
            last_day=datetime.date(2019, 7, 15),
            aproved=True
        )
        apply2 = HolidayApply(
            first_day=datetime.date(2019, 8, 3),
            last_day=datetime.date(2019, 8, 10),
            aproved=True
        )
        apply3 = HolidayApply(
            first_day=datetime.date(2018, 10, 18),
            last_day=datetime.date(2018, 10, 21),
            aproved=True
        )
        apply4 = HolidayApply(
            first_day=datetime.date(2018, 12, 28),
            last_day=datetime.date(2019, 1, 3),
            aproved=True
        )
        apply5 = HolidayApply(
            first_day=datetime.date(2018, 12, 28),
            last_day=datetime.date(2019, 1, 3),
            aproved=True
        )
        apply_pending = HolidayApply(
            first_day=datetime.date(2019, 4, 3),
            last_day=datetime.date(2019, 4, 7)
        )
        year = YearHolidays(year=2018, days=24)
        year2 = YearHolidays(year=2019, days=24)

        worker.manager_id = worker2.email
        worker3.manager_id = worker2.email
        apply1.user_id = worker.email
        apply2.user_id = worker3.email
        apply3.user_id = worker.email
        apply4.user_id = worker.email
        apply5.user_id = worker2.email
        apply_pending.user_id = worker.email

        self.session.add_all([worker, worker2, worker3])
        self.session.add_all([apply1, apply2, apply3, apply4, apply_pending, apply5])
        self.session.add_all([year, year2])

    def test_holidays_view(self):
        from .views.default import holidays

        holidays_view = holidays(dummy_request(self.session))

        # Vista principal
        self.assertEqual(holidays_view['user'], 'jordic@vinissimus.com')
        self.assertEqual(holidays_view['days'], 21)
        self.assertEqual(
            holidays_view['peticions'][0].scalar().user_id,
            'psanlorenzo@vinissimus.com'
        )
        self.assertFalse(
            holidays_view['peticions'][0].scalar().aproved
        )
        self.assertIs(
            type(holidays_view['peticions'][0].scalar().first_day),
            datetime.date
        )
        self.assertIs(
            type(holidays_view['peticions'][0].scalar().last_day),
            datetime.date
        )
        self.assertEqual(
            holidays_view['peticions_propies'][0].user_id,
            'jordic@vinissimus.com'
        )
        self.assertTrue(
            holidays_view['peticions_propies'][0].aproved
        )
        self.assertIs(
            type(holidays_view['peticions_propies'][0].first_day),
            datetime.date
        )
        self.assertIs(
            type(holidays_view['peticions_propies'][0].last_day),
            datetime.date
        )

    def test_holiday_apply_view(self):
        from .views.default import holiday_apply
        params = {'first_day': '2019-04-03', 'last_day': '2019-04-07'}

        holiday_apply_view = holiday_apply(dummy_request(self.session, params))

        # Vista holiday_apply
        self.assertEqual(
            holiday_apply_view['msg'],
            'Solicitud enviada!'
        )

    def test_holiday_accept_view(self):
        from .views.default import holiday_accept
        params = {'first_day': '2019-04-03', 'last_day': '2019-04-07', 'user': 'psanlorenzo@vinissimus.com', 'aprove': 'Acceptar'}

        holiday_accept_view = holiday_accept(dummy_request(self.session, params))

        # Vista holiday accept
        self.assertEqual(
            holiday_accept_view['msg'],
            'Solicitud aprovada!'
        )

    def test_holiday_edit_view(self):
        from .views.default import holiday_edit
        params = {'first_day': '2019-04-03', 'last_day': '2019-04-07', 'user': 'psanlorenzo@vinissimus.com'}

        holiday_edit_view = holiday_edit(dummy_request(self.session, params))

        # Vista holiday accept
        self.assertEqual(
            holiday_edit_view['user'].user_id,
            'psanlorenzo@vinissimus.com'
        )
        self.assertEqual(
            holiday_edit_view['user'].first_day,
            '2019-04-03'
        )
        self.assertEqual(
            holiday_edit_view['user'].last_day,
            '2019-04-07'
        )

    def test_holiday_edit_accept_view(self):
        from .views.default import holiday_edit_accept
        params = {'first_day': '2019-04-03', 'last_day': '2019-04-07', 'user': 'psanlorenzo@vinissimus.com', 'new_first_day': '2019-05-05', 'new_last_day': '2019-05-08'}

        holiday_edit_accept_view = holiday_edit_accept(dummy_request(self.session, params))

        # Vista holiday accept
        self.assertEqual(
            holiday_edit_accept_view['msg'],
            'Canvi realitzat'
        )