Better assert formatting for PKey

geofront-auth · Feb 13, 2015 · ac8fde7 · ac8fde7
1 parent 75a719b
commit ac8fde7
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -2,6 +2,7 @@
 import os
 import threading
 
+from paramiko.pkey import PKey
 from paramiko.rsakey import RSAKey
 from paramiko.sftp_client import SFTPClient
 from paramiko.transport import Transport
@@ -108,6 +109,16 @@ def pytest_addoption(parser):
                      **env_default('GITHUB_TEAM_SLUGS'))
 
 
+def pytest_assertrepr_compare(op, left, right):
+    if op == '==' and isinstance(left, PKey) and isinstance(right, PKey):
+        left_key = format_openssh_pubkey(left)
+        right_key = format_openssh_pubkey(right)
+        return [
+            '{!r} == {!r}'.format(left, right),
+            '   {} != {}'.format(left_key, right_key)
+        ]
+
+
 used_port = 0