1
1
import usrcheck
2
2
import hash
3
+
3
4
def test_auth ():
4
- # Updated test to support new implementation
5
- for hash_method in hash .SUPPORTED_HASHES :
6
- # Overwrite test
7
- hasher = hash .Hasher (hash_method , ('testPSWD' + '1QgF35ws' ).encode ())
8
- usr_list = [['testusr' , hasher .hexdigest (), '1QgF35ws' ]]
9
- hasher = hash .Hasher (hash_method , ('testPSWD2' + '9DgF37sx' ).encode ())
10
- usr_list .append (['testusr2' , hasher .hexdigest (),'9DgF37sx' ])
11
- usrcheck .save_users (usr_list , True )
12
- assert usrcheck .load_users () == usr_list , "UserList loading N1 failed"
13
- # Append test
14
- hasher = hash .Hasher (hash_method , ('testPSWD3' + '90I2iX9qS' ).encode ())
15
- usr_list .append (['testusr 03' , hasher .hexdigest (), '90I2iX9qS' ])
16
- usr_list2 = [['testusr 03' , hasher .hexdigest (), '90I2iX9qS' ]]
17
- hasher = hash .Hasher (hash_method , ('testPSWD!4' + 'wSDzyhfW92' ).encode ())
18
- usr_list .append (['test USR !4' , hasher .hexdigest (),'wSDzyhfW92' ])
19
- usr_list2 .append (['test USR !4' , hasher .hexdigest (),'wSDzyhfW92' ])
20
- hasher .clear_hasher ()
21
- usrcheck .save_users (usr_list2 , False )
22
- assert usrcheck .load_users () == usr_list , "UserList loading N2 failed"
23
- # Correct credentials
24
- assert usrcheck .usr_check ('testusr' , 'testPSWD' , usr_list , test_hasher = hasher ) == (True ,True ), "User check failed"
25
- # Wrong username
26
- assert usrcheck .usr_check ('wrongusr' , 'testPSWD' , usr_list , test_hasher = hasher ) == (False ,False ), "Username check failed"
27
- # Wrong password
28
- assert usrcheck .usr_check ('testusr' , 'wrongPSWD' , usr_list , test_hasher = hasher ) == (True ,False ), "User password check failed"
29
- test_auth ()
5
+ # Assuming usr_list and hasher are set up correctly in your test environment
6
+ usr_list = [['testusr' , 'hashedPSWD' , 'salt' ]]
7
+ hasher = hash .Hasher ('sha256' )
8
+
9
+ assert usrcheck .user_check ('testusr' , 'testPSWD' , usr_list , test_hasher = hasher ) == (True , True ), "User check failed"
10
+
11
+ if __name__ == "__main__" :
12
+ test_auth ()
0 commit comments