mirror of
https://github.com/progval/irctest.git
synced 2025-04-05 23:09:48 +00:00
remove draft/resume-0.5 tests
This commit is contained in:
Shivaram Lingamneni
committed by
Valentin Lorentz
Valentin Lorentz
parent
729b7cb8d8
commit
db0c64fa30
@ -1,260 +0,0 @@
|
||||
"""
|
||||
<https://github.com/DanielOaks/ircv3-specifications/blob/master+resume/extensions/resume.md>
|
||||
"""
|
||||
|
||||
import secrets
|
||||
|
||||
from irctest import cases
|
||||
from irctest.numerics import RPL_AWAY
|
||||
from irctest.patma import ANYDICT, ANYSTR
|
||||
|
||||
ANCIENT_TIMESTAMP = "2006-01-02T15:04:05.999Z"
|
||||
|
||||
|
||||
class ResumeTestCase(cases.BaseServerTestCase):
|
||||
@cases.mark_specifications("Oragono")
|
||||
def testNoResumeByDefault(self):
|
||||
self.connectClient(
|
||||
"bar", capabilities=["batch", "echo-message", "labeled-response"]
|
||||
)
|
||||
ms = self.getMessages(1)
|
||||
resume_messages = [m for m in ms if m.command == "RESUME"]
|
||||
self.assertEqual(
|
||||
resume_messages,
|
||||
[],
|
||||
"should not see RESUME messages unless explicitly negotiated",
|
||||
)
|
||||
|
||||
@cases.mark_specifications("Oragono")
|
||||
def testResume(self):
|
||||
chname = "#" + secrets.token_hex(12)
|
||||
self.connectClient(
|
||||
"observer", capabilities=["batch", "labeled-response", "server-time"]
|
||||
)
|
||||
ms = self.getMessages(1)
|
||||
|
||||
welcome = self.connectClient(
|
||||
"mainnick",
|
||||
capabilities=[
|
||||
"batch",
|
||||
"labeled-response",
|
||||
"server-time",
|
||||
"draft/resume-0.5",
|
||||
],
|
||||
)
|
||||
resume_messages = [m for m in welcome if m.command == "RESUME"]
|
||||
self.assertEqual(len(resume_messages), 1)
|
||||
self.assertEqual(resume_messages[0].params[0], "TOKEN")
|
||||
token = resume_messages[0].params[1]
|
||||
|
||||
self.joinChannel(1, chname)
|
||||
self.joinChannel(2, chname)
|
||||
self.sendLine(1, "PRIVMSG %s :hello friends" % (chname,))
|
||||
self.sendLine(1, "PRIVMSG mainnick :hello friend singular")
|
||||
self.getMessages(1)
|
||||
# should receive these messages
|
||||
privmsgs = [m for m in self.getMessages(2) if m.command == "PRIVMSG"]
|
||||
self.assertEqual(len(privmsgs), 2)
|
||||
privmsgs.sort(key=lambda m: m.params[0])
|
||||
self.assertMessageMatch(
|
||||
privmsgs[0], command="PRIVMSG", params=[chname, "hello friends"]
|
||||
)
|
||||
self.assertMessageMatch(
|
||||
privmsgs[1],
|
||||
command="PRIVMSG",
|
||||
params=["mainnick", "hello friend singular"],
|
||||
tags={"time": ANYSTR, **ANYDICT},
|
||||
)
|
||||
channelMsgTime = privmsgs[0].tags.get("time")
|
||||
|
||||
# tokens MUST be cryptographically secure; therefore, this token should be
|
||||
# invalid with probability at least 1 - 1/(2**128)
|
||||
bad_token = "a" * len(token)
|
||||
self.addClient()
|
||||
self.sendLine(3, "CAP LS")
|
||||
self.sendLine(3, "CAP REQ :batch labeled-response server-time draft/resume-0.5")
|
||||
self.sendLine(3, "NICK tempnick")
|
||||
self.sendLine(3, "USER tempuser 0 * tempuser")
|
||||
self.sendLine(3, " ".join(("RESUME", bad_token, ANCIENT_TIMESTAMP)))
|
||||
|
||||
# resume with a bad token MUST fail
|
||||
ms = self.getMessages(3)
|
||||
resume_err_messages = [
|
||||
m
|
||||
for m in ms
|
||||
if m.command == "FAIL" and m.params[:2] == ["RESUME", "INVALID_TOKEN"]
|
||||
]
|
||||
self.assertEqual(len(resume_err_messages), 1)
|
||||
# however, registration should proceed with the alternative nick
|
||||
self.sendLine(3, "CAP END")
|
||||
welcome_msgs = [
|
||||
m for m in self.getMessages(3) if m.command == "001"
|
||||
] # RPL_WELCOME
|
||||
self.assertEqual(welcome_msgs[0].params[0], "tempnick")
|
||||
|
||||
self.addClient()
|
||||
self.sendLine(4, "CAP LS")
|
||||
self.sendLine(4, "CAP REQ :batch labeled-response server-time draft/resume-0.5")
|
||||
self.sendLine(4, "NICK tempnick_")
|
||||
self.sendLine(4, "USER tempuser 0 * tempuser")
|
||||
# resume with a timestamp in the distant past
|
||||
self.sendLine(4, " ".join(("RESUME", token, ANCIENT_TIMESTAMP)))
|
||||
# successful resume does not require CAP END:
|
||||
# https://github.com/ircv3/ircv3-specifications/pull/306/files#r255318883
|
||||
ms = self.getMessages(4)
|
||||
|
||||
# now, do a valid resume with the correct token
|
||||
resume_messages = [m for m in ms if m.command == "RESUME"]
|
||||
self.assertEqual(len(resume_messages), 2)
|
||||
self.assertEqual(resume_messages[0].params[0], "TOKEN")
|
||||
new_token = resume_messages[0].params[1]
|
||||
self.assertNotEqual(
|
||||
token,
|
||||
new_token,
|
||||
"should receive a new, strong resume token; instead got " + new_token,
|
||||
)
|
||||
# success message
|
||||
self.assertMessageMatch(
|
||||
resume_messages[1], command="RESUME", params=["SUCCESS", "mainnick"]
|
||||
)
|
||||
|
||||
# test replay of messages
|
||||
privmsgs = [
|
||||
m for m in ms if m.command == "PRIVMSG" and m.prefix.startswith("observer")
|
||||
]
|
||||
self.assertEqual(len(privmsgs), 2)
|
||||
privmsgs.sort(key=lambda m: m.params[0])
|
||||
self.assertMessageMatch(
|
||||
privmsgs[0],
|
||||
command="PRIVMSG",
|
||||
params=[chname, "hello friends"],
|
||||
tags={"time": channelMsgTime, **ANYDICT},
|
||||
)
|
||||
# should replay with the original server-time
|
||||
# TODO this probably isn't testing anything because the timestamp only
|
||||
# has second resolution, hence will typically match by accident
|
||||
self.assertMessageMatch(
|
||||
privmsgs[1], command="PRIVMSG", params=["mainnick", "hello friend singular"]
|
||||
)
|
||||
|
||||
# legacy client should receive a QUIT and a JOIN
|
||||
quit, join = [m for m in self.getMessages(1) if m.command in ("QUIT", "JOIN")]
|
||||
self.assertEqual(quit.command, "QUIT")
|
||||
self.assertTrue(quit.prefix.startswith("mainnick"))
|
||||
self.assertMessageMatch(join, command="JOIN", params=[chname])
|
||||
self.assertTrue(join.prefix.startswith("mainnick"))
|
||||
|
||||
# original client should have been disconnected
|
||||
self.assertDisconnected(2)
|
||||
# new client should be receiving PRIVMSG sent to mainnick
|
||||
self.sendLine(1, "PRIVMSG mainnick :hello again")
|
||||
self.getMessages(1)
|
||||
self.assertMessageMatch(
|
||||
self.getMessage(4), command="PRIVMSG", params=["mainnick", "hello again"]
|
||||
)
|
||||
|
||||
# test chain-resuming (resuming the resumed connection, using the new token)
|
||||
self.addClient()
|
||||
self.sendLine(5, "CAP LS")
|
||||
self.sendLine(5, "CAP REQ :batch labeled-response server-time draft/resume-0.5")
|
||||
self.sendLine(5, "NICK tempnick_")
|
||||
self.sendLine(5, "USER tempuser 0 * tempuser")
|
||||
self.sendLine(5, "RESUME " + new_token)
|
||||
ms = self.getMessages(5)
|
||||
|
||||
resume_messages = [m for m in ms if m.command == "RESUME"]
|
||||
self.assertEqual(len(resume_messages), 2)
|
||||
self.assertEqual(resume_messages[0].params[0], "TOKEN")
|
||||
new_new_token = resume_messages[0].params[1]
|
||||
self.assertNotEqual(
|
||||
token,
|
||||
new_new_token,
|
||||
"should receive a new, strong resume token; instead got " + new_new_token,
|
||||
)
|
||||
self.assertNotEqual(
|
||||
new_token,
|
||||
new_new_token,
|
||||
"should receive a new, strong resume token; instead got " + new_new_token,
|
||||
)
|
||||
# success message
|
||||
self.assertMessageMatch(
|
||||
resume_messages[1], command="RESUME", params=["SUCCESS", "mainnick"]
|
||||
)
|
||||
|
||||
@cases.mark_specifications("Oragono")
|
||||
def testBRB(self):
|
||||
chname = "#" + secrets.token_hex(12)
|
||||
self.connectClient(
|
||||
"observer",
|
||||
capabilities=[
|
||||
"batch",
|
||||
"labeled-response",
|
||||
"message-tags",
|
||||
"server-time",
|
||||
"draft/resume-0.5",
|
||||
],
|
||||
)
|
||||
ms = self.getMessages(1)
|
||||
self.joinChannel(1, chname)
|
||||
|
||||
welcome = self.connectClient(
|
||||
"mainnick",
|
||||
capabilities=[
|
||||
"batch",
|
||||
"labeled-response",
|
||||
"server-time",
|
||||
"draft/resume-0.5",
|
||||
],
|
||||
)
|
||||
resume_messages = [m for m in welcome if m.command == "RESUME"]
|
||||
self.assertEqual(len(resume_messages), 1)
|
||||
self.assertEqual(resume_messages[0].params[0], "TOKEN")
|
||||
token = resume_messages[0].params[1]
|
||||
self.joinChannel(2, chname)
|
||||
|
||||
self.getMessages(1)
|
||||
self.sendLine(2, "BRB :software upgrade")
|
||||
# should receive, e.g., `BRB 210` (number of seconds)
|
||||
ms = [m for m in self.getMessages(2) if m.command == "BRB"]
|
||||
self.assertEqual(len(ms), 1)
|
||||
self.assertGreater(int(ms[0].params[0]), 1)
|
||||
# BRB disconnects you
|
||||
self.assertDisconnected(2)
|
||||
# without sending a QUIT line to friends
|
||||
self.assertEqual(self.getMessages(1), [])
|
||||
|
||||
self.sendLine(1, "PRIVMSG mainnick :hey there")
|
||||
# BRB message should be sent as an away message
|
||||
self.assertMessageMatch(
|
||||
self.getMessage(1),
|
||||
command=RPL_AWAY,
|
||||
params=["observer", "mainnick", "software upgrade"],
|
||||
)
|
||||
|
||||
self.addClient(3)
|
||||
self.sendLine(3, "CAP REQ :batch account-tag message-tags draft/resume-0.5")
|
||||
self.sendLine(3, " ".join(("RESUME", token, ANCIENT_TIMESTAMP)))
|
||||
ms = self.getMessages(3)
|
||||
|
||||
resume_messages = [m for m in ms if m.command == "RESUME"]
|
||||
self.assertEqual(len(resume_messages), 2)
|
||||
self.assertEqual(resume_messages[0].params[0], "TOKEN")
|
||||
self.assertMessageMatch(
|
||||
resume_messages[1], command="RESUME", params=["SUCCESS", "mainnick"]
|
||||
)
|
||||
|
||||
privmsgs = [
|
||||
m for m in ms if m.command == "PRIVMSG" and m.prefix.startswith("observer")
|
||||
]
|
||||
self.assertEqual(len(privmsgs), 1, privmsgs)
|
||||
self.assertMessageMatch(
|
||||
privmsgs[0],
|
||||
nick="observer",
|
||||
command="PRIVMSG",
|
||||
params=["mainnick", "hey there"],
|
||||
)
|
||||
|
||||
# friend with the resume cap should receive a RESUMED message
|
||||
resumed_messages = [m for m in self.getMessages(1) if m.command == "RESUMED"]
|
||||
self.assertEqual(len(resumed_messages), 1)
|
||||
self.assertTrue(resumed_messages[0].prefix.startswith("mainnick"))
|
||||
Reference in New Issue
Block a user