Add testing command

1 files changed, 9 insertions, 0 deletions

diff --git a/local/utility.py b/local/utility.py
index 7c66d9e..abb0915 100644
--- a/local/utility.py
+++ b/local/utility.py
@@ -1,9 +1,12 @@
 import urllib2
 
 import supybot.conf as conf
+import supybot.world as world
 import supybot.ircutils as ircutils
 import supybot.registry as registry
 
+import globals
+
 def registryValue(plugin, name, channel=None, value=True):
     group = conf.supybot.plugins.get(plugin)
     names = registry.split(name)
@@ -62,6 +65,12 @@ def getShortURL(longurl):
         url = response.info().getheader('Location')
     return ircutils.mircColor(url, "purple")
 
+def saveMessages(msgs):
+    """ Saves the last messages so that the plugin can be easily tested """
+    if world.testing:
+        return
+    globals.messageList = msgs
+
 # Possible colours:
 # white, black, (light/dark) blue, (light) green, red, brown, purple,
 # orange, yellow, teal, pink, light/dark gray/grey