<reponame>sportsracer/retwist
# coding: utf-8

import json
from uuid import UUID

import pytest
from twisted.web.error import Error
from twisted.web.test.requesthelper import DummyRequest

from retwist.param import BoolParam, EnumParam, IntParam, JsonParam, LangParam, Param, UUIDParam, VersionParam


@pytest.fixture
def dummy_request():
    """
    Dummy request with some arguments used by tests in this module.
    """
    request = DummyRequest(b"/")
    # Mimic the way Twisted parses URL parameters:
    request.args = {
        b"id": [b"1234"],
        b"count": [b"20"],
        b"parent_id": [],  # happens when you pass "parent_id="
        b"child_id": [b"a", b"b"],  # happens when you pass "child_id=a&child_id=b". Not supported by retwist
        b"debug": [b"true"],
        b"verbose": [b"false"],
        b"type": [b"int"],
        b"lang": [b"de"],
        b"v": [b"1.0"],
        b"key": [b"523f2850-5646-4832-9507-e99e144328c8"],
        b"payload": [json.dumps({"msg": u"Ümlauts supported"})],
    }
    return request


def test_param(dummy_request):
    """
    Test basic parsing behavior.
    """

    param = Param()

    val = param.parse_from_request("id", dummy_request)
    assert val == "1234"

    with pytest.raises(Error) as exc_info:
        param.parse_from_request("parent_id", dummy_request)
    assert exc_info.value.status == b"400"

    with pytest.raises(Error) as exc_info:
        param.parse_from_request("child_id", dummy_request)
    assert exc_info.value.status == b"400"

    with pytest.raises(ValueError):
        Param(required=True, default="required params shouldn't have a default")


def test_missing_params(dummy_request):
    """
    Test behavior when parameters are missing.
    """

    # Return default if parameter is missing

    param_with_default = Param(default="default")

    val = param_with_default.parse_from_request("missing_key", dummy_request)
    assert val == "default"

    # Return error if required parameter is missing

    required_param = Param(required=True)

    with pytest.raises(Error) as exc_info:
        required_param.parse_from_request("missing_key", dummy_request)
    assert exc_info.value.status == b"400"


def test_bool_param(dummy_request):

    bool_param = BoolParam(default=False)

    val = bool_param.parse_from_request("debug", dummy_request)
    assert val is True

    val = bool_param.parse_from_request("enabled", dummy_request)
    assert val is False

    val = bool_param.parse_from_request("missing_key", dummy_request)
    assert val is False

    with pytest.raises(Error) as exc_info:
        bool_param.parse_from_request("id", dummy_request)
    assert exc_info.value.status == b"400"


def test_int_param(dummy_request):

    int_param = IntParam()

    val = int_param.parse_from_request("count", dummy_request)
    assert val == 20

    # Test error on malformed input

    with pytest.raises(Error) as exc_info:
        val = int_param.parse_from_request("lang", dummy_request)
    assert exc_info.value.status == b"400"

    # Test minimum and maximum boundaries

    int_param = IntParam(min_val=0, max_val=20)

    val = int_param.parse_from_request("count", dummy_request)
    assert val == 20

    dummy_request.addArg(b"count", b"-1")
    with pytest.raises(Error) as exc_info:
        val = int_param.parse_from_request("count", dummy_request)
    assert exc_info.value.status == b"400"

    dummy_request.addArg(b"count", b"21")
    with pytest.raises(Error) as exc_info:
        val = int_param.parse_from_request("count", dummy_request)
    assert exc_info.value.status == b"400"


def test_enum_param(dummy_request):

    enum_param = EnumParam(["int", "float"])

    val = enum_param.parse_from_request("type", dummy_request)
    assert val == "int"

    with pytest.raises(Error) as exc_info:
        enum_param.parse_from_request("debug", dummy_request)
    assert exc_info.value.status == b"400"


def test_lang_param(dummy_request):

    lang_param = LangParam(default="en")

    val = lang_param.parse_from_request("lang", dummy_request)
    assert val == "de"

    # Fall back to default

    del dummy_request.args[b"lang"]

    val = lang_param.parse_from_request("lang", dummy_request)
    assert val == "en"

    # ... or parse from HTTP header

    dummy_request.requestHeaders.setRawHeaders("Accept-Language", ["fr-CH, fr;q=0.9, en;q=0.8, de;q=0.7, *;q=0.5"])

    val = lang_param.parse_from_request("lang", dummy_request)
    assert val == "fr-CH"

    # Don't crap out on malformed HTTP header

    dummy_request.requestHeaders.setRawHeaders("Accept-Language", ["fr-CH;q=;"])
    lang_param.parse_from_request("lang", dummy_request)


def test_version_param(dummy_request):

    version_param = VersionParam()

    val = version_param.parse_from_request("v", dummy_request)
    assert val == (1, 0)

    # Handle malformed version

    dummy_request.args[b"v"] = [b"derp"]

    with pytest.raises(Error) as exc_info:
        version_param.parse_from_request("v", dummy_request)
    assert exc_info.value.status == b"400"


