from wsgiref.validate import validator

import pytest

from paste import if_none_match

from ..common_wsgi import call_app

ETAG = "1234"


@pytest.fixture
def app():
    @validator
    @if_none_match
    @validator
    def app(environ, start_response):
        body = b"Hello, world!"
        etag_header = ("ETag", f'"{ETAG}"')
        if environ["REQUEST_METHOD"] == "HEAD":
            start_response(
                "200 OK",
                [
                    ("Content-Type", "text/plain"),
                    ("Content-Length", str(len(body))),
                    etag_header,
                ],
            )
            return []
        start_response("200 OK", [("Content-Type", "text/plain"), etag_header])
        return [body]

    return app


@pytest.mark.parametrize("method", ["POST", "PUT", "DELETE"])
def test_non_get_or_head_request_passthrough(app, method):
    environ = {"REQUEST_METHOD": method, "HTTP_IF_NONE_MATCH": f'"{ETAG}"'}
    response = call_app(app, environ)
    assert response.data == b"Hello, world!"
    assert response.status == "200 OK"
    assert ("Content-Type", "text/plain") in response.headers


def test_if_none_match_header_not_set(app):
    environ = {"REQUEST_METHOD": "GET"}
    response = call_app(app, environ)
    assert response.data == b"Hello, world!"
    assert response.status == "200 OK"
    assert ("Content-Type", "text/plain") in response.headers


@pytest.mark.parametrize("value", [f'"{ETAG}"', f'W/"{ETAG}"'])
def test_etag_matches_if_none_match_header(app, value):
    environ = {"REQUEST_METHOD": "GET", "HTTP_IF_NONE_MATCH": value}
    response = call_app(app, environ)
    assert response.data == b""
    assert response.status == "304 Not Modified"
    assert ("ETag", f'"{ETAG}"') in response.headers


@pytest.mark.parametrize(
    "value",
    [
        f'"{ETAG}", "54321"',
        f'"54321", "{ETAG}", "098765"',
        f'"54321", "{ETAG}"',
        f'W/"{ETAG}", "54321"',
        f'"54321", W/"{ETAG}", "098765"',
        f'"54321", W/"{ETAG}"',
    ],
)
def test_etag_matches_if_none_match_header_list(app, value):
    environ = {"REQUEST_METHOD": "GET", "HTTP_IF_NONE_MATCH": value}
    response = call_app(app, environ)
    assert response.data == b""
    assert response.status == "304 Not Modified"
    assert ("ETag", f'"{ETAG}"') in response.headers


def test_etag_matches_if_none_match_header_star(app):
    environ = {"REQUEST_METHOD": "GET", "HTTP_IF_NONE_MATCH": "*"}
    response = call_app(app, environ)
    assert response.data == b""
    assert response.status == "304 Not Modified"
    assert ("ETag", f'"{ETAG}"') in response.headers


def test_missing_etag_does_not_match_if_none_match_header_star():
    @validator
    @if_none_match
    @validator
    def app(environ, start_response):
        body = b"Hello, world!"
        if environ["REQUEST_METHOD"] == "HEAD":
            start_response(
                "200 OK",
                [
                    ("Content-Type", "text/plain"),
                    ("Content-Length", str(len(body))),
                ],
            )
            return []
        start_response("200 OK", [("Content-Type", "text/plain")])
        return [body]

    environ = {"REQUEST_METHOD": "GET", "HTTP_IF_NONE_MATCH": "*"}
    response = call_app(app, environ)
    assert response.data == b"Hello, world!"
    assert response.status == "200 OK"


def test_etag_does_not_match_if_none_match_header(app):
    environ = {"REQUEST_METHOD": "GET", "HTTP_IF_NONE_MATCH": '"54321"'}
    response = call_app(app, environ)
    assert response.data == b"Hello, world!"
    assert response.status == "200 OK"
    assert ("Content-Type", "text/plain") in response.headers


def test_etag_does_not_match_if_none_match_header_list(app):
    environ = {"REQUEST_METHOD": "GET", "HTTP_IF_NONE_MATCH": '"123","4567","*"'}
    response = call_app(app, environ)
    assert response.data == b"Hello, world!"
    assert response.status == "200 OK"
    assert ("Content-Type", "text/plain") in response.headers


def test_etag_does_not_match_if_none_match_malformed_header(app):
    environ = {"REQUEST_METHOD": "GET", "HTTP_IF_NONE_MATCH": "{ETAG}"}
    response = call_app(app, environ)
    assert response.data == b"400 Bad Request\n"
    assert response.status == "400 Bad Request"
    assert ("Content-Type", "text/plain") in response.headers