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