Added a test for get_correlation_id() to flask implementation

Author: chr1st1ank

tests/test_flask.py

@@ -63,6 +63,10 @@ def log_exception():
             logger.exception("Error occurred", exc_info=e)
         return {}
 
+    @app.route("/get-correlation-id")
+    def get_correlation_id():
+        return {'correlation_id': json_logging.get_correlation_id()}
+
     with app.test_client() as test_client:
         yield test_client, handler
 
@@ -121,6 +125,16 @@ def test_correlation_id_generated(client_and_log_handler):
     ), "autogenerated UUID doesn't have expected format"
 
 
+def test_get_correlation_id(client_and_log_handler):
+    """Test if json_logging.get_correlation_id() finds a given correlation ID"""
+    api_client, handler = client_and_log_handler
+
+    response = api_client.get("/get-correlation-id", headers={"X-Correlation-Id": "abc-def"})
+
+    assert response.status_code == 200
+    assert response.json["correlation_id"] == "abc-def"
+
+
 def test_extra_property(client_and_log_handler):
     """Test adding an extra property to a log message"""
     api_client, handler = client_and_log_handler