example/test.py

@@ -53,3 +53,14 @@ def test_client_receive_event(capsys, event_app_client):
     )
     captured = capsys.readouterr()
     assert captured.out == "Received event data: {'event': 'yes!'}\n"
+
+
+def test_status_and_healthy(event_app_client):
+    eac = event_app_client(target="say_hello", source="example/main.py")
+    response = eac.get("/healthy")
+    assert response.status_code == 200
+    assert response.text == "ok"
+
+    response = eac.get("/status")
+    assert response.status_code == 200
+    assert response.text == "ok"

pyproject.toml

 [tool.poetry]
 name = "ucam-faas"
-version = "0.1.1"
+version = "0.1.2"
 description = "Opinionated FaaS support framework extending Google's functions-framework"
 authors = ["University of Cambridge Information Services <devops-wilson@uis.cam.ac.uk>"]
 readme = "README.md"

ucam_faas/__init__.py

@@ -155,11 +155,21 @@ def _initialize_logging(debug):  # pragma: no cover
     return logger.handlers[:], structlog.wrap_logger(logger)
 
 
-def run_ucam_faas(target, source, host, port, debug):  # pragma: no cover
+def _initialize_ucam_faas_app(target, source, debug):
     handlers, logger = _initialize_logging(debug)
     app = functions_framework.create_app(target, source)
     app.logger.handlers = handlers
 
+    @app.route("/healthy")
+    @app.route("/status")
+    def get_status():
+        return "ok"
+
+    return app
+
+
+def run_ucam_faas(target, source, host, port, debug):  # pragma: no cover
+    app = _initialize_ucam_faas_app(target, source, debug)
     if debug:
         app.run(host, port, debug)
     else:

ucam_faas/testing.py

-import functions_framework
+from . import _initialize_ucam_faas_app
 
 try:
     from pytest import fixture
@@ -6,7 +6,7 @@ try:
     @fixture
     def event_app_client():
         def _event_app_client(target, source=None):
-            test_app = functions_framework.create_app(target, source)
+            test_app = _initialize_ucam_faas_app(target, source, True)
             return test_app.test_client()
 
         return _event_app_client