class TestUtils(APITestCase):
    def test_unique_device_code(self):
        first_code = id_generator()
        second_code = id_generator()

        self.assertNotEqual(first_code, second_code)

    def test_device_code_len(self):
        first_code = id_generator(100)
        second_code = id_generator(1)

        self.assertEqual(len(first_code), 100)
        self.assertEqual(len(second_code), 1)


class TestSprinkleScheduleOperations(APITestCase):

    def test_initial_schedule(self):
        user = User.objects.create_user('any_user', password='randomString')
        device = Device(name='one_device', owner=user)
        schedule_parameters = {
            'device': device,
            'hour': 10,
            'minute': 12,
            'am_pm': 'am',
            'on_monday': True,
            'on_tuesday': True,
            'on_wednesday': True,
            'on_thursday': True,
            'on_friday': True,
            'on_saturday': True,
            'on_sunday': True,
        }
        hello = SprinkleSchedule(**schedule_parameters)
        next_schedule = hello.next_day()
        print(next_schedule)