Ejemplo n.º 1
0
class DiningListTestCase(TestCase):
    @classmethod
    def setUpTestData(cls):
        cls.user = User.objects.create_user('ankie')
        cls.association = Association.objects.create()

    def setUp(self):
        self.dining_list = DiningList(date=date(2123, 1, 2),
                                      sign_up_deadline=datetime(
                                          2100, 2, 2, tzinfo=timezone.utc),
                                      association=self.association)

    def test_is_open(self):
        list = DiningList.objects.create(date=date(2015, 1, 1),
                                         association=self.association,
                                         sign_up_deadline=datetime(
                                             2015,
                                             1,
                                             1,
                                             17,
                                             00,
                                             tzinfo=timezone.utc))

        with patch.object(timezone,
                          'now',
                          return_value=datetime(2015,
                                                1,
                                                1,
                                                16,
                                                59,
                                                tzinfo=timezone.utc)):
            self.assertTrue(list.is_open())
        with patch.object(timezone,
                          'now',
                          return_value=datetime(2015,
                                                1,
                                                1,
                                                17,
                                                00,
                                                tzinfo=timezone.utc)):
            self.assertFalse(list.is_open())
        with patch.object(timezone,
                          'now',
                          return_value=datetime(2015,
                                                1,
                                                1,
                                                17,
                                                1,
                                                tzinfo=timezone.utc)):
            self.assertFalse(list.is_open())

    def test_is_owner(self):
        self.dining_list.save()
        self.dining_list.owners.add(self.user)
        self.assertTrue(self.dining_list.is_owner(self.user))

    def test_is_owner_false(self):
        self.dining_list.save()
        self.assertFalse(self.dining_list.is_owner(self.user))