Reworked so there's no double-import problem.

test/test.py

@@ -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):
+##     for filename in os.listdir(conf.logDir):
-        if filename == 'plugins':
+##         if filename == 'plugins':
-            continue
+##             continue
-        filename = os.path.join(conf.logDir, filename)
+##         filename = os.path.join(conf.logDir, filename)
-        os.remove(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:

test/test_Admin.py

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

test/test_Alias.py

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

test/test_Amazon.py

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

test/test_Babelfish.py

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

test/test_BadWords.py

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

test/test_Bugzilla.py

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

test/test_Channel.py

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

test/test_ChannelDB.py

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

test/test_Ctcp.py

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

test/test_DCC.py

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

test/test_Debian.py

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

test/test_Dict.py

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

test/test_Dunno.py

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

test/test_Ebay.py

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

test/test_Enforcer.py

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

test/test_Factoids.py

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

test/test_FixRelayBot.py

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

test/test_Friendly.py

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

test/test_Fun.py

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

test/test_FunDB.py

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

test/test_Gameknot.py

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

test/test_Google.py

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

test/test_Http.py

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

test/test_Karma.py

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

test/test_Lookup.py

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

test/test_Markov.py

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

test/test_Math.py

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

test/test_Misc.py

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

test/test_MoobotFactoids.py

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

test/test_Movies.py

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

test/test_Network.py

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

test/test_News.py

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

test/test_Note.py

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

test/test_OSU.py

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

test/test_Owner.py

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

test/test_Parter.py

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

test/test_Poll.py

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

test/test_Python.py

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

test/test_QuoteGrabs.py

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

test/test_Quotes.py

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

test/test_RSS.py

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

test/test_Relay.py

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

test/test_RootWarner.py

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

test/test_Services.py

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

test/test_Sourceforge.py

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

test/test_Status.py

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

test/test_Todo.py

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

test/test_Topic.py

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

test/test_URL.py

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

test/test_Unix.py

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

test/test_User.py

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

test/test_Utilities.py

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

test/test_callbacks.py

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

test/test_fix.py

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

test/test_ircdb.py

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

test/test_irclib.py

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

test/test_ircmsgs.py

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

test/test_ircutils.py

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

test/test_plugins.py

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

test/test_privmsgs.py

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

test/test_schedule.py

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

test/test_structures.py

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

test/test_utils.py

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

test/testsupport.py

@@ -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:
+