Fix str representation of TNEF

new file: tests/test_tnef.py modified: tnefparse/tnef.py
koodaamo · Nov 28, 2020 · a9172cb · a9172cb
1 parent 076fb99
commit a9172cb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/tests/test_tnef.py b/tests/test_tnef.py
@@ -0,0 +1,11 @@
+import os
+from tnefparse.tnef import TNEF
+
+THIS_DIR = os.path.dirname(os.path.abspath(__file__))
+
+
+def test_tnef_str_representation():
+    tnef_data = open(os.path.join(THIS_DIR, "examples", "two-files.tnef"), mode="rb").read()
+    t = TNEF(tnef_data)
+
+    assert str(t) == "<TNEF:0x237, 2 attachments>"
diff --git a/tnefparse/tnef.py b/tnefparse/tnef.py
@@ -337,7 +337,7 @@ def rtfbody(self):
 
     def __str__(self):
         atts = (", %i attachments" % len(self.attachments)) if self.attachments else ''
-        return f"<{self.__class__.__name__}:0x{self.key:2.2x}{atts}>"
+        return f"<{self.__class__.__name__}:0x{self.key:02x}{atts}>"
 
     def dump(self, force_strings=False):
         def get_data(a):