Added basic unit tests

Author: jamesroberts

README.md

@@ -27,7 +27,7 @@ def hello_world():
 
 
 if __name__ == "__main__":
-    fastwsgi.run(wsgi_app=app, host="0.0.0.0", port=5000)
+    fastwsgi.run(wsgi_app=app, host="127.0.0.1", port=5000)
 ```
 
 
@@ -39,9 +39,16 @@ def application(environ, start_response):
     return [b"Hello, World!"]
 
 if __name__ == "__main__":
-    fastwsgi.run(wsgi_app=application, host="0.0.0.0", port=5000)
+    fastwsgi.run(wsgi_app=application, host="127.0.0.1", port=5000)
 ```
 
+## Testing
+
+To run the test suite using [pytest](https://docs.pytest.org/en/latest/getting-started.html), run the following command:
+
+```bash
+python3 -m pytest
+```
 
 ## TODO
 

example.py

@@ -9,10 +9,5 @@ def hello_world():
     return "Hello, World!", 200
 
 
-def application(environ, start_response):
-    start_response('200 OK', [('Content-Type', 'text/html')])
-    return [b"Hello, World!"]
-
-
 if __name__ == "__main__":
-    fastwsgi.run(wsgi_app=application, host="0.0.0.0", port=5000)
+    fastwsgi.run(wsgi_app=app, host="0.0.0.0", port=5000)

tests/conftest.py

@@ -0,0 +1,25 @@
+import pytest
+import fastwsgi
+from multiprocessing import Process
+
+
+HOST = "127.0.0.1"
+PORT = 8080
+
+
+class ServerProcess:
+    def __init__(self, application, host=HOST, port=PORT) -> None:
+        self.process = Process(target=fastwsgi.run, args=(application, host, port))
+
+    def __enter__(self):
+        self.process.start()
+        return self.process
+
+    def __exit__(self, exc_type, exc_value, exc_tb):
+        if self.process.is_alive():
+            self.process.terminate()
+
+
+@pytest.fixture(autouse=True, scope="session")
+def server_process():
+    return ServerProcess

tests/test_flask.py

@@ -0,0 +1,16 @@
+import requests
+from flask import Flask
+
+app = Flask(__name__)
+
+
+@app.get("/")
+def hello_world():
+    return "Hello, World!", 200
+
+
+def test_uwsgi_hello_world(server_process):
+    with server_process(app):
+        result = requests.get("http://127.0.0.1:8080/")
+        assert result.status_code == 200
+        assert result.text == "Hello, World!"

tests/test_uwsgi.py

@@ -0,0 +1,13 @@
+import requests
+
+
+def wsgi_app(environ, start_response):
+    start_response('200 OK', [('Content-Type', 'text/html')])
+    return [b"Hello, World!"]
+
+
+def test_uwsgi_hello_world(server_process):
+    with server_process(wsgi_app):
+        result = requests.get("http://127.0.0.1:8080/")
+        assert result.status_code == 200
+        assert result.text == "Hello, World!"