def setUp(self):
self.parser = Parser()
self.obfuscator = Obfuscator()
simple_doc = FixtureFiles.documents.get(FixtureFiles.simple_edi_file)
#don't modify the shared document based on the simple_edi_file, create a copy to modify.
self.simpleEdiDocument = copy.deepcopy(simple_doc)
self.obfuscated_doc = self.obfuscator.obfuscate(self.simpleEdiDocument)
def setUp(self):
self.goodISAWithAsterisk = "ISA*00* *00* *12*8005551234AA *12*8005556789BB *110408*1221*U*00401*000006617*0*P*>~"
self.goodISAWithPipe = "ISA|00| |00| |12|8005551234AA |12|8005556789BB |110408|1221|U|00401|000006617|0|P|>~"
self.goodISAWithNewlineTerminator = "ISA*00* *00* *12*8005551234AA *12*8005556789BB *110408*1221*U*00401*000006617*0*P*>\n"
self.badISA = "PSA*00* *00* *12*8005551234AA *12*8005556789BB *110408*1221*U*00401*000006617*0*P*>~"
self.goodISAWithLeadingSpaces = " ISA*00* *00* *12*8005551234AA *12*8005556789BB *110408*1221*U*00401*000006617*0*P*>~"
self.twoSegments = "ISA*00* *00* *12*8005551234AA *12*8005556789BB *110408*1221*U*00401*000006617*0*P*>~IEA*0*000006617~"
self.newLineTerminator = "ISA*00* *00* *12*8005551234AA *12*8005556789BB *110408*1221*U*00401*000006617*0*P*>\nIEA*0*000006617\n"
self.extraTerminator = "ISA*00* *00* *12*8005551234AA *12*8005556789BB *110408*1221*U*00401*000006617*0*P*>~\nIEA*0*000006617~\n"
self.nonEDIFile = "<xml><test></test></xml>"
self.emptySegmentsWithNoTerminator = "ISA****************"
self.emptySegmentsWithTerminator = "ISA****************>~"
self.edi_with_ISA_GS_GE_IEA = "ISA*00* *00* *12*8005551234AA *12*8005556789BB *110408*1221*U*00401*000006617*0*P*>~GS*IN*8005551234A*8005556789B*20110408*1221*6617*X*004010~GE*6*6617~IEA*1*000006617"
self.parser = Parser()
def read_edi_file(self, file_path):
"""Read in a file and parse it to an EDI document
Attributes:
file_path -- path to the file to read.
"""
handle = open(file_path, 'r')
try:
edi_text = handle.read()
edi_document = Parser().parse_document(document_text=edi_text)
FixtureFiles.documents[file_path] = edi_document
finally:
handle.close()
return edi_document
def setUp(self):
self.parser = Parser()
self.simpleEdiDocument = FixtureFiles.documents.get(
FixtureFiles.simple_edi_file)
self.int_header = InterchangeHeader()
self.int_header.isa01.content = "00"
self.int_header.isa02.content = " "
self.int_header.isa03.content = "00"
self.int_header.isa04.content = " "
self.int_header.isa05.content = "12"
self.int_header.isa06.content = "8005551234AA "
self.int_header.isa07.content = "12"
self.int_header.isa08.content = "8005555678BB "
self.int_header.isa09.content = "131022"
self.int_header.isa10.content = "1400"
self.int_header.isa11.content = "U"
self.int_header.isa12.content = "00401"
self.int_header.isa13.content = "000003821"
self.int_header.isa14.content = "0"
self.int_header.isa15.content = "P"
self.int_header.isa16.content = ">"
def test_isa_format_as_edi(self):
good_isa_edi = \
"ISA*00* *00* *12*8005551234AA *12*8005556789BB *110408*1221*U*00401*000006617*0*P*>~"
good_isa_document = Parser().parse_document(good_isa_edi)
formatted_document = good_isa_document.format_as_edi()
self.assertEqual(good_isa_edi, formatted_document)
def test_interchange_header_trailer_format_as_edi(self):
interchange_edi = \
"ISA*00* *00* *12*8005551234AA *12*8005556789BB *110408*1221*U*00401*000006617*0*P*>\nIEA*0*000006617\n"
interchange_document = Parser().parse_document(interchange_edi)
formatted_document = interchange_document.format_as_edi()
self.assertEqual(interchange_edi, formatted_document)
def test_isa_with_alternate_separators_format_as_edi(self):
good_isa_edi = \
"ISA|22| |00| |12|8005551234AA |12|8005556789BB |110408|1221|U|00401|000006617|0|P|>\n"
good_isa_document = Parser().parse_document(good_isa_edi)
formatted_document = good_isa_document.format_as_edi()
self.assertEqual(good_isa_edi, formatted_document)
