From 355681fd042f4b7867f0fe83f5fe5fef04f07d8e Mon Sep 17 00:00:00 2001
From: farhaanbukhsh <farhaan.bukhsh@gmail.com>
Date: Jan 19 2016 15:21:02 +0000
Subject: Fix all tests in lib


---

diff --git a/pagure/lib/login.py b/pagure/lib/login.py
index 663ea4d..553886e 100644
--- a/pagure/lib/login.py
+++ b/pagure/lib/login.py
@@ -12,6 +12,8 @@ import random
 import string
 import bcrypt
 
+import hashlib
+import pagure
 from pagure.lib import model
 from kitchen.text.converters import to_unicode, to_bytes
 from cryptography.hazmat.primitives import constant_time
diff --git a/tests/test_progit_lib_login.py b/tests/test_progit_lib_login.py
index ea4a37e..4edae54 100644
--- a/tests/test_progit_lib_login.py
+++ b/tests/test_progit_lib_login.py
@@ -22,6 +22,7 @@ sys.path.insert(0, os.path.join(os.path.dirname(
     os.path.abspath(__file__)), '..'))
 
 import pagure.lib
+from pagure.exceptions import PagureException
 import pagure.lib.login
 import hashlib
 from pagure import APP
@@ -64,10 +65,17 @@ class PagureLibLogintests(tests.Modeltests):
             pagure.lib.login.check_password('foo', password)
         )
         password = '%s%s' % ('foo', APP.config.get('PASSWORD_SEED', None))
-        password = '$1$'+hashlib.sha512(password).hexdigest()
+        password = '$1$' + hashlib.sha512(password).hexdigest()
         self.assertTrue(
-             pagure.lib.login.check_password('foo', password)
+            pagure.lib.login.check_password('foo', password)
         )
+        password = '%s%s' % ('foo', APP.config.get('PASSWORD_SEED', None))
+        password = hashlib.sha512(password).hexdigest()
+        with self.assertRaises(PagureException):
+            pagure.lib.login.check_password('foo', password)
+        password = '$3$' + password
+        with self.assertRaises(PagureException):
+            pagure.lib.login.check_password('foo', password)
 
 if __name__ == '__main__':
     SUITE = unittest.TestLoader().loadTestsFromTestCase(PagureLibLogintests)