def test_json_param(dummy_request):

    json_param = JsonParam()

    val = json_param.parse_from_request("payload", dummy_request)
    assert val == {"msg": u"Ümlauts supported"}

    with pytest.raises(Error) as exc_info:
        json_param.parse(b'invalid: "json"')
    assert exc_info.value.status == b"400"


def test_json_param_with_schema():
    json_exceptions = pytest.importorskip("jsonschema.exceptions")

    schema = {
        "type": "object",
        "properties": {
            "x": {"type": "number"}
        },
        "additionalProperties": False
    }
    json_param = JsonParam(schema=schema)

    val = json_param.parse(b'{"x": 1337}')
    assert val == {"x": 1337}

    with pytest.raises(Error) as exc_info:
        json_param.parse(b'{"x": 1337, "y": "foo"}')
    assert exc_info.value.status == b"400"

    # Test that invalid schema gets rejected

    with pytest.raises(json_exceptions.SchemaError):
        JsonParam(schema={"type": "shmerg"})


def test_json_param_array():
    pytest.importorskip("jsonschema")

    json_param = JsonParam.array(items={"type": "number"}, max_items=1)

    val = json_param.parse(b'[1337]')
    assert val == [1337]

    with pytest.raises(Error) as exc_info:
        json_param.parse(b'["invalid"]')
    assert exc_info.value.status == b"400"

    with pytest.raises(Error) as exc_info:
        json_param.parse(b'[1337, 31337]')
    assert exc_info.value.status == b"400"

    uuid_param = JsonParam.array(items=JsonParam.UUID_TYPE)
    val = uuid_param.parse(b'["d46cb4aa-4e50-4a54-907c-a6db7ac9d646"]')
    assert val == ["d46cb4aa-4e50-4a54-907c-a6db7ac9d646"]

    with pytest.raises(Error) as exc_info:
        json_param.parse(b'["abcdef-invalid-uuid"]')
    assert exc_info.value.status == b"400"


def test_uuid_param(dummy_request):
    uuid_param = UUIDParam()

    val = uuid_param.parse_from_request("key", dummy_request)
    assert val == UUID("523f2850-5646-4832-9507-e99e144328c8")

    with pytest.raises(Error) as exc_info:
        uuid_param.parse(b'["abcdef-invalid-uuid"]')
    assert exc_info.value.status == b"400"

# coding: utf-8

import json
from uuid import UUID

import pytest
from twisted.web.error import Error
from twisted.web.test.requesthelper import DummyRequest

from retwist.param import BoolParam, EnumParam, IntParam, JsonParam, LangParam, Param, UUIDParam, VersionParam


@pytest.fixture
def dummy_request():
    """
    Dummy request with some arguments used by tests in this module.
    """
    request = DummyRequest(b"/")
    # Mimic the way Twisted parses URL parameters:
    request.args = {
        b"id": [b"1234"],
        b"count": [b"20"],
        b"parent_id": [],  # happens when you pass "parent_id="
        b"child_id": [b"a", b"b"],  # happens when you pass "child_id=a&child_id=b". Not supported by retwist
        b"debug": [b"true"],
        b"verbose": [b"false"],
        b"type": [b"int"],
        b"lang": [b"de"],
        b"v": [b"1.0"],
        b"key": [b"523f2850-5646-4832-9507-e99e144328c8"],
        b"payload": [json.dumps({"msg": u"Ümlauts supported"})],
    }
    return request


def test_param(dummy_request):
    """
    Test basic parsing behavior.
    """

    param = Param()

    val = param.parse_from_request("id", dummy_request)
    assert val == "1234"

    with pytest.raises(Error) as exc_info:
        param.parse_from_request("parent_id", dummy_request)
    assert exc_info.value.status == b"400"

    with pytest.raises(Error) as exc_info:
        param.parse_from_request("child_id", dummy_request)
    assert exc_info.value.status == b"400"

    with pytest.raises(ValueError):
        Param(required=True, default="required params shouldn't have a default")


def test_missing_params(dummy_request):
    """
    Test behavior when parameters are missing.
    """

    # Return default if parameter is missing

    param_with_default = Param(default="default")

    val = param_with_default.parse_from_request("missing_key", dummy_request)
    assert val == "default"

    # Return error if required parameter is missing

    required_param = Param(required=True)

    with pytest.raises(Error) as exc_info:
        required_param.parse_from_request("missing_key", dummy_request)
    assert exc_info.value.status == b"400"


def test_bool_param(dummy_request):

    bool_param = BoolParam(default=False)

    val = bool_param.parse_from_request("debug", dummy_request)
    assert val is True

    val = bool_param.parse_from_request("enabled", dummy_request)
    assert val is False

    val = bool_param.parse_from_request("missing_key", dummy_request)
    assert val is False

    with pytest.raises(Error) as exc_info:
        bool_param.parse_from_request("id", dummy_request)
    assert exc_info.value.status == b"400"


