mirror of https://github.com/progval/irctest.git
135 lines
4.2 KiB
Python
135 lines
4.2 KiB
Python
"""
|
||
`Ergo <https://ergo.chat/>`_-specific tests of non-Unicode filtering
|
||
|
||
<https://ircv3.net/specs/extensions/utf8-only>`_
|
||
"""
|
||
|
||
from irctest import cases, runner
|
||
from irctest.numerics import ERR_ERRONEUSNICKNAME
|
||
from irctest.patma import ANYSTR
|
||
|
||
|
||
class Utf8TestCase(cases.BaseServerTestCase):
|
||
@cases.mark_specifications("Ergo")
|
||
def testNonUtf8Filtering(self):
|
||
self.connectClient(
|
||
"bar",
|
||
capabilities=["batch", "echo-message", "labeled-response"],
|
||
)
|
||
self.joinChannel(1, "#qux")
|
||
self.sendLine(1, b"@label=xyz PRIVMSG #qux hi\xaa")
|
||
self.assertMessageMatch(
|
||
self.getMessage(1),
|
||
command="FAIL",
|
||
params=["PRIVMSG", "INVALID_UTF8", ANYSTR],
|
||
tags={"label": "xyz"},
|
||
)
|
||
|
||
@cases.mark_isupport("UTF8ONLY")
|
||
def testUtf8Validation(self):
|
||
self.connectClient("foo")
|
||
self.connectClient("bar")
|
||
|
||
if "UTF8ONLY" not in self.server_support:
|
||
raise runner.IsupportTokenNotSupported("UTF8ONLY")
|
||
|
||
self.sendLine(1, "PRIVMSG bar hi")
|
||
self.getMessages(1) # synchronize
|
||
ms = self.getMessages(2)
|
||
self.assertMessageMatch(
|
||
[m for m in ms if m.command == "PRIVMSG"][0], params=["bar", "hi"]
|
||
)
|
||
|
||
self.sendLine(1, b"PRIVMSG bar hi\xaa")
|
||
|
||
m = self.getMessage(1)
|
||
assert m.command in ("FAIL", "WARN", "ERROR")
|
||
|
||
if m.command in ("FAIL", "WARN"):
|
||
self.assertMessageMatch(m, params=["PRIVMSG", "INVALID_UTF8", ANYSTR])
|
||
|
||
def testNonutf8Realname(self):
|
||
self.connectClient("foo")
|
||
if "UTF8ONLY" not in self.server_support:
|
||
raise runner.IsupportTokenNotSupported("UTF8ONLY")
|
||
|
||
self.addClient()
|
||
self.sendLine(2, "NICK bar")
|
||
self.clients[2].conn.sendall(b"USER username * * :i\xe8rc\xe9\r\n")
|
||
|
||
d = b""
|
||
while True:
|
||
try:
|
||
buf = self.clients[2].conn.recv(1024)
|
||
except TimeoutError:
|
||
break
|
||
if d and not buf:
|
||
break
|
||
d += buf
|
||
if b"FAIL " in d or b"ERROR " in d or b"468 " in d: # ERR_INVALIDUSERNAME
|
||
return # nothing more to test
|
||
self.assertIn(b"001 ", d)
|
||
|
||
self.sendLine(2, "WHOIS bar")
|
||
self.getMessages(2)
|
||
|
||
def testNonutf8Username(self):
|
||
self.connectClient("foo")
|
||
if "UTF8ONLY" not in self.server_support:
|
||
raise runner.IsupportTokenNotSupported("UTF8ONLY")
|
||
|
||
self.addClient()
|
||
self.sendLine(2, "NICK bar")
|
||
self.clients[2].conn.sendall(b"USER \xe8rc\xe9 * * :readlname\r\n")
|
||
|
||
d = b""
|
||
while True:
|
||
try:
|
||
buf = self.clients[2].conn.recv(1024)
|
||
except TimeoutError:
|
||
break
|
||
if d and not buf:
|
||
break
|
||
d += buf
|
||
if b"FAIL " in d or b"ERROR " in d or b"468 " in d: # ERR_INVALIDUSERNAME
|
||
return # nothing more to test
|
||
self.assertIn(b"001 ", d)
|
||
|
||
self.sendLine(2, "WHOIS bar")
|
||
self.getMessages(2)
|
||
|
||
|
||
class ErgoUtf8NickEnabledTestCase(cases.BaseServerTestCase):
|
||
@staticmethod
|
||
def config() -> cases.TestCaseControllerConfig:
|
||
return cases.TestCaseControllerConfig(
|
||
ergo_config=lambda config: config["server"].update(
|
||
{"casemapping": "precis"},
|
||
)
|
||
)
|
||
|
||
@cases.mark_specifications("Ergo")
|
||
def testUtf8NonAsciiNick(self):
|
||
"""Ergo accepts certain non-ASCII UTF8 nicknames if PRECIS is enabled."""
|
||
self.connectClient("Işıl")
|
||
self.joinChannel(1, "#test")
|
||
|
||
self.connectClient("Claire")
|
||
self.joinChannel(2, "#test")
|
||
|
||
self.sendLine(1, "PRIVMSG #test :hi there")
|
||
self.getMessages(1)
|
||
self.assertMessageMatch(
|
||
self.getMessage(2), nick="Işıl", params=["#test", "hi there"]
|
||
)
|
||
|
||
|
||
class ErgoUtf8NickDisabledTestCase(cases.BaseServerTestCase):
|
||
@cases.mark_specifications("Ergo")
|
||
def testUtf8NonAsciiNick(self):
|
||
"""Ergo rejects non-ASCII nicknames in its default configuration."""
|
||
self.addClient(1)
|
||
self.sendLine(1, "USER u s e r")
|
||
self.sendLine(1, "NICK Işıl")
|
||
self.assertMessageMatch(self.getMessage(1), command=ERR_ERRONEUSNICKNAME)
|