def test_replace_item(): repo = Repository() repo.add_item(0) repo.add_item(1) repo.add_item(1) repo.add_item(1) repo.replace_item(lambda item: item == 1, 0) assert repo.get_items() == [0, 0, 1, 1]
def test_replace_item(self): repo = Repository() repo.add_item(0) repo.add_item(1) repo.add_item(1) uid = repo.add_item(1) repo.replace_item(0, condition=lambda item: item == 1) self.assertEqual(repo.get_items(), [0, 0, 1, 1]) repo.replace_item(2, uid=uid) self.assertEqual(repo.get_items(), [0, 0, 1, 2])
def test_remove_items(): repo = Repository() repo.add_item(0) repo.add_item(1) repo.add_item(1) repo.add_item(1) repo.remove_item(lambda item: item == 1) assert repo.get_items() == [0, 1, 1] repo.remove_items(lambda item: item == 1) assert repo.get_items() == [0]
def test_generate_id(self): repo = Repository() uids = [] for i in range(1000): uid = repo.add_item(1) self.assertNotIn(uid, uids) uids.append(uid)
def test_remove_items(self): repo = Repository() repo.add_item(0) repo.add_item(1) repo.add_item(1) repo.add_item(1) repo.remove_item(condition=lambda item: item == 1) self.assertEqual(repo.get_items(), [0, 1, 1]) uid = repo.add_item(2) repo.remove_item(uid=uid) self.assertEqual(repo.get_items(), [0, 1, 1]) repo.remove_items(lambda item: item == 1) self.assertEqual(repo.get_items(), [0])
def test_get_items(self): repo = Repository() repo.add_item(0) repo.add_item(1) self.assertEqual(repo.get_item(condition=lambda item: item == 1), 1) uid = repo.add_item(2) self.assertEqual(repo.get_item(uid=uid), 2) repo.add_item(1) self.assertEqual(repo.get_items(), [0, 1, 2, 1]) self.assertEqual(repo.get_items(lambda item: item == 1), [1, 1]) self.assertEqual(repo.get_items(lambda item: item > 2), []) repo.add_item(3) self.assertEqual(repo.get_items(lambda item: item > 2), [3])
def test_get_items(): repo = Repository() repo.add_item(0) repo.add_item(1) assert repo.get_item(lambda item: item == 1) == 1 repo.add_item(1) assert repo.get_items() == [0, 1, 1] assert repo.get_items(lambda item: item == 1) == [1, 1] assert not repo.get_items(lambda item: item > 1) repo.add_item(2) assert repo.get_items(lambda item: item > 1) == [2]
def test_activity_service(self): a_repo = Repository() p_repo = Repository() p_repo.add_item(Person('1', get_random_name(), get_random_phone_number()), get_uid=lambda p: p.uid) p_repo.add_item(Person('2', get_random_name(), get_random_phone_number()), get_uid=lambda p: p.uid) p_repo.add_item(Person('3', get_random_name(), get_random_phone_number()), get_uid=lambda p: p.uid) service = ActivityService(a_repo, p_repo) activity1 = Activity('1234', ['1', '2'], get_random_date(), get_random_time(), 'test1') activity2 = Activity('4321', ['1', '3'], get_random_date(), get_random_time(), 'test2') service.add_activity(activity1) self.assertEqual(service.get_activities(), [activity1]) service.add_activity(activity2) self.assertEqual(service.get_activities(), [activity1, activity2]) self.assertEqual(service.get_activity(activity1.uid), activity1) service.remove_activity(activity2.uid) self.assertEqual(service.get_activities(), [activity1]) CommandSystem.global_undo() self.assertEqual(service.get_activities(), [activity1, activity2]) CommandSystem.global_redo() self.assertEqual(service.get_activities(), [activity1]) CommandSystem.global_undo() new_activity = Activity(activity2.uid, ['1'], activity2.date, activity2.time, activity2.description) service.update_activity(activity2.uid, new_activity) self.assertEqual(service.get_activities(), [activity1, new_activity]) self.assertEqual(service.get_date_activities(activity1.date), [activity1]) self.assertEqual(service.get_busiest_days(), [(activity1.date, 1), (new_activity.date, 1)]) self.assertEqual(service.get_person_activities('2'), [activity1]) self.assertEqual(service.search_activities(None, None, 'test1'), [activity1])
def test_add_item(): repo = Repository() repo.add_item(1) assert repo.get_items() == [1]
def test_add_item(self): repo = Repository() repo.add_item(1) self.assertEqual(repo.get_items(), [1])