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

Reworked so there's no double-import problem.

This commit is contained in:
Jeremy Fincher 2003-12-02 12:27:45 +00:00
parent da92309ba4
commit 2a9d9d1b5d
65 changed files with 412 additions and 372 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

315
test/test.py
View File

@ -31,78 +31,21 @@
import supybot
import logging
import conf
conf.dataDir = 'test-data'
conf.confDir = 'test-conf'
conf.logDir = 'test-log'
conf.replyWhenNotCommand = False
conf.minimumLogPriority = logging.CRITICAL
conf.stdoutLogging = False
import fix
import gc
import re
import sys
import imp
import glob
import time
started = time.time()
import os.path
import unittest
import world
world.startedAt = started
import ircdb
import irclib
import drivers
import ircmsgs
import ircutils
import callbacks
import Owner
import Misc
originalCallbacksGetHelp = callbacks.getHelp
lastGetHelp = 'x'*1000
def cachingGetHelp(method, name=None):
global lastGetHelp
lastGetHelp = originalCallbacksGetHelp(method, name)
return lastGetHelp
callbacks.getHelp = cachingGetHelp
nicks = ['fatjim','scn','moshez','LordVan','MetaCosm','pythong','fishfart',
'alb','d0rt','jemfinch','StyxAlso','fors','deltab','gd',
'hellz_hunter','are_j|pub_comp','jason_','dreid','sayke_','winjer',
'TenOfTen','GoNoVas','queuetue','the|zzz','Hellfried','Therion',
'shro','DaCa','rexec','polin8','r0ky','aaron_','ironfroggy','eugene',
'faassen','tirloni','mackstann','Yhg1s','ElBarono','vegai','shang',
'typo_','kikoforgetme','asqui','TazyTiggy','fab','nixman','liiwi',
'AdamV','paolo','red_one','_AleX_','lament','jamessan','supybot',
'macr0_zzz','plaisthos','redghost','disco','mphardy','gt3','mathie',
'jonez','r0ky-office','tic','d33p','ES3merge','talin','af','flippo',
'sholden','ameoba','shepherg','j2','Acapnotic','dash','merlin262',
'Taaus','_moshez','rik','jafo__','blk-majik','JT__','itamar',
'kermit-','davidmccabe','glyph','jojo','dave_p','goo','hyjinx',
'SamB','exarkun','drewp','Ragica','skylan','redgore','k3','Ra1stlin',
'StevenK','carball','h3x','carljm','_jacob','teratorn','frangen',
'phed','datazone','Yaggo','acct_','nowhere','pyn','ThomasWaldmann',
'dunker','pilotLight','brainless','LoganH_','jmpnz','steinn',
'EliasREC','lowks__','OldSmrf','Mad77','snibril','delta','psy',
'skimpIzu','Kengur','MoonFallen','kotkis','Hyperi']
fd = file(os.path.join('test', 'rfc2812.msgs'), 'r')
rawmsgs = [line.strip() for line in fd]
fd.close()
msgs = []
for s in rawmsgs:
try:
msgs.append(ircmsgs.IrcMsg(s))
except:
print 'IrcMsg constructor failed: %r' % s
nicks += [msg.nick for msg in msgs if msg.nick]
class path(str):
"""A class to represent platform-independent paths."""
@ -112,251 +55,6 @@ class path(str):
def __eq__(self, other):
return self._r.split(self) == self._r.split(other)
class TimeoutError(AssertionError):
def __str__(self):
return '%r timed out' % self.args[0]
class PluginTestCase(unittest.TestCase):
"""Subclass this to write a test case for a plugin. See test/test_Fun.py
for an example.
"""
timeout = 10
plugins = None
cleanConfDir = True
cleanDataDir = True
def setUp(self, nick='test'):
# Set conf variables appropriately.
conf.prefixChars = '@'
conf.replyWhenNotCommand = False
self.myVerbose = world.myVerbose
if self.cleanConfDir:
for filename in os.listdir(conf.confDir):
os.remove(os.path.join(conf.confDir, filename))
if self.cleanDataDir:
for filename in os.listdir(conf.dataDir):
os.remove(os.path.join(conf.dataDir, filename))
ircdb.users.reload()
ircdb.channels.reload()
if self.plugins is None:
raise ValueError, 'PluginTestCase must have a "plugins" attribute.'
self.nick = nick
self.prefix = ircutils.joinHostmask(nick, 'user', 'host.domain.tld')
self.irc = irclib.Irc(nick)
while self.irc.takeMsg():
pass
_ = Owner.loadPluginClass(self.irc, Owner)
_ = Owner.loadPluginClass(self.irc, Misc)
if isinstance(self.plugins, str):
if self.plugins not in ('Owner', 'Misc'):
module = Owner.loadPluginModule(self.plugins)
cb = Owner.loadPluginClass(self.irc, module)
else:
for name in self.plugins:
if name not in ('Owner', 'Misc'):
module = Owner.loadPluginModule(name)
cb = Owner.loadPluginClass(self.irc, module)
def tearDown(self):
self.irc.die()
gc.collect()
def _feedMsg(self, query, timeout=None):
if timeout is None:
timeout = self.timeout
if self.myVerbose:
print # Extra newline, so it's pretty.
msg = ircmsgs.privmsg(self.irc.nick, query, prefix=self.prefix)
if self.myVerbose:
print 'Feeding: %r' % msg
self.irc.feedMsg(msg)
fed = time.time()
response = self.irc.takeMsg()
while response is None and time.time() - fed < timeout:
time.sleep(0.1) # So it doesn't suck up 100% cpu.
drivers.run()
response = self.irc.takeMsg()
if self.myVerbose:
print 'Response: %r' % response
return response
def getMsg(self, query, timeout=None):
return self._feedMsg(query, timeout=timeout)
def feedMsg(self, query):
"""Just feeds it a message, that's all."""
self.irc.feedMsg(ircmsgs.privmsg(self.irc.nick, query,
prefix=self.prefix))
# These assertError/assertNoError are somewhat fragile. The proper way to
# do them would be to use a proxy for the irc object and intercept .error.
# But that would be hard, so I don't bother. When this breaks, it'll get
# fixed, but not until then.
def assertError(self, query):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
if lastGetHelp not in m.args[1]:
self.failUnless(m.args[1].startswith('Error:'),
'%r did not error: %s' % (query, m.args[1]))
def assertNotError(self, query):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failIf(m.args[1].startswith('Error:'),
'%r errored: %s' % (query, m.args[1]))
self.failIf(lastGetHelp in m.args[1],
'%r returned the help string.' % query)
def assertHelp(self, query):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failUnless(lastGetHelp in m.args[1])
def assertNoResponse(self, query, timeout=None):
m = self._feedMsg(query, timeout)
self.failIf(m, 'Unexpected response: %r' % m)
def assertResponse(self, query, expectedResponse):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.assertEqual(m.args[1], expectedResponse,
'%r != %r' % (expectedResponse, m.args[1]))
def assertRegexp(self, query, regexp, flags=re.I):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failUnless(re.search(regexp, m.args[1], flags),
'%r does not match %r' % (m.args[1], regexp))
def assertNotRegexp(self, query, regexp, flags=re.I):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failUnless(re.search(regexp, m.args[1], flags) is None,
'%r matched %r' % (m.args[1], regexp))
def assertRegexps(self, query, regexps):
started = time.time()
total = len(regexps)*self.timeout
while regexps and time.time() - started < total:
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
regexp = regexps.pop(0)
self.failUnless(re.search(regexp, m.args[1]),
'%r does not match %r' % (m.args[1], regexp))
self.failIf(time.time() - started > total)
def assertResponses(self, query, expectedResponses):
responses = []
started = time.time()
while len(responses) < len(expectedResponses) and \
time.time() - started > len(expectedResponses)*self.timeout:
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
responses.append(m)
self.assertEqual(len(expectedResponses), len(responses))
for (m, expected) in zip(responses, expectedResponses):
self.assertEqual(m.args[1], expected)
def assertAction(self, query, expectedResponse=None):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failUnless(ircmsgs.isAction(m))
if expectedResponse is not None:
self.assertEqual(ircmsgs.unAction(m), expectedResponse)
def assertActionRegexp(self, query, regexp, flags=re.I):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failUnless(ircmsgs.isAction(m))
s = ircmsgs.unAction(m)
self.failUnless(re.search(regexp, s, flags),
'%r does not match %r' % (s, regexp))
class ChannelPluginTestCase(PluginTestCase):
channel = '#test'
def setUp(self):
PluginTestCase.setUp(self)
self.irc.feedMsg(ircmsgs.join(self.channel, prefix=self.prefix))
def _feedMsg(self, query, timeout=None):
if timeout is None:
timeout = self.timeout
if self.myVerbose:
print # Newline, just like PluginTestCase.
if query[0] not in conf.prefixChars:
query = conf.prefixChars[0] + query
msg = ircmsgs.privmsg(self.channel, query, prefix=self.prefix)
if self.myVerbose:
print 'Feeding: %r' % msg
self.irc.feedMsg(msg)
fed = time.time()
response = self.irc.takeMsg()
while response is None and time.time() - fed < timeout:
drivers.run()
response = self.irc.takeMsg()
if response is not None:
if response.command == 'PRIVMSG':
args = list(response.args)
# Strip off nick: at beginning of response.
if args[1].startswith(self.nick) or \
args[1].startswith(ircutils.nickFromHostmask(self.prefix)):
args[1] = args[1].split(' ', 1)[1]
ret = ircmsgs.privmsg(*args)
else:
ret = response
else:
ret = None
if self.myVerbose:
print 'Returning: %r' % ret
return ret
def feedMsg(self, query):
"""Just feeds it a message, that's all."""
self.irc.feedMsg(ircmsgs.privmsg(self.channel, query,
prefix=self.prefix))
class PluginDocumentation:
def testAllCommandsHaveHelp(self):
for cb in self.irc.callbacks:
if isinstance(cb, callbacks.PrivmsgRegexp):
continue
if hasattr(cb, 'isCommand'):
for attr in cb.__class__.__dict__:
if cb.isCommand(attr):
self.failUnless(getattr(cb, attr).__doc__,
'%s has no syntax' % attr)
def testAllCommandsHaveMorehelp(self):
for cb in self.irc.callbacks:
if isinstance(cb, callbacks.PrivmsgRegexp):
continue
if hasattr(cb, 'isCommand'):
for attr in cb.__class__.__dict__:
if cb.isCommand(attr):
command = getattr(cb, attr)
helps = command.__doc__
self.failUnless(helps and len(helps.splitlines()) >= 3,
'%s has no help' % attr)
def testPluginHasDocumentation(self):
for cb in self.irc.callbacks:
m = sys.modules[cb.__class__.__module__]
self.failIf(m.__doc__ is None,
'%s has no module documentation'%cb.__class__.__name__)
if __name__ == '__main__':
import optparse
@ -369,11 +67,11 @@ if __name__ == '__main__':
if not os.path.exists(conf.logDir):
os.mkdir(conf.logDir)
for filename in os.listdir(conf.logDir):
if filename == 'plugins':
continue
filename = os.path.join(conf.logDir, filename)
os.remove(filename)
## for filename in os.listdir(conf.logDir):
## if filename == 'plugins':
## continue
## filename = os.path.join(conf.logDir, filename)
## os.remove(filename)
pluginLogDir = os.path.join(conf.logDir, 'plugins')
for filename in os.listdir(pluginLogDir):
os.remove(os.path.join(pluginLogDir, filename))
@ -421,6 +119,5 @@ if __name__ == '__main__':
runner = unittest.TextTestRunner(verbosity=2)
runner.run(suite)
print 'Total asserts: %s' % unittest.asserts
world.testing = False
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78:

