val/irctest
1
0
Fork 0
mirror of https://github.com/progval/irctest.git synced 2025-04-05 23:09:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

expect batches

Browse Source
This commit is contained in:
Valentin Lorentz
2023-04-15 17:52:54 +02:00
parent e17ab347f4
commit 88f8be0ff6
1 changed files with 31 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
irctest/server_tests/metadata_2.py
View File

@ -5,13 +5,26 @@
""" """
from irctest import cases from irctest import cases
from irctest.patma import ANYSTR, StrRe from irctest.patma import ANYDICT, ANYSTR, StrRe
class MetadataTestCase(cases.BaseServerTestCase): class MetadataTestCase(cases.BaseServerTestCase):
valid_metadata_keys = {"valid_key1", "valid_key2"} valid_metadata_keys = {"valid_key1", "valid_key2"}
invalid_metadata_keys = {"invalid_key1", "invalid_key2"} invalid_metadata_keys = {"invalid_key1", "invalid_key2"}
def getBatchMessages(self, client):
messages = self.getMessages(1)
first_msg = messages.pop(0)
last_msg = messages.pop(-1)
self.assertMessageMatch(
first_msg, command="BATCH", params=[StrRe(r"\+.*"), "metadata"]
)
batch_id = first_msg.params[0][1:]
self.assertMessageMatch(last_msg, command="BATCH", params=["-" + batch_id])
return (batch_id, messages)
@cases.mark_specifications("IRCv3") @cases.mark_specifications("IRCv3")
def testGetOneUnsetValid(self): def testGetOneUnsetValid(self):
"""<http://ircv3.net/specs/core/metadata-3.2.html#metadata-get>""" """<http://ircv3.net/specs/core/metadata-3.2.html#metadata-get>"""
@ -19,12 +32,15 @@ class MetadataTestCase(cases.BaseServerTestCase):
"foo", capabilities=["draft/metadata-2", "batch"], skip_if_cap_nak=True "foo", capabilities=["draft/metadata-2", "batch"], skip_if_cap_nak=True
) )
self.sendLine(1, "METADATA * GET valid_key1") self.sendLine(1, "METADATA * GET valid_key1")
m = self.getMessage(1)
(batch_id, messages) = self.getBatchMessages(1)
self.assertEqual(len(messages), 1, fail_msg="Expected one ERR_NOMATCHINGKEY")
self.assertMessageMatch( self.assertMessageMatch(
m, messages[0],
tags={"batch": batch_id, **ANYDICT},
command="766", # ERR_NOMATCHINGKEY command="766", # ERR_NOMATCHINGKEY
fail_msg="Did not reply with 766 (ERR_NOMATCHINGKEY) to a " fail_msg="Did not reply with 766 (ERR_NOMATCHINGKEY) to a "
"request to an unset valid METADATA key.", "request to an unset valid METADATA key: {msg}",
) )
@cases.mark_specifications("IRCv3") @cases.mark_specifications("IRCv3")
@ -37,28 +53,28 @@ class MetadataTestCase(cases.BaseServerTestCase):
"foo", capabilities=["draft/metadata-2", "batch"], skip_if_cap_nak=True "foo", capabilities=["draft/metadata-2", "batch"], skip_if_cap_nak=True
) )
self.sendLine(1, "METADATA * GET valid_key1 valid_key2") self.sendLine(1, "METADATA * GET valid_key1 valid_key2")
m = self.getMessage(1) (batch_id, messages) = self.getBatchMessages(1)
self.assertEqual(len(messages), 2, fail_msg="Expected two ERR_NOMATCHINGKEY")
self.assertMessageMatch( self.assertMessageMatch(
m, messages[0],
command="766", # RPL_NOMATCHINGKEY command="766", # RPL_NOMATCHINGKEY
fail_msg="Did not reply with 766 (RPL_NOMATCHINGKEY) to a " fail_msg="Did not reply with 766 (RPL_NOMATCHINGKEY) to a "
"request to two unset valid METADATA key: {msg}", "request to two unset valid METADATA key: {msg}",
) )
self.assertMessageMatch( self.assertMessageMatch(
m, messages[0],
params=["foo", "foo", "valid_key1", ANYSTR], params=["foo", "foo", "valid_key1", ANYSTR],
fail_msg="Response to “METADATA * GET valid_key1 valid_key2” " fail_msg="Response to “METADATA * GET valid_key1 valid_key2” "
"did not respond to valid_key1 first: {msg}", "did not respond to valid_key1 first: {msg}",
) )
m = self.getMessage(1)
self.assertMessageMatch( self.assertMessageMatch(
m, messages[1],
command="766", # ERR_NOMATCHINGKEY command="766", # ERR_NOMATCHINGKEY
fail_msg="Did not reply with two 766 (ERR_NOMATCHINGKEY) to a " fail_msg="Did not reply with two 766 (ERR_NOMATCHINGKEY) to a "
"request to two unset valid METADATA key: {msg}", "request to two unset valid METADATA key: {msg}",
) )
self.assertMessageMatch( self.assertMessageMatch(
m, messages[1],
params=["foo", "foo", "valid_key2", ANYSTR], params=["foo", "foo", "valid_key2", ANYSTR],
fail_msg="Response to “METADATA * GET valid_key1 valid_key2” " fail_msg="Response to “METADATA * GET valid_key1 valid_key2” "
"did not respond to valid_key2 as second response: {msg}", "did not respond to valid_key2 as second response: {msg}",
@ -75,12 +91,8 @@ class MetadataTestCase(cases.BaseServerTestCase):
"foo", capabilities=["draft/metadata-2", "batch"], skip_if_cap_nak=True "foo", capabilities=["draft/metadata-2", "batch"], skip_if_cap_nak=True
) )
self.sendLine(1, "METADATA * LIST") self.sendLine(1, "METADATA * LIST")
m = self.getMessage(1) (batch_id, messages) = self.getBatchMessages(1)
self.assertMessageMatch( self.assertEqual(len(messages), 0, fail_msg="Expected empty batch")
m,
command="762", # RPL_METADATAEND
params=["foo", ANYSTR],
)
@cases.mark_specifications("IRCv3") @cases.mark_specifications("IRCv3")
def testListInvalidTarget(self): def testListInvalidTarget(self):
@ -125,8 +137,10 @@ class MetadataTestCase(cases.BaseServerTestCase):
if target == "*": if target == "*":
target = StrRe(r"(\*|foo)") target = StrRe(r"(\*|foo)")
(batch_id, messages) = self.getBatchMessages(1)
self.assertEqual(len(messages), 1, fail_msg="Expected one RPL_KEYVALUE")
self.assertMessageMatch( self.assertMessageMatch(
self.getMessage(1), messages[0],
command="761", # RPL_KEYVALUE command="761", # RPL_KEYVALUE
params=["foo", target, key, ANYSTR, value], params=["foo", target, key, ANYSTR, value],
) )