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)