Skip to content

Commit 0a9d24d

Browse files
committed
tests: Fix request signer test
1 parent 8d268fc commit 0a9d24d

2 files changed

Lines changed: 8 additions & 6 deletions

File tree

tests/conftest.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from datetime import datetime
22

3-
from ecdsa.keys import SigningKey
3+
from ecdsa.keys import SigningKey, VerifyingKey
44
from httpx import AsyncClient
55
import pytest
66
import pytest_asyncio
@@ -45,6 +45,9 @@ def ecdsa_signing_key_str() -> str:
4545
def ecdsa_signing_key(ecdsa_signing_key_str: str) -> SigningKey:
4646
return SigningKey.from_pem(ecdsa_signing_key_str)
4747

48+
@pytest.fixture(scope="session")
49+
def ecdsa_verifying_key(ecdsa_signing_key: SigningKey) -> VerifyingKey:
50+
return ecdsa_signing_key.get_verifying_key()
4851

4952
@pytest.fixture(scope="session")
5053
def synthetic_request_signer(ecdsa_signing_key) -> RequestSigner:

tests/test_request_signer.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import base64
22
from binascii import unhexlify
3-
4-
from ecdsa.keys import VerifyingKey
3+
import pytest
4+
from ecdsa.keys import VerifyingKey, BadSignatureError
55

66
from xbox.webapi.common.request_signer import RequestSigner
77

@@ -85,15 +85,14 @@ def test_synthetic_verify_digest(
8585
"Fe3R7GuZfvGA350cH7hWgg4HIHjaD9lGYiwxki6bNyGnB8dMEIfEmBiuNuGUfWjY5lL2h44X/VMGOkPIezVb7Q=="
8686
)
8787
invalid_signature = b"\xFF" + bytes(signature)[1:]
88-
8988
success = synthetic_request_signer.verify_digest(signature, message)
90-
success_invalid = synthetic_request_signer.verify_digest(invalid_signature, message)
9189
success_via_vk = synthetic_request_signer.verify_digest(
9290
signature, message, ecdsa_verifying_key
9391
)
92+
with pytest.raises(BadSignatureError):
93+
synthetic_request_signer.verify_digest(invalid_signature, message)
9494

9595
assert success is True
96-
assert success_invalid is False
9796
assert success_via_vk is True
9897

9998

0 commit comments

Comments
 (0)