def test_int_param(dummy_request):

    int_param = IntParam()

    val = int_param.parse_from_request("count", dummy_request)
    assert val == 20

    # Test error on malformed input

    with pytest.raises(Error) as exc_info:
        val = int_param.parse_from_request("lang", dummy_request)
    assert exc_info.value.status == b"400"

    # Test minimum and maximum boundaries

    int_param = IntParam(min_val=0, max_val=20)

    val = int_param.parse_from_request("count", dummy_request)
    assert val == 20

    dummy_request.addArg(b"count", b"-1")
    with pytest.raises(Error) as exc_info:
        val = int_param.parse_from_request("count", dummy_request)
    assert exc_info.value.status == b"400"

    dummy_request.addArg(b"count", b"21")
    with pytest.raises(Error) as exc_info:
        val = int_param.parse_from_request("count", dummy_request)
    assert exc_info.value.status == b"400"


def test_enum_param(dummy_request):

    enum_param = EnumParam(["int", "float"])

    val = enum_param.parse_from_request("type", dummy_request)
    assert val == "int"

    with pytest.raises(Error) as exc_info:
        enum_param.parse_from_request("debug", dummy_request)
    assert exc_info.value.status == b"400"


def test_lang_param(dummy_request):

    lang_param = LangParam(default="en")

    val = lang_param.parse_from_request("lang", dummy_request)
    assert val == "de"

    # Fall back to default

    del dummy_request.args[b"lang"]

    val = lang_param.parse_from_request("lang", dummy_request)
    assert val == "en"

    # ... or parse from HTTP header

    dummy_request.requestHeaders.setRawHeaders("Accept-Language", ["fr-CH, fr;q=0.9, en;q=0.8, de;q=0.7, *;q=0.5"])

    val = lang_param.parse_from_request("lang", dummy_request)
    assert val == "fr-CH"

    # Don't crap out on malformed HTTP header

    dummy_request.requestHeaders.setRawHeaders("Accept-Language", ["fr-CH;q=;"])
    lang_param.parse_from_request("lang", dummy_request)


def test_version_param(dummy_request):

    version_param = VersionParam()

    val = version_param.parse_from_request("v", dummy_request)
    assert val == (1, 0)

    # Handle malformed version

    dummy_request.args[b"v"] = [b"derp"]

    with pytest.raises(Error) as exc_info:
        version_param.parse_from_request("v", dummy_request)
    assert exc_info.value.status == b"400"


def test_json_param(dummy_request):

    json_param = JsonParam()

    val = json_param.parse_from_request("payload", dummy_request)
    assert val == {"msg": u"Ümlauts supported"}

    with pytest.raises(Error) as exc_info:
        json_param.parse(b'invalid: "json"')
    assert exc_info.value.status == b"400"


def test_json_param_with_schema():
    json_exceptions = pytest.importorskip("jsonschema.exceptions")

    schema = {
        "type": "object",
        "properties": {
            "x": {"type": "number"}
        },
        "additionalProperties": False
    }
    json_param = JsonParam(schema=schema)

    val = json_param.parse(b'{"x": 1337}')
    assert val == {"x": 1337}

    with pytest.raises(Error) as exc_info:
        json_param.parse(b'{"x": 1337, "y": "foo"}')
    assert exc_info.value.status == b"400"

    # Test that invalid schema gets rejected

    with pytest.raises(json_exceptions.SchemaError):
        JsonParam(schema={"type": "shmerg"})


def test_json_param_array():
    pytest.importorskip("jsonschema")

    json_param = JsonParam.array(items={"type": "number"}, max_items=1)

    val = json_param.parse(b'[1337]')
    assert val == [1337]

    with pytest.raises(Error) as exc_info:
        json_param.parse(b'["invalid"]')
    assert exc_info.value.status == b"400"

    with pytest.raises(Error) as exc_info:
        json_param.parse(b'[1337, 31337]')
    assert exc_info.value.status == b"400"

    uuid_param = JsonParam.array(items=JsonParam.UUID_TYPE)
    val = uuid_param.parse(b'["d46cb4aa-4e50-4a54-907c-a6db7ac9d646"]')
    assert val == ["d46cb4aa-4e50-4a54-907c-a6db7ac9d646"]

    with pytest.raises(Error) as exc_info:
        json_param.parse(b'["abcdef-invalid-uuid"]')
    assert exc_info.value.status == b"400"


def test_uuid_param(dummy_request):
    uuid_param = UUIDParam()

    val = uuid_param.parse_from_request("key", dummy_request)
    assert val == UUID("523f2850-5646-4832-9507-e99e144328c8")

    with pytest.raises(Error) as exc_info:
        uuid_param.parse(b'["abcdef-invalid-uuid"]')
    assert exc_info.value.status == b"400"
# coding: utf-8 Dummy request with some arguments used by tests in this module. # Mimic the way Twisted parses URL parameters: # happens when you pass "parent_id=" # happens when you pass "child_id=a&child_id=b". Not supported by retwist Test basic parsing behavior. Test behavior when parameters are missing. # Return default if parameter is missing # Return error if required parameter is missing # Test error on malformed input # Test minimum and maximum boundaries # Fall back to default # ... or parse from HTTP header # Don't crap out on malformed HTTP header # Handle malformed version # Test that invalid schema gets rejected