def testHasRecordWithTag(self):
rs = MemoryRecordSet()
person = test_person(0)
person.freeze()
rs.add_record(person, ['tagA'])
self.assertTrue(rs.has_record_with_tag('tagA'))
def testSize(self):
rs = MemoryRecordSet()
person = test_person(0)
person.freeze()
rs.add_record(person)
person1 = test_person(1)
person1.freeze()
rs.add_record(person1)
self.assertEqual(rs.size, person.size + person1.size)
def testCount(self):
rs = MemoryRecordSet()
person = test_person(0)
person.freeze()
rs.add_record(person)
person1 = test_person(1)
person1.freeze()
rs.add_record(person1)
self.assertEqual(rs.count, 2)
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))
def testGetRecord(self):
rs = MemoryRecordSet()
person = test_person(0)
person.freeze()
rs.add_record(person)
person1 = test_person(1)
person1.freeze()
rs.add_record(person1)
self.assertEqual(rs.get_record(person.serial), person)
def testFindOneRecordWithTag(self):
rs = MemoryRecordSet()
person = test_person(0)
person.freeze()
rs.add_record(person, ['tagA'])
self.assertEqual(list(rs.find_records_with_tag('tagA')),
[person, ])
person1 = test_person(1)
person1.freeze()
rs.add_record(person1, ['tagB'])
self.assertEqual(list(rs.find_records_with_tag('tagA')),
[person, ])
def testFindMultipleRecordWithTag(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.assertEqual(list(sorted(rs.find_records_with_tag('tagA'))),
sorted([person, person1]))
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 testAddRecord(self):
rs = MemoryRecordSet()
person = test_person(0)
person.freeze()
rs.add_record(person)
def testAddRecordWithTag(self):
rs = MemoryRecordSet()
person = test_person(0)
person.freeze()
rs.add_record(person, ['tagA'])
