From b96f828e7be6b0494625ad9b24ea3715a8400d18 Mon Sep 17 00:00:00 2001
From: Tomas Kopecek <tkopecek@redhat.com>
Date: Oct 15 2019 10:55:14 +0000
Subject: fix test for RHEL6


---

diff --git a/tests/test_cli/test_edit_user.py b/tests/test_cli/test_edit_user.py
index 7d9768e..c3ee1dd 100644
--- a/tests/test_cli/test_edit_user.py
+++ b/tests/test_cli/test_edit_user.py
@@ -116,7 +116,11 @@ Options:
         # Finally, assert that things were called as we expected.
         activate_session_mock.assert_not_called()
         session.editUser.assert_not_called()
-        self.assertEqual(cm.exception.code, 2)
+        if isinstance(cm.exception, int):
+            # RHEL6
+            self.assertEqual(cm.exception, 2)
+        else:
+            self.assertEqual(cm.exception.code, 2)
 
 
 if __name__ == '__main__':