val
/
irctest
mirror of https://github.com/progval/irctest.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test +R user mode as implemented in Ergo (#168)

This commit is contained in:
Shivaram Lingamneni 2022-05-13 13:49:40 -04:00 committed by GitHub
parent 627f0b6415
commit 83017483ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 124 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

124
irctest/server_tests/umodes/registeredonly.py Normal file
View File

@ -0,0 +1,124 @@
"""
Test the registered-only DM user mode (commonly +R).
"""
from irctest import cases
from irctest.numerics import ERR_NEEDREGGEDNICK
@cases.mark_services
class RegisteredOnlyUmodeTestCase(cases.BaseServerTestCase):
@cases.mark_specifications("Ergo")
def testRegisteredOnlyUserMode(self):
"""Test the +R registered-only mode."""
self.controller.registerUser(self, "evan", "sesame")
self.controller.registerUser(self, "carmen", "pink")
self.connectClient(
"evan",
name="evan",
account="evan",
password="sesame",
capabilities=["sasl"],
)
self.connectClient("shivaram", name="shivaram")
self.sendLine("evan", "MODE evan +R")
self.assertMessageMatch(
self.getMessage("evan"), command="MODE", params=["evan", "+R"]
)
# this DM should be blocked by +R registered-only
self.getMessages("shivaram")
self.sendLine("shivaram", "PRIVMSG evan :hey there")
self.assertMessageMatch(
self.getMessage("shivaram"),
command=ERR_NEEDREGGEDNICK,
)
self.assertEqual(self.getMessages("evan"), [])
self.connectClient(
"carmen",
name="carmen",
account="carmen",
password="pink",
capabilities=["sasl"],
)
self.getMessages("evan")
self.sendLine("carmen", "PRIVMSG evan :hey there")
self.assertEqual(self.getMessages("carmen"), [])
# this message should go through fine:
self.assertMessageMatch(
self.getMessage("evan"),
command="PRIVMSG",
params=["evan", "hey there"],
)
@cases.mark_specifications("Ergo")
def testRegisteredOnlyUserModeAcceptCommand(self):
"""Test that the ACCEPT command can authorize another user
to send the accept-er direct messages, overriding the
+R registered-only mode."""
self.controller.registerUser(self, "evan", "sesame")
self.connectClient(
"evan",
name="evan",
account="evan",
password="sesame",
capabilities=["sasl"],
)
self.connectClient("shivaram", name="shivaram")
self.sendLine("evan", "MODE evan +R")
self.assertMessageMatch(
self.getMessage("evan"), command="MODE", params=["evan", "+R"]
)
self.sendLine("evan", "ACCEPT shivaram")
self.getMessages("evan")
self.sendLine("shivaram", "PRIVMSG evan :hey there")
self.assertEqual(self.getMessages("shivaram"), [])
self.assertMessageMatch(
self.getMessage("evan"),
command="PRIVMSG",
params=["evan", "hey there"],
)
self.sendLine("evan", "ACCEPT -shivaram")
self.getMessages("evan")
self.sendLine("shivaram", "PRIVMSG evan :how's it going")
self.assertMessageMatch(
self.getMessage("shivaram"),
command=ERR_NEEDREGGEDNICK,
)
self.assertEqual(self.getMessages("evan"), [])
@cases.mark_specifications("Ergo")
def testRegisteredOnlyUserModeAutoAcceptOnDM(self):
"""Test that sending someone a DM automatically authorizes them to
reply, overriding the +R registered-only mode."""
self.controller.registerUser(self, "evan", "sesame")
self.connectClient(
"evan",
name="evan",
account="evan",
password="sesame",
capabilities=["sasl"],
)
self.connectClient("shivaram", name="shivaram")
self.sendLine("evan", "MODE evan +R")
self.assertMessageMatch(
self.getMessage("evan"), command="MODE", params=["evan", "+R"]
)
self.sendLine("evan", "PRIVMSG shivaram :hey there")
self.getMessages("evan")
self.assertMessageMatch(
self.getMessage("shivaram"),
command="PRIVMSG",
params=["shivaram", "hey there"],
)
self.sendLine("shivaram", "PRIVMSG evan :how's it going")
self.assertEqual(self.getMessages("shivaram"), [])
self.assertMessageMatch(
self.getMessage("evan"),
command="PRIVMSG",
params=["evan", "how's it going"],
)