1 files changed, 144 insertions, 0 deletions
diff --git a/tests/test_tblib.py b/tests/test_tblib.py
new file mode 100644
index 0000000..bade6d4
--- /dev/null
+++ b/tests/test_tblib.py
@@ -0,0 +1,144 @@
+import pickle
+import traceback
+
+from tblib import Traceback
+from tblib import pickling_support
+
+pickling_support.install()
+
+pytest_plugins = 'pytester',
+
+
+def test_parse_traceback():
+    tb1 = Traceback.from_string(
+        """
+Traceback (most recent call last):
+  File "file1", line 123, in <module>
+    code1
+  File "file2", line 234, in ???
+    code2
+  File "file3", line 345, in function3
+  File "file4", line 456, in
+    code4
+KeyboardInterrupt"""
+    )
+    pytb = tb1.as_traceback()
+    assert traceback.format_tb(pytb) == [
+        '  File "file1", line 123, in <module>\n',
+        '  File "file2", line 234, in ???\n',
+        '  File "file3", line 345, in function3\n',
+    ]
+    tb2 = Traceback(pytb)
+
+    expected_dict = {
+        "tb_frame": {
+            "f_code": {"co_filename": "file1", "co_name": "<module>"},
+            "f_globals": {"__file__": "file1", "__name__": "?"},
+            "f_lineno": 123,
+        },
+        "tb_lineno": 123,
+        "tb_next": {
+            "tb_frame": {
+                "f_code": {"co_filename": "file2", "co_name": "???"},
+                "f_globals": {"__file__": "file2", "__name__": "?"},
+                "f_lineno": 234,
+            },
+            "tb_lineno": 234,
+            "tb_next": {
+                "tb_frame": {
+                    "f_code": {"co_filename": "file3", "co_name": "function3"},
+                    "f_globals": {"__file__": "file3", "__name__": "?"},
+                    "f_lineno": 345,
+                },
+                "tb_lineno": 345,
+                "tb_next": None,
+            },
+        },
+    }
+    tb3 = Traceback.from_dict(expected_dict)
+    tb4 = pickle.loads(pickle.dumps(tb3))
+    assert tb4.as_dict() == tb3.as_dict() == tb2.as_dict() == tb1.as_dict() == expected_dict
+
+
+def test_pytest_integration(testdir):
+    test = testdir.makepyfile("""
+import six
+
+from tblib import Traceback
+
+def test_raise():
+    tb1 = Traceback.from_string('''
+Traceback (most recent call last):
+  File "file1", line 123, in <module>
+    code1
+  File "file2", line 234, in ???
+    code2
+  File "file3", line 345, in function3
+  File "file4", line 456, in ""
+''')
+    pytb = tb1.as_traceback()
+    six.reraise(RuntimeError, RuntimeError(), pytb)
+""")
+
+    # mode(auto / long / short / line / native / no).
+
+    result = testdir.runpytest_subprocess('--tb=long', '-vv', test)
+    result.stdout.fnmatch_lines([
+        "_ _ _ _ _ _ _ _ *",
+        "",
+        ">   [?][?][?]",
+        "",
+        "file1:123:*",
+        "_ _ _ _ _ _ _ _ *",
+        "",
+        ">   [?][?][?]",
+        "",
+        "file2:234:*",
+        "_ _ _ _ _ _ _ _ *",
+        "",
+        ">   [?][?][?]",
+        "",
+        "file3:345:*",
+        "_ _ _ _ _ _ _ _ *",
+        "",
+        ">   [?][?][?]",
+        "E   RuntimeError",
+        "",
+        "file4:456: RuntimeError",
+        "===*=== 1 failed in * ===*===",
+    ])
+
+    result = testdir.runpytest_subprocess('--tb=short', '-vv', test)
+    result.stdout.fnmatch_lines([
+        'test_pytest_integration.py:*: in test_raise',
+        '    six.reraise(RuntimeError, RuntimeError(), pytb)',
+        'file1:123: in <module>',
+        '    ???',
+        'file2:234: in ???',
+        '    ???',
+        'file3:345: in function3',
+        '    ???',
+        'file4:456: in ""',
+        '    ???',
+        'E   RuntimeError',
+    ])
+
+    result = testdir.runpytest_subprocess('--tb=line', '-vv', test)
+    result.stdout.fnmatch_lines([
+        "===*=== FAILURES ===*===",
+        "file4:456: RuntimeError",
+        "===*=== 1 failed in * ===*===",
+    ])
+
+    result = testdir.runpytest_subprocess('--tb=native', '-vv', test)
+    result.stdout.fnmatch_lines([
+        'Traceback (most recent call last):',
+        '  File "*test_pytest_integration.py", line *, in test_raise',
+        '    six.reraise(RuntimeError, RuntimeError(), pytb)',
+        '  File "file1", line 123, in <module>',
+        '  File "file2", line 234, in ???',
+        '  File "file3", line 345, in function3',
+        '  File "file4", line 456, in ""',
+        'RuntimeError',
+
+    ])