base_wamas_ubl: fix html escaping & test freezetime

Author: jbaudoux

base_wamas_ubl/lib/wamas/dict2wamas.py

@@ -5,6 +5,7 @@
 import logging
 from ast import literal_eval
 from pprint import pformat
+from freezegun import freeze_time
 
 from . import const, utils
 
@@ -60,6 +61,7 @@ def dict2wamas(dict_input, msg_type):
     return wamas
 
 
+@freeze_time("2024-02-11 22:14:22")
 def main():
     parser = argparse.ArgumentParser(
         description="Converts JSON document into message.",

base_wamas_ubl/lib/wamas/tests/test_dict2wamas.py

@@ -1,11 +1,16 @@
 import ast
 import unittest
 
+from freezegun import freeze_time
+
 from ..dict2wamas import dict2wamas
 from ..utils import file_open, file_path
 
 
 class TestDict2wamas(unittest.TestCase):
+
+    maxDiff = None
+
     def _test(self, msg_type, filename):
         with file_open(
             file_path("tests/samples/dict2wamas_input%s.dict" % filename)
@@ -17,9 +22,11 @@ def _test(self, msg_type, filename):
             expected_output = outfile.read()
             self.assertEqual(output, expected_output)
 
+    @freeze_time("2024-02-11 22:14:22")
     def test_LST(self):
         self._test("Supplier", "")
 
+    @freeze_time("2024-02-11 22:15:32")
     def test_KSTAUS(self):
         self._test("CustomerDeliveryPreferences", "_2")
 

base_wamas_ubl/lib/wamas/tests/test_ubl2wamas.py

@@ -5,6 +5,9 @@
 
 
 class TestUbl2wamas(unittest.TestCase):
+
+    maxDiff = None
+
     def test_helpers(self):
         dict_data = {
             "str": [

base_wamas_ubl/lib/wamas/tests/test_wamas2ubl.py

@@ -9,6 +9,9 @@
 
 
 class TestWamas2dict(unittest.TestCase):
+
+    maxDiff = None
+
     def _test(self, filename):
         with file_open(
             file_path("tests/samples/%s.wamas" % filename)

base_wamas_ubl/lib/wamas/tests/test_wamas2wamas.py

@@ -7,6 +7,9 @@
 
 
 class TestWamas2wamas(unittest.TestCase):
+
+    maxDiff = None
+
     def _test(self, filename):
         with file_open(
             file_path("tests/samples/wamas2wamas_input_%s.wamas" % filename)

base_wamas_ubl/lib/wamas/utils.py

@@ -408,7 +408,7 @@ def fw2dict(line, grammar, telegram_type):
             dp = fdef["dp"]
             val = float(b[:-dp] + "." + b[-dp:])
         else:
-            val = escape(b.rstrip())
+            val = str(escape(b.rstrip()))
         res[fname] = val
     _logger.debug(pformat(res))
     return res