Mikaela/Limnoria
1
0
Fork 0
mirror of https://github.com/Mikaela/Limnoria.git synced 2025-01-23 10:34:19 +01:00
Code Issues Projects Releases Wiki Activity

Made PluginTestCase handle threaded plugins properly.

Browse Source
This commit is contained in:
Jeremy Fincher 2003-08-11 05:09:59 +00:00
parent 2c87c183b2
commit 1a05dc957b
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
test/test.py
View File

@ -38,6 +38,7 @@ from fix import *
import sys import sys
import glob import glob
import time
import os.path import os.path
import unittest import unittest
@ -81,6 +82,7 @@ class PluginTestCase(unittest.TestCase):
"""Subclass this to write a test case for a plugin. See test_FunCommands """Subclass this to write a test case for a plugin. See test_FunCommands
for an example. for an example.
""" """
timeout = 10
def setUp(self, nick='test'): def setUp(self, nick='test'):
self.nick = nick self.nick = nick
self.prefix = ircutils.joinHostmask(nick, 'user', 'host.domain.tld') self.prefix = ircutils.joinHostmask(nick, 'user', 'host.domain.tld')
@ -91,19 +93,24 @@ class PluginTestCase(unittest.TestCase):
def assertResponse(self, query, expectedResponse): def assertResponse(self, query, expectedResponse):
self.irc.feedMsg(ircmsgs.privmsg(self.nick, query, prefix=self.prefix)) self.irc.feedMsg(ircmsgs.privmsg(self.nick, query, prefix=self.prefix))
fed = time.time()
response = self.irc.takeMsg()
while response is None and time.time() - fed < self.timeout:
response = self.irc.takeMsg() response = self.irc.takeMsg()
self.failUnless(response) self.failUnless(response)
self.assertEqual(response.args[1], expectedResponse) self.assertEqual(response.args[1], expectedResponse)
def assertResponses(self, query, expectedResponses): def assertResponses(self, query, expectedResponses):
self.irc.feedMsg(ircmsgs.privmsg(self.nick, query, prefix=self.prefix)) self.irc.feedMsg(ircmsgs.privmsg(self.nick, query, prefix=self.prefix))
fed = time.time()
responses = [] responses = []
while 1: while len(responses) < len(expectedResponses) and \
time.time() - fed < self.timeout :
m = self.irc.takeMsg() m = self.irc.takeMsg()
if m: if m:
responses.append(m) responses.append(m)
else: else:
break time.sleep(.1)
self.assertEqual(len(expectedResponses), len(responses)) self.assertEqual(len(expectedResponses), len(responses))
for (response, expected) in zip(responses, expectedResponses): for (response, expected) in zip(responses, expectedResponses):
self.assertEqual(response.args[1], expected) self.assertEqual(response.args[1], expected)
@ -119,5 +126,5 @@ if __name__ == '__main__':
suite = unittest.defaultTestLoader.loadTestsFromNames(names) suite = unittest.defaultTestLoader.loadTestsFromNames(names)
runner = unittest.TextTestRunner(verbosity=2) runner = unittest.TextTestRunner(verbosity=2)
runner.run(suite) runner.run(suite)
print 'Total asserts: %s' % unittest.asserts ## print 'Total asserts: %s' % unittest.asserts
world.testing = False ## world.testing = False