def test_usage_pattern(self): sched = schedules() sched.add(self.A) sched.add(self.C) self.assertTrue(self.A in sched) self.assertTrue(self.C in sched) self.assertFalse(self.B in sched) self.assertRaises(schedule_error, sched.add, self.B) self.assertFalse(self.B in sched) self.assertEqual(sched, sched) sched.remove(self.A) self.assertFalse(self.A in sched) self.assertFalse(self.B in sched) self.assertTrue(self.C in sched) sched.remove(self.C) self.assertFalse(self.B in sched) self.assertFalse(self.C in sched)
def test_equality(self): sched1 = schedules() sched2 = schedules() self.assertEqual(sched1, sched2) sched1.add(self.A) sched1.add(self.B) sched2.add(self.A) sched2.add(self.B) sched2.add(self.C) self.assertNotEqual(sched1, sched2) sched1.add(self.C) self.assertEqual(sched1, sched2)
def test_store(self): fake_file = self.mocker.mock() fake_file.store_object('test_schedule', IN(self.tasks)) self.mocker.count(len(self.tasks)) self.mocker.replay() schedule = schedules('test_schedule') for task in self.tasks: schedule.add(task) schedule.store(fake_file)
def test_save_and_restore(self): sched1 = schedules() sched1.add(self.A) sched1.add(self.B) sched1.add(self.C) store1 = file('test_schedules_and_file.sqlite') sched1.store(store1) del sched1 del store1 store2 = file('test_schedules_and_file.sqlite') sched2 = schedules.load(store2) self.assertEqual(set([x.begins for x in sched2.tasks]), set([3, 5, 7]))
def test_usage_pattern(self): sched1 = schedules() sched1.add(self.A) sched1.add(self.B) sched1.add(self.C) sched1.add(self.D) sched1.add(self.E) store1 = file('test_system.sqlite') sched1.store(store1) del store1 store2 = file('test_system.sqlite') sched2 = schedules.load(store2) self.assertEqual(sched1, sched2) sched2.remove(self.D) sched2.remove(self.E) self.assertNotEqual(sched1, sched2) sched2.add(self.F) self.assertTrue(self.F in sched2) self.assertFalse(self.F in sched1) self.assertRaises(schedule_error, sched2.add, self.D) self.assertRaises(schedule_error, sched2.add, self.E) self.assertTrue(self.A in sched1) self.assertTrue(self.B in sched1) self.assertTrue(self.C in sched1) self.assertTrue(self.D in sched1) self.assertTrue(self.E in sched1) self.assertFalse(self.F in sched1) self.assertTrue(self.A in sched2) self.assertTrue(self.B in sched2) self.assertTrue(self.C in sched2) self.assertFalse(self.D in sched2) self.assertFalse(self.E in sched2) self.assertTrue(self.F in sched2)
def test_add_distinct_exclude(self): schedule = schedules() schedule.add(self.distinct_include) schedule.add(self.distinct_exclude)
def test_in_after_add(self): schedule = schedules() schedule.add(self.fake) self.assertTrue(self.fake in schedule)
def test_in_before_add(self): schedule = schedules() self.assertFalse(self.fake in schedule)
def test_add_over_distinct_include(self): schedule = schedules() schedule.add(self.distinct_include) schedule.add(self.overlap_include)
def test_add_over_distinct_exclude(self): schedule = schedules() schedule.add(self.distinct_exclude) self.assertRaises(schedule_error, schedule.add, self.overlap_include)