Andros Fenollosa
bfd8b510b6
Clean architecture (core use cases + infra gateways), X25519/AES-GCM E2E encryption, tkinter GUI, and a pytest suite. 45 tests passing.
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
"""Tests for composing outgoing encrypted frames."""
|
|
|
|
import base64
|
|
import json
|
|
|
|
from neotalk.core.entities.constants import FrameType, MessageKind
|
|
from neotalk.core.entities.responses import ResponseTypes
|
|
from neotalk.core.use_cases.conversation.compose_message import (
|
|
compose_message_use_case,
|
|
compose_typing_use_case,
|
|
)
|
|
|
|
|
|
def test_compose_message_seals_the_text(crypto, shared_key):
|
|
# When composing a message
|
|
response = compose_message_use_case(" hola bob ", crypto, shared_key)
|
|
|
|
# Then the frame is secure and its body decrypts to the trimmed text
|
|
assert response["type"] == ResponseTypes.SUCCESS
|
|
frame = response["data"]["frame"]
|
|
assert frame["type"] == FrameType.SECURE
|
|
ciphertext = base64.b64decode(frame["body"])
|
|
payload = json.loads(crypto.decrypt(shared_key, ciphertext))
|
|
assert payload == {"kind": MessageKind.MESSAGE, "text": "hola bob"}
|
|
|
|
|
|
def test_compose_message_rejects_empty_text(crypto, shared_key):
|
|
# When composing whitespace
|
|
response = compose_message_use_case(" ", crypto, shared_key)
|
|
|
|
# Then it is a parameters error
|
|
assert response["type"] == ResponseTypes.PARAMETERS_ERROR
|
|
|
|
|
|
def test_compose_message_rejects_overly_long_text(crypto, shared_key):
|
|
# When composing a huge message
|
|
response = compose_message_use_case("x" * 5000, crypto, shared_key)
|
|
|
|
# Then it is a parameters error
|
|
assert response["type"] == ResponseTypes.PARAMETERS_ERROR
|
|
|
|
|
|
def test_compose_typing_has_no_text(crypto, shared_key):
|
|
# When composing a typing signal
|
|
response = compose_typing_use_case(crypto, shared_key)
|
|
|
|
# Then the sealed payload is just the typing kind
|
|
frame = response["data"]["frame"]
|
|
ciphertext = base64.b64decode(frame["body"])
|
|
payload = json.loads(crypto.decrypt(shared_key, ciphertext))
|
|
assert payload == {"kind": MessageKind.TYPING}
|