def testRemoveRecord(self): rs = MemoryRecordSet() person = test_person(0) person.freeze() rs.add_record(person, ['tagA', ]) person1 = test_person(1) person1.freeze() rs.add_record(person1, ['tagA', ]) self.assertTrue(rs.has_record(person.serial)) self.assertEqual(sorted(list(rs.find_records_with_tag('tagA'))), sorted([person, person1])) rs.remove_record(person.serial) self.assertFalse(rs.has_record(person.serial)) self.assertEqual(sorted(list(rs.find_records_with_tag('tagA'))), sorted([person1]))
def testHasRecord(self): rs = MemoryRecordSet() person = test_person(0) person.freeze() rs.add_record(person) person2 = test_person(1) person2.freeze() rs.add_record(person2) self.assertTrue(rs.has_record(person.serial))