Fix unparsing malformed entities, bump v1.10.10

This commit is contained in:
Lonami Exo
2019-12-30 10:57:03 +01:00
parent be8838b5f8
commit d196c89825
6 changed files with 94 additions and 11 deletions

View File

@@ -1,13 +1,14 @@
"""
tests for telethon.crypto.rsa
Tests for `telethon.crypto.rsa`.
"""
import pytest
from telethon.crypto import rsa
@pytest.fixture
def server_key_fp():
"""factory to return a key, old if so chosen"""
"""Factory to return a key, old if so chosen."""
def _server_key_fp(old: bool):
for fp, data in rsa._server_keys.items():
_, old_key = data
@@ -16,22 +17,26 @@ def server_key_fp():
return _server_key_fp
def test_encryption_inv_key():
"""test for #1324"""
"""Test for #1324."""
assert rsa.encrypt("invalid", b"testdata") is None
def test_encryption_old_key(server_key_fp):
"""test for #1324"""
"""Test for #1324."""
assert rsa.encrypt(server_key_fp(old=True), b"testdata") is None
def test_encryption_allowed_old_key(server_key_fp):
data = rsa.encrypt(server_key_fp(old=True), b"testdata", use_old=True)
# we can't verify the data is actually valid because we don't have
# We can't verify the data is actually valid because we don't have
# the decryption keys
assert data is not None and len(data) == 256
def test_encryption_current_key(server_key_fp):
data = rsa.encrypt(server_key_fp(old=False), b"testdata")
# we can't verify the data is actually valid because we don't have
# We can't verify the data is actually valid because we don't have
# the decryption keys
assert data is not None and len(data) == 256