def setUp(self) -> None: self.field_definition = FieldDefinition('dob', 'XML') self.file_definition = FileDefinition('XML', [self.field_definition], has_header=True, delimiter='|', record_element_name='person', input_directory='input', completed_directory='completed', file_mask='mask')
def setUp(self): super(FixedFileServiceTests, self).create_file_definitions(fc.FIXED) self.file_definition = FileDefinition(fc.FIXED, self.field_definitions, True) self.file_service = create_file_service(self.file_definition) self.file_name = 'Valid-fixed-09032019.txt' with open(self.file_name, 'w') as f: f.write('Name Age DOB \n') f.write('Aaron 39 09/04/1980\n') f.write('Gene 61 01/15/1958\n') f.write('Xander 4 11/22/2014\n') f.write('Mason 12 04/13/2007\n')
def setUp(self): super().create_file_definitions(fc.DELIMITED) self.file_definition = FileDefinition(fc.DELIMITED, self.field_definitions, True, '|') self.file_service = FlatFileService(self.file_definition) self.file_name = 'Valid-delimited-09032019.txt' with open(self.file_name, 'w') as f: f.write('Name|DOB|Age\n') f.write('Aaron|09/04/1980|39\n') f.write('Gene|01/15/1958|61\n') f.write('Xander|11/22/2014|4\n') f.write('Mason|04/13/2007|12\n')
def setUp(self): super(XmlFileServiceTests, self).create_file_definitions(fc.XML) self.file_definition = FileDefinition(fc.XML, self.field_definitions, record_element_name='person') self.file_service = XmlFileService(self.file_definition) self.file_name = 'Valid-xml-09032019.txt' with open(self.file_name, 'w') as f: self.file = f f.write('<people>\n') self.write_xml_record('Aaron', 39, '09/04/1980') self.write_xml_record('Gene', 61, '01/15/1958') self.write_xml_record('Xander', 4, '11/22/2014') self.write_xml_record('Mason', 12, '04/13/2007') f.write('</people>\n')
def create_file_definition(file_type, delimiter=None, record_element_name=None, start_position=None, field_length=None, position_in_row=None): field_definitions = [ FieldDefinition('name', file_type, start_position=start_position, field_length=field_length, position_in_row=position_in_row) ] return FileDefinition(file_type, field_definitions, record_element_name=record_element_name, delimiter=delimiter)
def setUp(self) -> None: self.file_definition = FileDefinition.create_from_dict({ fc.FILE_TYPE: fc.DELIMITED, fc.DELIMITER: ',', fc.RECORD_ELEMENT_NAME: 'record', fc.HAS_HEADER: True, fc.INPUT_DIRECTORY: 'input', fc.COMPLETED_DIRECTORY: 'completed', fc.FILE_MASK: 'mask', fc.FIELD_DEFINITIONS: [{ fc.FIELD_NAME: 'name', fc.POSITION_IN_ROW: 0 }] })
def test_file_type_missing(self): with self.assertRaises(ValueError): FileDefinition('', [self.file_definition])
def test_xml_file_record_element_name_missing(self): with self.assertRaises(ValueError): FileDefinition(fc.XML, [self.field_definition], record_element_name='')
def test_delimited_file_delimiter_missing(self): with self.assertRaises(ValueError): FileDefinition(fc.DELIMITED, [self.field_definition], delimiter='')
def test_file_definitions_missing(self): with self.assertRaises(ValueError): FileDefinition('XML', [])
def test_process_records_not_implemented(self): file_service = FileService( FileDefinition(fc.FIXED, [FieldDefinition('hello', 'file_type')])) with self.assertRaises(NotImplementedError): file_service.process_file_records(None)