Mikaela
/
Limnoria
mirror of https://github.com/Mikaela/Limnoria.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Include start and end messages in a batch. 

They may have useful server tags (especially when we start
supporting labeled-response).
This commit is contained in:
Valentin Lorentz 2020-05-06 18:29:17 +02:00
parent d627ba7683
commit cc0af4e790
2 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/irclib.py
View File

@ -845,9 +845,10 @@ class IrcState(IrcCommandDispatcher, log.Firewalled):
batch_type = msg.args[1]
batch_arguments = tuple(msg.args[2:])
self.batches[batch_name] = Batch(type=batch_type,
arguments=batch_arguments, messages=[])
arguments=batch_arguments, messages=[msg])
elif msg.args[0].startswith('-'):
batch = self.batches.pop(batch_name)
batch.messages.append(msg)
msg.tag('batch', batch)
else:
assert False, msg.args[0]

14
test/test_irclib.py
View File

@ -748,12 +748,13 @@ class IrcTestCase(SupyTestCase):
def testBatch(self):
self.irc.reset()
self.irc.feedMsg(ircmsgs.IrcMsg(':someuser1 JOIN #foo'))
self.irc.feedMsg(ircmsgs.IrcMsg(':host BATCH +name netjoin'))
m1 = ircmsgs.IrcMsg('@batch=name :someuser2 JOIN #foo')
m1 = ircmsgs.IrcMsg(':host BATCH +name netjoin')
self.irc.feedMsg(m1)
self.irc.feedMsg(ircmsgs.IrcMsg(':someuser3 JOIN #foo'))
m2 = ircmsgs.IrcMsg('@batch=name :someuser4 JOIN #foo')
m2 = ircmsgs.IrcMsg('@batch=name :someuser2 JOIN #foo')
self.irc.feedMsg(m2)
self.irc.feedMsg(ircmsgs.IrcMsg(':someuser3 JOIN #foo'))
m3 = ircmsgs.IrcMsg('@batch=name :someuser4 JOIN #foo')
self.irc.feedMsg(m3)
class Callback(irclib.IrcCallback):
batch = None
def name(self):
@ -763,10 +764,11 @@ class IrcTestCase(SupyTestCase):
c = Callback()
self.irc.addCallback(c)
try:
self.irc.feedMsg(ircmsgs.IrcMsg(':host BATCH -name'))
m4 = ircmsgs.IrcMsg(':host BATCH -name')
self.irc.feedMsg(m4)
finally:
self.irc.removeCallback(c.name())
self.assertEqual(c.batch, irclib.Batch('netjoin', (), [m1, m2]))
self.assertEqual(c.batch, irclib.Batch('netjoin', (), [m1, m2, m3, m4]))
class SaslTestCase(SupyTestCase):
def setUp(self):