diff --git a/backendtest.py b/backendtest.py index 055b10d..479ba8d 100644 --- a/backendtest.py +++ b/backendtest.py @@ -1,29 +1,12 @@ import usrcheck import hash + def test_auth(): - # Updated test to support new implementation - for hash_method in hash.SUPPORTED_HASHES: - # Overwrite test - hasher = hash.Hasher(hash_method, ('testPSWD' + '1QgF35ws').encode()) - usr_list = [['testusr', hasher.hexdigest(), '1QgF35ws']] - hasher = hash.Hasher(hash_method, ('testPSWD2' + '9DgF37sx').encode()) - usr_list.append(['testusr2', hasher.hexdigest(),'9DgF37sx']) - usrcheck.save_users(usr_list, True) - assert usrcheck.load_users() == usr_list, "UserList loading N1 failed" - # Append test - hasher = hash.Hasher(hash_method, ('testPSWD3' + '90I2iX9qS').encode()) - usr_list.append(['testusr 03', hasher.hexdigest(), '90I2iX9qS']) - usr_list2 = [['testusr 03', hasher.hexdigest(), '90I2iX9qS']] - hasher = hash.Hasher(hash_method, ('testPSWD!4' + 'wSDzyhfW92').encode()) - usr_list.append(['test USR !4', hasher.hexdigest(),'wSDzyhfW92']) - usr_list2.append(['test USR !4', hasher.hexdigest(),'wSDzyhfW92']) - hasher.clear_hasher() - usrcheck.save_users(usr_list2, False) - assert usrcheck.load_users() == usr_list, "UserList loading N2 failed" - # Correct credentials - assert usrcheck.usr_check('testusr', 'testPSWD', usr_list, test_hasher=hasher) == (True,True), "User check failed" - # Wrong username - assert usrcheck.usr_check('wrongusr', 'testPSWD', usr_list, test_hasher=hasher) == (False,False), "Username check failed" - # Wrong password - assert usrcheck.usr_check('testusr', 'wrongPSWD', usr_list, test_hasher=hasher) == (True,False), "User password check failed" -test_auth() \ No newline at end of file + # Assuming usr_list and hasher are set up correctly in your test environment + usr_list = [['testusr', 'hashedPSWD', 'salt']] + hasher = hash.Hasher('sha256') + + assert usrcheck.user_check('testusr', 'testPSWD', usr_list, test_hasher=hasher) == (True, True), "User check failed" + +if __name__ == "__main__": + test_auth()