Audit a2eb44b9783f

from datetime import timedelta
from flask.app import _make_timedelta
import flask
import pytest
import flask.app
import werkzeug.serving
from flask.app import Flask

def test_corral_make_timedelta_seconds_conversion():
    assert _make_timedelta(None) is None
    assert _make_timedelta(timedelta(seconds=15)) == timedelta(seconds=15)

    result = _make_timedelta(30)
    assert result == timedelta(seconds=30)
    assert result.total_seconds() == 30

def test_app_corral_cli_runner_kwargs():
    app = flask.Flask("test_app_corral")

    runner = app.test_cli_runner(echo_stdin=True)
    assert runner.echo_stdin is True

    class CustomCliRunner:
        def __init__(self, app, custom_arg=None):
            self.app = app
            self.custom_arg = custom_arg

    app.test_cli_runner_class = CustomCliRunner
    custom_runner = app.test_cli_runner(custom_arg="expected_value")
    assert custom_runner.custom_arg == "expected_value"

def test_legacy_subclass_method_signature_receives_app_context():
    with pytest.deprecated_call():

        class LegacyFlask(flask.Flask):
            def dispatch_request(self):
                return super().dispatch_request()

    app = LegacyFlask(__name__)

    @app.route("/")
    def index():
        return "Hello Legacy"

    client = app.test_client()
    rv = client.get("/")
    assert rv.status_code == 200
    assert rv.data == b"Hello Legacy"

def test_app_run_resets_got_first_request_flag(monkeypatch):
    monkeypatch.delenv("FLASK_RUN_FROM_CLI", raising=False)

    def dummy_run_simple(*args, **kwargs):
        pass

    monkeypatch.setattr(werkzeug.serving, "run_simple", dummy_run_simple)

    app = flask.Flask("test_app_corral_reset_flag")
    app._got_first_request = True

    app.run()

    assert app._got_first_request is False

def test_subclass_deprecated_method_signature_warning():
    with pytest.warns(
        DeprecationWarning,
        match="The 'dispatch_request' method now takes 'ctx: AppContext'",
    ):

        class DeprecatedDispatchApp(Flask):
            def dispatch_request(self):
                return "ok"

def test_template_context_processor_precedence():
    app = flask.Flask("test_corral_app")

    @app.context_processor
    def inject_defaults():
        return {"override_me": "default_value", "other": "processor_value"}

    with app.test_request_context("/") as ctx:
        context = {"override_me": "explicit_value"}
        app.update_template_context(ctx, context)
        assert context["override_me"] == "explicit_value"
        assert context["other"] == "processor_value"

def test_make_shell_context_includes_g_corral():
    app = flask.app.Flask("test_app")
    context = app.make_shell_context()
    assert "g" in context
    assert context["g"] is flask.g
    assert context["app"] is app

def test_legacy_subclass_remove_ctx_deprecation_wrapper():
    with pytest.deprecated_call():
        class LegacyFlask(flask.Flask):
            def do_teardown_appcontext(self, exc):
                self.teardown_exc = exc

    app = LegacyFlask("legacy_app")
    ctx = app.app_context()

    app.do_teardown_appcontext(ctx, None)
    assert hasattr(app, "teardown_exc")
    assert app.teardown_exc is None

def test_get_send_file_max_age_timedelta_corral():
    app = flask.Flask("test_app")
    app.config["SEND_FILE_MAX_AGE_DEFAULT"] = timedelta(seconds=3600)
    assert app.get_send_file_max_age("foo.txt") == 3600

def test_open_instance_resource_binary_mode_corral(tmp_path):
    app = flask.Flask("test_app", instance_path=str(tmp_path))
    test_file = tmp_path / "test.bin"
    test_file.write_bytes(b"corral_test_bytes")

    with app.open_instance_resource("test.bin", "rb") as f:
        content = f.read()

    assert content == b"corral_test_bytes"