Add some tests before the end of the day

Author: theangryangel

wip.py renamed to tests/test_mapping.py

@@ -1,5 +1,21 @@
+#    pydifact - a python edifact library
+#    Copyright (C) 2021 Karl Southern
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU Lesser General Public License as published
+#    by the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU Lesser General Public License for more details.
+#
+#    You should have received a copy of the GNU Lesser General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+import unittest
 from pydifact.segmentcollection import Interchange
-from pydifact import Segment, mapping, serializer
+from pydifact import Segment, mapping
 
 
 class BGM(Segment):
@@ -14,42 +30,6 @@ class LIN(Segment):
     tag = "LIN"
 
 
-interchange = Interchange.from_str(
-    """UNA:+.?*'
-UNB+UNOA:4+5021376940009:14+1111111111111:14+200421:1000+0001+ORDERS'
-UNH+1+ORDERS:D:01B:UN:EAN010'
-BGM+220+123456+9'
-DTM+137:20150410:102'
-DTM+2:20150710:102'
-FTX+DIN+++DELIVERY INSTRUCTIONS DESCRIPTION'
-NAD+BY+5021376940009::9'
-NAD+SU+1111111111111::9'
-RFF+IA:123456'
-NAD+ST+++Mr John Smith+The Bungalow:Hastings Road+Preston+:::Lancashire+SW1A 1AA'
-CTA+LB+:Mr John Smith'
-COM+01772 999999:TE'
-[email protected]:EM'
-CUX+2:GBP:9'
-TDT+20+++++SD'
-LIN+1++121354654:BP'
-IMD+F++:::TPRG item description'
-QTY+21:2'
-MOA+203:200.00'
-PRI+AAA:100.00'
-RFF+LI:1'
-LIN+1++121354654:BP'
-IMD+F++:::TPRG item description'
-QTY+21:2'
-MOA+203:200.00'
-PRI+AAA:100.00'
-RFF+LI:1'
-UNS+S'
-CNT+2:1'
-UNT+1+27'
-UNZ+1+0001'"""
-)
-
-
 class OrderLine(mapping.SegmentGroup):
     line_id = mapping.Segment("LIN", mandatory=True)
     description = mapping.Segment("IMD", mandatory=True)
@@ -81,37 +61,65 @@ class Order(mapping.SegmentGroup):
     cnt = mapping.Segment("CNT", mandatory=True)
 
 
-TYPE_TO_PARSER_DICT = {"ORDERS": Order}
-
-message = next(interchange.get_messages())
+SAMPLE = """UNA:+.?*'
+UNB+UNOA:4+5021376940009:14+1111111111111:14+200421:1000+0001+ORDERS'
+UNH+1+ORDERS:D:01B:UN:EAN010'
+BGM+220+123456+9'
+DTM+137:20150410:102'
+DTM+2:20150710:102'
+FTX+DIN+++DELIVERY INSTRUCTIONS DESCRIPTION'
+NAD+BY+5021376940009::9'
+NAD+SU+1111111111111::9'
+RFF+IA:123456'
+NAD+ST+++Mr John Smith+The Bungalow:Hastings Road+Preston+:::Lancashire+SW1A 1AA'
+CTA+LB+:Mr John Smith'
+COM+01772 999999:TE'
+[email protected]:EM'
+CUX+2:GBP:9'
+TDT+20+++++SD'
+LIN+1++121354654:BP'
+IMD+F++:::TPRG item description'
+QTY+21:2'
+MOA+203:200.00'
+PRI+AAA:100.00'
+RFF+LI:1'
+LIN+1++121354654:BP'
+IMD+F++:::TPRG item description'
+QTY+21:2'
+MOA+203:200.00'
+PRI+AAA:100.00'
+RFF+LI:1'
+UNS+S'
+CNT+2:1'
+UNT+1+27'
+UNZ+1+0001'"""
 
-cls = TYPE_TO_PARSER_DICT.get(message.type)
-if not cls:
-    raise NotImplementedError("Unsupported message type '{}'".format(message.type))
 
-obj = cls()
-obj.from_message(message)
+class MappingTest(unittest.TestCase):
 
-reconstituted = obj.to_message(message.reference_number, message.identifier)
+    def test_read_interchange(self):
+        interchange = Interchange.from_str(SAMPLE)
+        message = next(interchange.get_messages())
 
-# print(str(obj))
-# print(obj.purchase_order_id[0])
+        try:
+            obj = Order()
+            obj.from_message(message)
+        except Exception as err:
+            raise AssertionError(
+                "Could not read Message into Order mapping! {}".format(
+                    repr(err)
+                )
+            )
 
-assert isinstance(obj.purchase_order_id.to_segments(), BGM)
+    def test_ensure_mapped_bgm_segment(self):
+        interchange = Interchange.from_str(SAMPLE)
+        message = next(interchange.get_messages())
 
-# print(message.segments)
+        obj = Order()
+        obj.from_message(message)
 
-assert str(message) == str(
-    reconstituted
-), "Original message should match reconstituted message"
+        self.assertTrue(isinstance(obj.purchase_order_id.to_segments(), BGM))
 
-reconstituted_interchange = Interchange(
-    interchange.sender, interchange.recipient,
-    interchange.control_reference,
-    interchange.syntax_identifier,
-    interchange.delimiters,
-    interchange.timestamp
-)
-reconstituted_interchange.add_message(reconstituted)
 
-assert reconstituted_interchange.serialize() == interchange.serialize(), "Original interchange should match reconstituted interchange"
+if __name__ == "__main__":
+    unittest.main()