2
test/test_Admin.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import conf

2
test/test_Alias.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
Alias = Owner.loadPluginModule('Alias')

2
test/test_Amazon.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class AmazonTestCase(PluginTestCase, PluginDocumentation):
plugins = ('Amazon',)

2
test/test_Babelfish.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class BabelFishTestCase(PluginTestCase, PluginDocumentation):
plugins = ('Babelfish',)

2
test/test_BadWords.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class BadWordsTestCase(PluginTestCase, PluginDocumentation):
plugins = ('BadWords', 'Utilities')

2
test/test_Bugzilla.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class BugzillaTest(PluginTestCase, PluginDocumentation):
plugins = ('Bugzilla',)

2
test/test_Channel.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import conf
import ircdb

2
test/test_ChannelDB.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import ircdb

2
test/test_Ctcp.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class CtcpTestCase(PluginTestCase):
plugins = ('Ctcp',)

2
test/test_DCC.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import time
import telnetlib

2
test/test_Debian.py
View File

@ -32,7 +32,7 @@
import os
import time
from test import *
from testsupport import *
class DebianTestCase(PluginTestCase, PluginDocumentation):
plugins = ('Debian',)

2
test/test_Dict.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class DictTestCase(PluginTestCase, PluginDocumentation):
plugins = ('Dict', 'Misc')

