def test_higher_priority_1(self):
"""Both exercise groups are occupied by Bolek and Lolek. Tola will
enqueue to both with different priorities. She will end up in the group
of higher priority regardless of the order in which Bolek and Lolek free
up the places.
"""
with patch(RECORDS_DATETIME, mock_datetime(2011, 10, 1, 12)):
self.assertTrue(
Record.enqueue_student(self.bolek,
self.cooking_exercise_group_1))
self.assertTrue(
Record.enqueue_student(self.lolek,
self.cooking_exercise_group_2))
with patch(RECORDS_DATETIME, mock_datetime(2011, 10, 1, 13)):
self.assertTrue(
Record.enqueue_student(self.tola,
self.cooking_exercise_group_1))
self.assertTrue(
Record.set_queue_priority(self.tola,
self.cooking_exercise_group_1, 7))
self.assertTrue(
Record.enqueue_student(self.tola,
self.cooking_exercise_group_2))
self.assertTrue(
Record.set_queue_priority(self.tola,
self.cooking_exercise_group_2, 8))
self.assertTrue(
Record.is_recorded(self.tola, self.cooking_exercise_group_1))
self.assertTrue(
Record.is_recorded(self.tola, self.cooking_exercise_group_2))
self.assertFalse(
Record.is_enrolled(self.tola, self.cooking_exercise_group_1))
self.assertFalse(
Record.is_enrolled(self.tola, self.cooking_exercise_group_2))
with patch(RECORDS_DATETIME, mock_datetime(2011, 10, 8, 12)):
self.assertCountEqual(
Record.remove_from_group(self.bolek,
self.cooking_exercise_group_1),
[self.cooking_exercise_group_1.pk])
with patch(RECORDS_DATETIME, mock_datetime(2011, 10, 8, 13)):
self.assertCountEqual(
Record.remove_from_group(self.lolek,
self.cooking_exercise_group_2),
[self.cooking_exercise_group_2.pk])
self.assertFalse(
Record.is_recorded(self.tola, self.cooking_exercise_group_1))
self.assertTrue(
Record.is_enrolled(self.tola, self.cooking_exercise_group_2))
def test_higher_priority_2(self):
"""The only difference between this test and the one above is the order
in which Bolek and Lolek leave their groups.
"""
with patch(RECORDS_DATETIME, mock_datetime(2011, 10, 1, 12)):
self.assertTrue(
Record.enqueue_student(self.bolek,
self.cooking_exercise_group_1))
self.assertTrue(
Record.enqueue_student(self.lolek,
self.cooking_exercise_group_2))
with patch(RECORDS_DATETIME, mock_datetime(2011, 10, 1, 13)):
self.assertTrue(
Record.enqueue_student(self.tola,
self.cooking_exercise_group_1))
self.assertTrue(
Record.set_queue_priority(self.tola,
self.cooking_exercise_group_1, 7))
self.assertTrue(
Record.enqueue_student(self.tola,
self.cooking_exercise_group_2))
self.assertTrue(
Record.set_queue_priority(self.tola,
self.cooking_exercise_group_2, 8))
self.assertTrue(
Record.is_recorded(self.tola, self.cooking_exercise_group_1))
self.assertTrue(
Record.is_recorded(self.tola, self.cooking_exercise_group_2))
self.assertFalse(
Record.is_enrolled(self.tola, self.cooking_exercise_group_1))
self.assertFalse(
Record.is_enrolled(self.tola, self.cooking_exercise_group_2))
with patch(RECORDS_DATETIME, mock_datetime(2011, 10, 8, 12)):
self.assertCountEqual(
Record.remove_from_group(self.lolek,
self.cooking_exercise_group_2),
[self.cooking_exercise_group_2.pk])
with patch(RECORDS_DATETIME, mock_datetime(2011, 10, 8, 13)):
self.assertCountEqual(
Record.remove_from_group(self.bolek,
self.cooking_exercise_group_1),
[self.cooking_exercise_group_1.pk])
self.assertFalse(
Record.is_recorded(self.tola, self.cooking_exercise_group_1))
self.assertTrue(
Record.is_enrolled(self.tola, self.cooking_exercise_group_2))
