3 changed files with 29 additions and 0 deletions
@ -0,0 +1,28 @@ |
|||
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 |
Loading…
Reference in new issue