Adding the test cases

tests/test_get_send_file_max_age.py

@@ -0,0 +1,43 @@
+import pytest
+from unittest.mock import MagicMock, patch
+from datetime import timedelta
+from flask import Flask, current_app
+from flask.app import Flask as FlaskApp
+from flask import branch_coverage, track_coverage
+import json
+
+def save_coverage_to_json(file_path='/Users/jannesvandenbogert/Documents/GitHub/flask/coverage_result.json'):
+    with open(file_path, 'w') as json_file:
+        json.dump(branch_coverage, json_file, indent=4)
+
+def create_test_app(config):
+    app = Flask(__name__)
+    app.config.update(config)
+    return app
+
+def test_get_send_file_max_age_none():
+    app = create_test_app({"SEND_FILE_MAX_AGE_DEFAULT": None})
+    with app.app_context():
+        max_age = app.get_send_file_max_age("test.txt")
+        assert max_age is None
+        assert branch_coverage["get_send_file_max_age_1"] == True
+
+def test_get_send_file_max_age_timedelta():
+    app = create_test_app({"SEND_FILE_MAX_AGE_DEFAULT": timedelta(hours=1)})
+    with app.app_context():
+        max_age = app.get_send_file_max_age("test.txt")
+        assert max_age == 3600
+        assert branch_coverage["get_send_file_max_age_2"] == True
+
+def test_get_send_file_max_age_int():
+    app = create_test_app({"SEND_FILE_MAX_AGE_DEFAULT": 3600})
+    with app.app_context():
+        max_age = app.get_send_file_max_age("test.txt")
+        assert max_age == 3600
+        assert branch_coverage["get_send_file_max_age_3"] == True
+
+def test_branch_coverage():
+    save_coverage_to_json()
+    assert branch_coverage["get_send_file_max_age_1"] == True
+    assert branch_coverage["get_send_file_max_age_2"] == True
+    assert branch_coverage["get_send_file_max_age_3"] == True