Make tests look Rubyish

2 files changed, 115 insertions, 0 deletions

diff --git a/local/testing/ExpectationPluginTestCase.py b/local/testing/ExpectationPluginTestCase.py
new file mode 100644
index 0000000..af6e56d
--- /dev/null
+++ b/local/testing/ExpectationPluginTestCase.py
@@ -0,0 +1,115 @@
+###
+from supybot.log import info
+from supybot.test import *
+
+from sys import stdout
+from time import sleep
+
+import re
+import urllib
+
+from ..utility import clean
+
+class ExpectationPluginTestCase(PluginTestCase):
+    plugins       = {}
+
+    def describe(self, query, *args):
+        m = self._feedMsg('get ' + query)
+        manyEs = tcolors.FAIL + 'E' * len(args) + tcolors.ENDC
+        if m is None:
+            print manyEs
+            raise TimeoutError, query
+        if m.args[1].startswith('Error:'):
+            print manyEs
+            self.fail('%r errored: %s' % (query, m.args[1]))
+
+        errors = []
+        for expectation in args:
+            expectation.reply = m.args[1]
+            if expectation.evaluate():
+                stdout.write(tcolors.OKGREEN + '.' + tcolors.ENDC)
+            else:
+                stdout.write(tcolors.FAIL + 'F' + tcolors.ENDC)
+                errors.append(expectation.getSummary())
+            stdout.flush()
+        stdout.write('')
+        stdout.flush()
+
+        if errors:
+            print "\n%sWhile describing %s" % (tcolors.FAIL, query)
+            for error in errors:
+                print "- Failed to assert that %s" % (error,)
+            print tcolors.ENDC
+            self.fail()
+
+    def sendRequest(self, file):
+        """ Opens the `samples` folder and sends a file as a request
+            to the plugin's server """
+        if file in self.files:
+            content = self.files[file]
+        else:
+            with open('samples/' + file + '.json', 'r') as content_file:
+                content = content_file.read()
+            self.files[file] = content
+        res = urllib.urlopen('http://localhost:' + str(self.port), 'payload=' + content)
+
+    def testDocumentation(self):
+        if self.__class__ == ExpectationPluginTestCase:
+            return
+        else:
+            PluginTestCase.testDocumentation(self)
+
+    def setUp(self):
+        # Prevent supybot's testDocumentation from going mad
+        PluginTestCase.setUp(self)
+
+    def tearDown(self):
+        # Add a space between our LEDs and python's OK message
+        PluginTestCase.tearDown(self)
+        stdout.write(' ')
+        stdout.flush()
+
+class tcolors:
+    HEADER  = '\033[95m'
+    OKBLUE  = '\033[94m'
+    OKGREEN = '\033[92m'
+    WARNING = '\033[93m'
+    FAIL    = '\033[91m'
+    ENDC    = '\033[0m'
+
+def expect(query):
+    m = GithubTestCase._feedMsg('get' + query)
+    return Expectation(m.args[1])
+
+def it():
+    return Expectation()
+
+class Expectation:
+    def __init__(self):
+        self.error = ''
+        self.should = self
+        self.to     = self
+
+    def evaluate(self):
+        return self.assertion()
+
+    def cleanReply(self):
+        return clean(self.reply)
+
+    def getSummary(self):
+        if self.assertionParameter:
+            return self.summary % (self.cleanReply(), self.assertionParameter)
+        else:
+            return self.summary % (self.cleanReply())
+
+    def contain(self, what):
+        self.assertion = self.contains
+        self.assertionParameter = what
+        self.summary = "'%s' contains '%s'"
+        return self
+
+    def contains(self, flags=re.I):
+        return re.search(self.assertionParameter, self.cleanReply(), flags)
+
+class Object(object):
+    pass
diff --git a/local/testing/__init__.py b/local/testing/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/local/testing/__init__.py