2
test/test_Dunno.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
try:
import sqlite

2
test/test_Ebay.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class EbayTest(PluginTestCase, PluginDocumentation):
plugins = ('Ebay',)

2
test/test_Enforcer.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class EnforcerTestCase(PluginTestCase, PluginDocumentation):
plugins = ('Enforcer',)

2
test/test_Factoids.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
try:
import sqlite

2
test/test_FixRelayBot.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import ircmsgs

2
test/test_Friendly.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class FriendlyTestCase(PluginTestCase):
plugins = ('Friendly',)

2
test/test_Fun.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import re

2
test/test_FunDB.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import ircdb

2
test/test_Gameknot.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import utils

2
test/test_Google.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class GoogleTestCase(ChannelPluginTestCase, PluginDocumentation):
plugins = ('Google',)

2
test/test_Http.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class HttpTest(PluginTestCase, PluginDocumentation):
plugins = ('Http',)

2
test/test_Karma.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import conf

2
test/test_Lookup.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import os

2
test/test_Markov.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
try:
import sqlite

2
test/test_Math.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class MathTestCase(PluginTestCase, PluginDocumentation):
plugins = ('Math',)

2
test/test_Misc.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class MiscTestCase(ChannelPluginTestCase, PluginDocumentation):
plugins = ('Misc', 'Utilities', 'Gameknot', 'Ctcp')

