File size: 2,910 Bytes
977bd69 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | from __future__ import annotations
import hashlib
import hmac
from app.wa import (
Deduper,
cards_to_text,
parse_incoming,
session_id_for,
to_whatsapp_text,
verify_signature,
)
def _sig(secret: str, body: bytes) -> str:
return "sha256=" + hmac.new(secret.encode(), body, hashlib.sha256).hexdigest()
def test_verify_signature_roundtrip():
body = b'{"hola": 1}'
assert verify_signature("s3cret", body, _sig("s3cret", body)) is True
assert verify_signature("s3cret", body, _sig("wrong", body)) is False
assert verify_signature("s3cret", body, None) is False
assert verify_signature("s3cret", body, "md5=abc") is False
def test_to_whatsapp_text_converts_markdown():
md = "**Manguera plana** de PVC\n\n- [Ver carrito](https://x.com/cart/1:1)\n## Detalles"
out = to_whatsapp_text(md)
assert "*Manguera plana*" in out
assert "**" not in out
assert "Ver carrito: https://x.com/cart/1:1" in out
assert "##" not in out
def test_cards_to_text_caps_and_links():
cards = [
{"title": f"Prod {i}", "price": "9,99 €", "add_url": f"https://s/cart/{i}:1"}
for i in range(5)
]
out = cards_to_text(cards)
assert out.count("•") == 3 # capped
assert "*Prod 0*" in out and "https://s/cart/0:1" in out
def test_session_id_deterministic_and_fits_column():
a = session_id_for(1, "34600111222")
assert a == session_id_for(1, "34600111222") # stable per phone
assert a != session_id_for(2, "34600111222") # tenant-scoped
assert a != session_id_for(1, "34600999888") # phone-scoped
assert len(a) <= 40 and a.startswith("wa")
def test_deduper_remembers_and_evicts():
d = Deduper(maxlen=2)
assert d.seen("a") is False
assert d.seen("a") is True
d.seen("b"), d.seen("c") # evicts "a"
assert d.seen("a") is False
def test_parse_incoming_extracts_text_and_skips_statuses():
payload = {
"entry": [
{
"changes": [
{
"value": {
"metadata": {"phone_number_id": "111"},
"messages": [
{"id": "wamid.1", "from": "34600111222", "type": "text",
"text": {"body": "hola"}},
{"id": "wamid.2", "from": "34600111222", "type": "image"},
],
}
},
{"value": {"metadata": {"phone_number_id": "111"},
"statuses": [{"id": "wamid.1", "status": "delivered"}]}},
]
}
]
}
msgs = parse_incoming(payload)
assert len(msgs) == 1
assert msgs[0].wamid == "wamid.1"
assert msgs[0].phone_number_id == "111"
assert msgs[0].from_phone == "34600111222"
assert msgs[0].text == "hola"
|