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))
