You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
1.3 KiB
28 lines
1.3 KiB
import pytest
|
|
from gauth_reencoder import decode_totp
|
|
from gauth_reencoder import generate_new_url
|
|
|
|
@pytest.mark.parametrize("url, account, issuer, secret, algorithm, digits", [
|
|
("otpauth://totp/DEF?algorithm=&digits=&issuer=ABC&secret=jidowio390u20", "DEF", "ABC", "jidowio390u20", "", ""),
|
|
("otpauth://totp/Label?secret=jidowio390u20&issuer=Issuer", "Label", "Issuer", "jidowio390u20", "", ""),
|
|
])
|
|
def test_decode_totp(url: str, account: str, issuer: str, secret: str, algorithm: str, digits: str):
|
|
decoded = decode_totp(url)
|
|
assert decoded["account"] == account
|
|
assert decoded["issuer"] == issuer
|
|
assert decoded["secret"] == secret
|
|
assert decoded["algorithm"] == algorithm
|
|
assert decoded["digits"] == digits
|
|
|
|
@pytest.mark.parametrize("url, account, issuer, secret, algorithm, digits", [
|
|
("otpauth://totp/DEF?algorithm=&digits=&issuer=ABC&secret=jidowio390u20", "DEF", "ABC", "jidowio390u20", "", ""),
|
|
])
|
|
def test_generate_new_url(url: str, account: str, issuer: str, secret: str, algorithm: str, digits: str):
|
|
decoded = dict()
|
|
decoded["account"] = account
|
|
decoded["issuer"] = issuer
|
|
decoded["secret"] = secret
|
|
decoded["algorithm"] = algorithm
|
|
decoded["digits"] = digits
|
|
new_url = generate_new_url(decoded)
|
|
assert new_url == url
|