2
test/test_MoobotFactoids.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
try:
import sqlite

2
test/test_Movies.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class MoviesTestCase(PluginTestCase, PluginDocumentation):
plugins = ('Movies',)

2
test/test_Network.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class NetworkTestCase(PluginTestCase, PluginDocumentation):
plugins = ['Network']

2
test/test_News.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import time

2
test/test_Note.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import utils
import ircdb

2
test/test_OSU.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class OSUTestCase(PluginTestCase, PluginDocumentation):
plugins = ('OSU',)

2
test/test_Owner.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import conf
import Owner

2
test/test_Parter.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import ircmsgs

2
test/test_Poll.py
View File

@ -30,7 +30,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import utils
import ircdb

2
test/test_Python.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import os

2
test/test_QuoteGrabs.py
View File

@ -31,7 +31,7 @@
import sets
from test import *
from testsupport import *
try:
import sqlite

2
test/test_Quotes.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
try:
import sqlite

2
test/test_RSS.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class RSSTestCase(PluginTestCase, PluginDocumentation):
plugins = ('RSS',)

2
test/test_Relay.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class RelayTestCase(PluginTestCase, PluginDocumentation):
plugins = ('Relay',)

2
test/test_RootWarner.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class RootWarnerTestCase(PluginTestCase):
plugins = ('RootWarner',)

2
test/test_Services.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class ServicesTestCase(PluginTestCase, PluginDocumentation):
plugins = ('Services',)

2
test/test_Sourceforge.py
View File

