def test_encode_token():
jwt = Jwt()
first_name = 'john'
last_name = 'smith'
email = '*****@*****.**'
role = 'admin'
response = jwt.encode_token(first_name, last_name, email, role)
assert type(response) == str
def test_get_role():
jwt = Jwt()
first_name = 'john'
last_name = 'smith'
email = '*****@*****.**'
role = 'admin'
response = jwt.encode_token(first_name, last_name, email, role)
get_role = jwt.get_role(response)
assert role == get_role
def test_decode_token():
jwt = Jwt()
first_name = 'john'
last_name = 'smith'
email = '*****@*****.**'
role = 'admin'
response = jwt.encode_token(first_name, last_name, email, role)
payload = {
'first_name': first_name,
'last_name': last_name,
'email': email,
'role': role
}
decode_payload = jwt.decode_token(response)
assert payload == decode_payload
def test_valid_token():
jwt = Jwt()
jwt = Jwt(
token=jwt.encode_token('john', 'smith', '*****@*****.**', 'admin'))
assert jwt.is_valid == True
def test_invalid_token():
jwt = Jwt(token="randomstringthatisnotvalid")
assert jwt.is_valid == False