30 lines
787 B
Python
30 lines
787 B
Python
from slidge_whatsapp.group import replace_whatsapp_mentions
|
|
|
|
|
|
def test_replace_whatsapp_mentions():
|
|
text = "Hayo @1234, it's cool in here in with @5678!! @123333"
|
|
|
|
assert (
|
|
replace_whatsapp_mentions(
|
|
text,
|
|
{"+1234": "bibi", "+5678": "baba"},
|
|
)
|
|
== "Hayo bibi, it's cool in here in with baba!! @123333"
|
|
)
|
|
|
|
assert replace_whatsapp_mentions(text, {}) == text
|
|
|
|
assert replace_whatsapp_mentions(text, {"+123333": "prout"}) == text.replace(
|
|
"@123333", "prout"
|
|
)
|
|
|
|
assert (
|
|
replace_whatsapp_mentions("+1234", {"+1234": "bibi", "+5678": "baba"})
|
|
== "+1234"
|
|
)
|
|
|
|
assert (
|
|
replace_whatsapp_mentions("@1234@1234@123", {"+1234": "bibi", "+5678": "baba"})
|
|
== "bibibibi@123"
|
|
)
|