"""custom-html — upgrade stage (D2): deploy the previous published version, write data, upgrade
to the current/$REF version, and assert the app stays healthy and data survives."""
import os
import sys

import pytest

sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "runner"))
from harness import lifecycle  # noqa: E402

MARKER_PATH = "/usr/share/nginx/html/ci-marker.txt"


@pytest.fixture
def old_app(recipe, app_domain, request):
    prev = lifecycle.previous_version(recipe)
    if not prev:
        pytest.skip(f"{recipe}: no previous published version to upgrade from")
    lifecycle.janitor()
    request.addfinalizer(lambda: lifecycle.teardown_app(app_domain))
    lifecycle.deploy_app(recipe, app_domain, version=prev)
    lifecycle.wait_healthy(app_domain)
    return app_domain, prev


def test_upgrade_preserves_data(old_app):
    domain, prev = old_app
    # write a data marker into the served volume (nginx serves /usr/share/nginx/html)
    lifecycle.exec_in_app(domain, ["sh", "-c", f"echo upgrade-survives > {MARKER_PATH}"])
    assert lifecycle.http_body(domain, "/ci-marker.txt").strip() == "upgrade-survives"

    # upgrade previous -> current/$REF
    lifecycle.upgrade_app(domain, version=os.environ.get("VERSION") or None)
    lifecycle.wait_healthy(domain)

    # app healthy and the data written before the upgrade is still there
    assert lifecycle.http_get(domain, "/") == 200
    assert lifecycle.http_body(domain, "/ci-marker.txt").strip() == "upgrade-survives", \
        "data did not survive the upgrade"