def testRecordSource(self):
dst = test_person(1)
src0 = test_person(0)
src1 = test_animal(0)
dst.note_src_record(src0)
dst.note_src_record(src1)
self.assertIn(src0.serial, dst.get_src_record_serials())
self.assertIn(src1.serial, dst.get_src_record_serials())
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 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 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 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 testSize(self):
rs = self._createRecordSet()
initial = rs.size
person = test_person(0)
person.freeze()
rs.add_record(person)
person1 = test_person(1)
person1.freeze()
rs.add_record(person1)
self.assertEqual(rs.size - initial, person.size + person1.size)
def testCount(self):
rs = self._createRecordSet()
initial = rs.count
person = test_person(0)
person.freeze()
rs.add_record(person)
person1 = test_person(1)
person1.freeze()
rs.add_record(person1)
self.assertEqual(rs.count - initial, 2)
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 _createRecordSet(self):
rs = EtlRecordSet(size_until_disk=50)
for i in range(5):
person = test_person(0)
person.freeze()
rs.add_record(person)
assert(rs.on_disk)
return rs
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 testConvertsToDisk(self):
rs = EtlRecordSet(size_until_disk=50)
self.assertFalse(rs.on_disk)
for i in range(5):
person = test_person(0)
person.freeze()
rs.add_record(person)
self.assertTrue(rs.on_disk)
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 testClone(self):
rec = test_person(0)
cloned = rec.clone()
self.assertEqual(rec, cloned)
def testAssertNotFrozen(self):
rec = test_person(0)
rec.assert_not_frozen()
rec.freeze()
with self.assertRaises(EtlRecordFrozen):
rec.assert_not_frozen()
def testCantUpdateFrozen(self):
rec = test_person(1)
rec.freeze()
with self.assertRaises(EtlRecordFrozen):
rec['first'] = "new"
def testFreeze(self):
rec = test_person(1)
rec.freeze()
self.assertTrue(rec.is_frozen,
"Record should be frozen")
def testSetFieldValue(self):
rec = test_person(1)
rec['first'] = "Jane"
rec['age'] = 20
self.assertEquals(rec, test_person(1))
def testGetFieldValue(self):
rec = test_person(0)
self.assertEquals(rec['first'], "John")
self.assertEquals(rec['last'], "Doe")
self.assertEquals(rec['age'], 22)
def testCeckRecordStruct(self):
person = test_person(0)
self.assertIsNone(PersonTestScehma().check_record_struct(person))
def testFieldNames(self):
self.assertEquals(test_person(0).field_names(),
['first', 'last', 'age'])
def testIfSchemasNotEquals(self):
self.assertNotEqual(test_person(0), test_animal(0))
def testIfValuesNotEquals(self):
self.assertNotEqual(test_person(0), test_person(1))
def testEquals(self):
self.assertEqual(test_person(0), test_person(0))
def testSerial(self):
self.assertIsNotNone(test_person(0).serial,
"Serial should not be None")
def testAddRecord(self):
rs = MemoryRecordSet()
person = test_person(0)
person.freeze()
rs.add_record(person)
def testSourceProcessor(self):
rec = test_person(0)
rec.set_source('processor1', 'output1')
self.assertEqual(rec.source_processor_name, 'processor1')
self.assertEqual(rec.source_processor_output_name, 'output1')
def testSize(self):
rec = test_person(0)
self.assertLess(abs(rec.size - 20), 5) # w/in 5 of 20
def testValues(self):
self.assertEqual(test_person(0).values,
{'first': "John", 'last': "Doe", 'age': 22})
def testFieldNamesWithoutSchema(self):
rec = test_person(0)
rec.set_schema(None)
self.assertEquals(sorted(rec.field_names()),
sorted(['first', 'last', 'age']))
