Write test for mockup, #16

bddrest/mockup_server.py

@@ -0,0 +1,21 @@
+import contextlib
+import threading
+from wsgiref.simple_server import WSGIRequestHandler, WSGIServer
+
+
+@contextlib.contextmanager
+def http_server(app=None, handler_class=WSGIRequestHandler,
+                server_class=WSGIServer, bind=('localhost', 8082)):
+
+    server = server_class(bind, handler_class)
+    if app:
+        assert isinstance(server, WSGIServer)
+        server.set_app(app)
+    thread = threading.Thread(
+        target=server.serve_forever, name='sa-media test server.', daemon=True
+    )
+    thread.start()
+    url = f'http://{server.server_address[0]}:{server.server_address[1]}'
+    yield server, url
+    server.shutdown()
+    thread.join()

bddrest/tests/test_mockup_server.py

@@ -0,0 +1,27 @@
+import unittest
+
+import requests
+from nanohttp import Controller, html, Application
+
+from bddrest.mockup_server import http_server
+
+
+class App(Application):
+    def __init__(self):
+        super().__init__(root=Root())
+
+
+class Root(Controller):
+    @html
+    def index(self):
+        return 'test'
+
+
+class MockUpServerTestCase(unittest.TestCase):
+    with http_server(app=App(), bind=('localhost', 8087)) as (server, url):
+        response = requests.get(url)
+        assert response.content == b'test'
+
+
+if __name__ == '__main__':
+    unittest.main()

setup.py

@@ -10,9 +10,11 @@
 
 
 dependencies = [
+    'nanohttp',
     'webtest',
     'pyyaml',
-    'argcomplete'
+    'argcomplete',
+    'requests'
 ]