Tests: simple python application.

1 files changed, 43 insertions, 0 deletions

diff --git a/test/unit.py b/test/unit.py
index b510b42c..d3d00fe0 100644
--- a/test/unit.py
+++ b/test/unit.py
@@ -7,6 +7,7 @@ import shutil
 import socket

 import tempfile

 import unittest

+from requests import Request, Session

 from subprocess import call

 from multiprocessing import Process

 

@@ -94,6 +95,12 @@ class TestUnit(unittest.TestCase):
             self.testdir + '/unit.log', self.testdir + '/control.unit.sock'):

             exit("Could not start unit")

 

+    def python_application(self, name, code):

+        os.mkdir(self.testdir + '/' + name)

+

+        with open(self.testdir + '/' + name + '/wsgi.py', 'w') as f:

+            f.write(code)

+

     def _stop(self):

         with open(self.testdir + '/unit.pid', 'r') as f:

             pid = f.read().rstrip()

@@ -200,3 +207,39 @@ class TestUnitControl(TestUnit):
     def _body_json(self, resp):

         m = re.search('.*?\x0d\x0a?\x0d\x0a?(.*)', resp, re.M | re.S)

         return json.loads(m.group(1))

+

+class TestUnitHTTP():

+

+    @classmethod

+    def http(self, method, host='127.0.0.1:7080', uri='/', **kwargs):

+        if 'sess' in kwargs:

+            sess = kwargs['sess']

+        else:

+            sess = Session()

+

+        body = None

+        if 'body' in kwargs:

+            body = kwargs['body']

+

+        headers = None

+        if 'headers' in kwargs:

+            headers = kwargs['headers']

+

+        req = Request('POST', 'http://' + host + uri, headers=headers)

+        prepped = req.prepare()

+

+        prepped.body = body

+

+        r = sess.send(prepped)

+

+        if 'keep' not in kwargs:

+            sess.close()

+            return r

+

+        return (r, sess)

+

+    def get(**kwargs):

+        return TestUnitHTTP.http('GET', **kwargs)

+

+    def post(**kwargs):

+        return TestUnitHTTP.http('POST', **kwargs)