1
1
import usrcheck
2
2
import hashlib
3
3
def test_auth ():
4
- usrcheck .save_users ([['testusr' ,hashlib .sha256 ('testPSWD' .encode ()).hexdigest ()]],True )
5
- assert usrcheck .load_users () == [['testusr' ,hashlib .sha256 ('testPSWD' .encode ()).hexdigest ()]], "Userlist loading failed"
6
- assert usrcheck .usr_check ('testusr' , 'testPSWD' ,[['testusr' ,hashlib .sha256 ('testPSWD' .encode ()).hexdigest ()]]) == (True ,True ), "User check failed"
4
+ usr_list = [['testusr' ,hashlib .sha256 (str ('testPSWD' + '1QgF35ws' ).encode ()).hexdigest (),'1QgF35ws' ]]
5
+ usrcheck .save_users (usr_list ,True )
6
+ assert usrcheck .load_users () == usr_list , "Userlist loading failed"
7
+ assert usrcheck .usr_check ('testusr' , 'testPSWD' ,usr_list ) == (True ,True ), "User check failed"
7
8
#Wrong username
8
- assert usrcheck .usr_check ('wrongusr' , 'testPSWD' ,[[ 'testusr' , hashlib . sha256 ( 'testPSWD' . encode ()). hexdigest ()]] ) == (False ,False ), "User check failed"
9
+ assert usrcheck .usr_check ('wrongusr' , 'testPSWD' ,usr_list ) == (False ,False ), "User check failed"
9
10
#Wrong password
10
- assert usrcheck .usr_check ('testusr' , 'wrongPSWD' ,[[ 'testusr' , hashlib . sha256 ( 'testPSWD' . encode ()). hexdigest ()]] ) == (True ,False ), "User check failed"
11
+ assert usrcheck .usr_check ('testusr' , 'wrongPSWD' ,usr_list ) == (True ,False ), "User check failed"
11
12
test_auth ()
0 commit comments