@ -31,7 +31,7 @@
import re
from test import *
from testsupport import *
class SourceforgeTest(ChannelPluginTestCase, PluginDocumentation):
plugins = ('Sourceforge',)

2
test/test_Status.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import sys

2
test/test_Todo.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
try:
import sqlite

2
test/test_Topic.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class TopicTestCase(ChannelPluginTestCase, PluginDocumentation):
plugins = ('Topic',)

2
test/test_URL.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
urls = """
http://www.ureg.ohio-state.edu/courses/book3.asp

2
test/test_Unix.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import os

2
test/test_User.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import ircdb

2
test/test_Utilities.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
class UtilitiesTestCase(PluginTestCase, PluginDocumentation):
plugins = ('Utilities', 'Status')

2
test/test_callbacks.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import conf
import utils

2
test/test_fix.py
View File

@ -31,7 +31,7 @@
## from __future__ import generators
from test import *
from testsupport import *
import itertools

2
test/test_ircdb.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import os
import unittest

2
test/test_irclib.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import copy
import pickle

2
test/test_ircmsgs.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import copy
import pickle

2
test/test_ircutils.py
View File

@ -30,7 +30,7 @@
###
from test import *
from testsupport import *
import copy
import random

2
test/test_plugins.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import sets

2
test/test_privmsgs.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import ircmsgs
import privmsgs

2
test/test_schedule.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import time

2
test/test_structures.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import pickle

2
test/test_utils.py
View File

@ -29,7 +29,7 @@
# POSSIBILITY OF SUCH DAMAGE.
###
from test import *
from testsupport import *
import sets
import utils

343
test/testsupport.py Normal file
View File

@ -0,0 +1,343 @@
#!/usr/bin/env python
###
# Copyright (c) 2002, Jeremiah Fincher
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions, and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions, and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of the author of this software nor the name of
# contributors to this software may be used to endorse or promote products
# derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
###
import fix
import gc
import os
import re
import sys
import time
started = time.time()
import unittest
import conf
import ircdb
import world
world.startedAt = started
import irclib
import drivers
import ircmsgs
import ircutils
import callbacks
import Owner
import Misc
nicks = ['fatjim','scn','moshez','LordVan','MetaCosm','pythong','fishfart',
'alb','d0rt','jemfinch','StyxAlso','fors','deltab','gd',
'hellz_hunter','are_j|pub_comp','jason_','dreid','sayke_','winjer',
'TenOfTen','GoNoVas','queuetue','the|zzz','Hellfried','Therion',
'shro','DaCa','rexec','polin8','r0ky','aaron_','ironfroggy','eugene',
'faassen','tirloni','mackstann','Yhg1s','ElBarono','vegai','shang',
'typo_','kikoforgetme','asqui','TazyTiggy','fab','nixman','liiwi',
'AdamV','paolo','red_one','_AleX_','lament','jamessan','supybot',
'macr0_zzz','plaisthos','redghost','disco','mphardy','gt3','mathie',
'jonez','r0ky-office','tic','d33p','ES3merge','talin','af','flippo',
'sholden','ameoba','shepherg','j2','Acapnotic','dash','merlin262',
'Taaus','_moshez','rik','jafo__','blk-majik','JT__','itamar',
'kermit-','davidmccabe','glyph','jojo','dave_p','goo','hyjinx',
'SamB','exarkun','drewp','Ragica','skylan','redgore','k3','Ra1stlin',
'StevenK','carball','h3x','carljm','_jacob','teratorn','frangen',
'phed','datazone','Yaggo','acct_','nowhere','pyn','ThomasWaldmann',
'dunker','pilotLight','brainless','LoganH_','jmpnz','steinn',
'EliasREC','lowks__','OldSmrf','Mad77','snibril','delta','psy',
'skimpIzu','Kengur','MoonFallen','kotkis','Hyperi']
fd = file(os.path.join('test', 'rfc2812.msgs'), 'r')
rawmsgs = [line.strip() for line in fd]
fd.close()
msgs = []
for s in rawmsgs:
try:
msgs.append(ircmsgs.IrcMsg(s))
except:
print 'IrcMsg constructor failed: %r' % s
nicks += [msg.nick for msg in msgs if msg.nick]
originalCallbacksGetHelp = callbacks.getHelp
lastGetHelp = 'x'*1000
def cachingGetHelp(method, name=None):
global lastGetHelp
lastGetHelp = originalCallbacksGetHelp(method, name)
return lastGetHelp
callbacks.getHelp = cachingGetHelp
class TimeoutError(AssertionError):
def __str__(self):
return '%r timed out' % self.args[0]
class PluginTestCase(unittest.TestCase):
"""Subclass this to write a test case for a plugin. See test/test_Fun.py
for an example.
"""
timeout = 10
plugins = None
cleanConfDir = True
cleanDataDir = True
def setUp(self, nick='test'):
# Set conf variables appropriately.
conf.prefixChars = '@'
conf.replyWhenNotCommand = False
self.myVerbose = world.myVerbose
if self.cleanConfDir:
for filename in os.listdir(conf.confDir):
os.remove(os.path.join(conf.confDir, filename))
if self.cleanDataDir:
for filename in os.listdir(conf.dataDir):
os.remove(os.path.join(conf.dataDir, filename))
ircdb.users.reload()
ircdb.channels.reload()
if self.plugins is None:
raise ValueError, 'PluginTestCase must have a "plugins" attribute.'
self.nick = nick
self.prefix = ircutils.joinHostmask(nick, 'user', 'host.domain.tld')
self.irc = irclib.Irc(nick)
while self.irc.takeMsg():
pass
_ = Owner.loadPluginClass(self.irc, Owner)
_ = Owner.loadPluginClass(self.irc, Misc)
if isinstance(self.plugins, str):
if self.plugins not in ('Owner', 'Misc'):
module = Owner.loadPluginModule(self.plugins)
cb = Owner.loadPluginClass(self.irc, module)
else:
for name in self.plugins:
if name not in ('Owner', 'Misc'):
module = Owner.loadPluginModule(name)
cb = Owner.loadPluginClass(self.irc, module)
def tearDown(self):
self.irc.die()
gc.collect()
def _feedMsg(self, query, timeout=None):
if timeout is None:
timeout = self.timeout
if self.myVerbose:
print # Extra newline, so it's pretty.
msg = ircmsgs.privmsg(self.irc.nick, query, prefix=self.prefix)
if self.myVerbose:
print 'Feeding: %r' % msg
self.irc.feedMsg(msg)
fed = time.time()
response = self.irc.takeMsg()
while response is None and time.time() - fed < timeout:
time.sleep(0.1) # So it doesn't suck up 100% cpu.
drivers.run()
response = self.irc.takeMsg()
if self.myVerbose:
print 'Response: %r' % response
return response
def getMsg(self, query, timeout=None):
return self._feedMsg(query, timeout=timeout)
def feedMsg(self, query):
"""Just feeds it a message, that's all."""
self.irc.feedMsg(ircmsgs.privmsg(self.irc.nick, query,
prefix=self.prefix))
# These assertError/assertNoError are somewhat fragile. The proper way to
# do them would be to use a proxy for the irc object and intercept .error.
# But that would be hard, so I don't bother. When this breaks, it'll get
# fixed, but not until then.
def assertError(self, query):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
if lastGetHelp not in m.args[1]:
self.failUnless(m.args[1].startswith('Error:'),
'%r did not error: %s' % (query, m.args[1]))
def assertNotError(self, query):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failIf(m.args[1].startswith('Error:'),
'%r errored: %s' % (query, m.args[1]))
self.failIf(lastGetHelp in m.args[1],
'%r returned the help string.' % query)
def assertHelp(self, query):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failUnless(lastGetHelp in m.args[1])
def assertNoResponse(self, query, timeout=None):
m = self._feedMsg(query, timeout)
self.failIf(m, 'Unexpected response: %r' % m)
def assertResponse(self, query, expectedResponse):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.assertEqual(m.args[1], expectedResponse,
'%r != %r' % (expectedResponse, m.args[1]))
def assertRegexp(self, query, regexp, flags=re.I):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failUnless(re.search(regexp, m.args[1], flags),
'%r does not match %r' % (m.args[1], regexp))
def assertNotRegexp(self, query, regexp, flags=re.I):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failUnless(re.search(regexp, m.args[1], flags) is None,
'%r matched %r' % (m.args[1], regexp))
def assertRegexps(self, query, regexps):
started = time.time()
total = len(regexps)*self.timeout
while regexps and time.time() - started < total:
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
regexp = regexps.pop(0)
self.failUnless(re.search(regexp, m.args[1]),
'%r does not match %r' % (m.args[1], regexp))
self.failIf(time.time() - started > total)
def assertResponses(self, query, expectedResponses):
responses = []
started = time.time()
while len(responses) < len(expectedResponses) and \
time.time() - started > len(expectedResponses)*self.timeout:
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
responses.append(m)
self.assertEqual(len(expectedResponses), len(responses))
for (m, expected) in zip(responses, expectedResponses):
self.assertEqual(m.args[1], expected)
def assertAction(self, query, expectedResponse=None):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failUnless(ircmsgs.isAction(m))
if expectedResponse is not None:
self.assertEqual(ircmsgs.unAction(m), expectedResponse)
def assertActionRegexp(self, query, regexp, flags=re.I):
m = self._feedMsg(query)
if m is None:
raise TimeoutError, query
self.failUnless(ircmsgs.isAction(m))
s = ircmsgs.unAction(m)
self.failUnless(re.search(regexp, s, flags),
'%r does not match %r' % (s, regexp))
class ChannelPluginTestCase(PluginTestCase):
channel = '#test'
def setUp(self):
PluginTestCase.setUp(self)
self.irc.feedMsg(ircmsgs.join(self.channel, prefix=self.prefix))
def _feedMsg(self, query, timeout=None):
if timeout is None:
timeout = self.timeout
if self.myVerbose:
print # Newline, just like PluginTestCase.
if query[0] not in conf.prefixChars:
query = conf.prefixChars[0] + query
msg = ircmsgs.privmsg(self.channel, query, prefix=self.prefix)
if self.myVerbose:
print 'Feeding: %r' % msg
self.irc.feedMsg(msg)
fed = time.time()
response = self.irc.takeMsg()
while response is None and time.time() - fed < timeout:
drivers.run()
response = self.irc.takeMsg()
if response is not None:
if response.command == 'PRIVMSG':
args = list(response.args)
# Strip off nick: at beginning of response.
if args[1].startswith(self.nick) or \
args[1].startswith(ircutils.nickFromHostmask(self.prefix)):
args[1] = args[1].split(' ', 1)[1]
ret = ircmsgs.privmsg(*args)
else:
ret = response
else:
ret = None
if self.myVerbose:
print 'Returning: %r' % ret
return ret
def feedMsg(self, query):
"""Just feeds it a message, that's all."""
self.irc.feedMsg(ircmsgs.privmsg(self.channel, query,
prefix=self.prefix))
class PluginDocumentation:
def testAllCommandsHaveHelp(self):
for cb in self.irc.callbacks:
if isinstance(cb, callbacks.PrivmsgRegexp):
continue
if hasattr(cb, 'isCommand'):
for attr in cb.__class__.__dict__:
if cb.isCommand(attr):
self.failUnless(getattr(cb, attr).__doc__,
'%s has no syntax' % attr)
def testAllCommandsHaveMorehelp(self):
for cb in self.irc.callbacks:
if isinstance(cb, callbacks.PrivmsgRegexp):
continue
if hasattr(cb, 'isCommand'):
for attr in cb.__class__.__dict__:
if cb.isCommand(attr):
command = getattr(cb, attr)
helps = command.__doc__
self.failUnless(helps and len(helps.splitlines()) >= 3,
'%s has no help' % attr)
def testPluginHasDocumentation(self):
for cb in self.irc.callbacks:
m = sys.modules[cb.__class__.__module__]
self.failIf(m.__doc__ is None,
'%s has no module documentation'%cb.__class__.__name__)
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: