aboutsummaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/hazmat/primitives/twofactor/test_hotp.py8
-rw-r--r--tests/hazmat/primitives/twofactor/test_totp.py5
2 files changed, 4 insertions, 9 deletions
diff --git a/tests/hazmat/primitives/twofactor/test_hotp.py b/tests/hazmat/primitives/twofactor/test_hotp.py
index ba40488a..3359dac2 100644
--- a/tests/hazmat/primitives/twofactor/test_hotp.py
+++ b/tests/hazmat/primitives/twofactor/test_hotp.py
@@ -14,7 +14,6 @@ from cryptography.hazmat.primitives import hashes
from cryptography.hazmat.primitives.hashes import MD5, SHA1
from cryptography.hazmat.primitives.twofactor import InvalidToken
from cryptography.hazmat.primitives.twofactor.hotp import HOTP
-from cryptography.hazmat.primitives.twofactor.utils import get_provisioning_uri
from ....utils import (
load_nist_vectors, load_vectors_from_file, raises_unsupported_algorithm
@@ -97,18 +96,15 @@ class TestHOTP(object):
secret = b"12345678901234567890"
hotp = HOTP(secret, 6, SHA1(), backend)
- assert get_provisioning_uri(hotp, "Alice Smith", counter=1) == (
+ assert hotp.get_provisioning_uri("Alice Smith", 1) == (
"otpauth://hotp/Alice%20Smith?digits=6&secret=GEZDGNBV"
"GY3TQOJQGEZDGNBVGY3TQOJQ&algorithm=SHA1&counter=1")
- assert get_provisioning_uri(hotp, "Alice Smith", 'Foo', counter=1) == (
+ assert hotp.get_provisioning_uri("Alice Smith", 1, issuer='Foo') == (
"otpauth://hotp/Foo:Alice%20Smith?digits=6&secret=GEZD"
"GNBVGY3TQOJQGEZDGNBVGY3TQOJQ&algorithm=SHA1&issuer=Foo"
"&counter=1")
- with pytest.raises(RuntimeError):
- get_provisioning_uri(hotp, "Alice Smith", 'World') # counter lost
-
def test_invalid_backend():
secret = b"12345678901234567890"
diff --git a/tests/hazmat/primitives/twofactor/test_totp.py b/tests/hazmat/primitives/twofactor/test_totp.py
index 94c696f9..cd841ba6 100644
--- a/tests/hazmat/primitives/twofactor/test_totp.py
+++ b/tests/hazmat/primitives/twofactor/test_totp.py
@@ -11,7 +11,6 @@ from cryptography.hazmat.backends.interfaces import HMACBackend
from cryptography.hazmat.primitives import hashes
from cryptography.hazmat.primitives.twofactor import InvalidToken
from cryptography.hazmat.primitives.twofactor.totp import TOTP
-from cryptography.hazmat.primitives.twofactor.utils import get_provisioning_uri
from ....utils import (
load_nist_vectors, load_vectors_from_file, raises_unsupported_algorithm
@@ -131,11 +130,11 @@ class TestTOTP(object):
secret = b"12345678901234567890"
totp = TOTP(secret, 6, hashes.SHA1(), 30, backend=backend)
- assert get_provisioning_uri(totp, "Alice Smith") == (
+ assert totp.get_provisioning_uri("Alice Smith") == (
"otpauth://totp/Alice%20Smith?digits=6&secret=GEZDGNBVG"
"Y3TQOJQGEZDGNBVGY3TQOJQ&algorithm=SHA1&period=30")
- assert get_provisioning_uri(totp, "Alice Smith", 'World') == (
+ assert totp.get_provisioning_uri("Alice Smith", 'World') == (
"otpauth://totp/World:Alice%20Smith?digits=6&secret=GEZ"
"DGNBVGY3TQOJQGEZDGNBVGY3TQOJQ&algorithm=SHA1&issuer=World"
"&period=30")