import pytest
from requests import ConnectionError

@pytest.mark.parametrize("path", ["web1", "web2"])
def test_web1_http_redirects_to_https(docker_compose, nginxproxy, path):
    r = nginxproxy.get("http://www.nginx-proxy.tld/%s/port" % path, allow_redirects=False)
    assert r.status_code == 301
    assert "Location" in r.headers
    assert "https://www.nginx-proxy.tld/%s/port" % path == r.headers['Location']

@pytest.mark.parametrize("path,port", [("web1", 81), ("web2", 82)])
def test_web1_https_is_forwarded(docker_compose, nginxproxy, path, port):
    r = nginxproxy.get("https://www.nginx-proxy.tld/%s/port" % path, allow_redirects=False)
    assert r.status_code == 200
    assert "answer from port %d\n" % port in r.text


@pytest.mark.parametrize("port", [81, 82])
def test_acme_challenge_does_not_work(docker_compose, nginxproxy, acme_challenge_path, port):
    with pytest.raises(ConnectionError):
        nginxproxy.get(
            f"http://www.nginx-proxy.tld:{port}/{acme_challenge_path}",
            allow_redirects=False
        )