mirror of
https://github.com/kennethreitz/responder.git
synced 2026-07-21 18:39:29 +00:00
103 lines
2.7 KiB
Python
103 lines
2.7 KiB
Python
import inspect
|
|
|
|
import pytest
|
|
|
|
from responder import models
|
|
from responder.models import CaseInsensitiveDict
|
|
|
|
_default_query = "q=%7b%20hello%20%7d&name=myname&user_name=test_user"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"query, expected",
|
|
[
|
|
pytest.param(
|
|
_default_query,
|
|
{"q": ["{ hello }"], "name": ["myname"], "user_name": ["test_user"]},
|
|
id="parse query with unique keys",
|
|
),
|
|
pytest.param(
|
|
"q=1&q=2&q=3", {"q": ["1", "2", "3"]}, id="parse query with the same key"
|
|
),
|
|
],
|
|
)
|
|
def test_query_dict(query, expected):
|
|
d = models.QueryDict(query)
|
|
assert d == expected
|
|
|
|
|
|
def test_query_dict_get():
|
|
d = models.QueryDict(_default_query)
|
|
|
|
assert d["user_name"] == "test_user"
|
|
assert d.get("key_none_exist") is None
|
|
|
|
|
|
def test_query_dict_get_list():
|
|
d = models.QueryDict(_default_query)
|
|
|
|
assert d.get_list("user_name") == ["test_user"]
|
|
assert d.get_list("key_none_exist") == []
|
|
assert d.get_list("key_none_exist", ["foo"]) == ["foo"]
|
|
|
|
|
|
def test_query_dict_keeps_blank_values():
|
|
d = models.QueryDict("key=value&empty=&multi=&multi=two")
|
|
|
|
assert d["empty"] == ""
|
|
assert d.get("empty") == ""
|
|
assert d.get_list("multi") == ["", "two"]
|
|
|
|
|
|
def test_query_dict_items_list():
|
|
d = models.QueryDict(_default_query)
|
|
|
|
items_list = d.items_list()
|
|
assert inspect.isgenerator(items_list)
|
|
assert dict(items_list) == {
|
|
"q": ["{ hello }"],
|
|
"name": ["myname"],
|
|
"user_name": ["test_user"],
|
|
}
|
|
|
|
|
|
def test_query_dict_items():
|
|
d = models.QueryDict(_default_query)
|
|
|
|
items = d.items()
|
|
assert inspect.isgenerator(items)
|
|
assert dict(items) == {"q": "{ hello }", "name": "myname", "user_name": "test_user"}
|
|
|
|
|
|
class TestCaseInsensitiveDict:
|
|
def test_set_and_get(self):
|
|
d = CaseInsensitiveDict()
|
|
d["Content-Type"] = "text/html"
|
|
assert d["content-type"] == "text/html"
|
|
assert d["CONTENT-TYPE"] == "text/html"
|
|
|
|
def test_contains(self):
|
|
d = CaseInsensitiveDict()
|
|
d["X-Custom"] = "value"
|
|
assert "x-custom" in d
|
|
assert "X-CUSTOM" in d
|
|
assert "missing" not in d
|
|
|
|
def test_get_default(self):
|
|
d = CaseInsensitiveDict()
|
|
assert d.get("missing") is None
|
|
assert d.get("missing", "default") == "default"
|
|
d["Key"] = "val"
|
|
assert d.get("KEY") == "val"
|
|
|
|
def test_update(self):
|
|
d = CaseInsensitiveDict()
|
|
d.update({"Content-Type": "text/html", "Accept": "json"})
|
|
assert d["content-type"] == "text/html"
|
|
assert d["accept"] == "json"
|
|
|
|
def test_update_kwargs(self):
|
|
d = CaseInsensitiveDict()
|
|
d.update(key1="val1", key2="val2")
|
|
assert d["key1"] == "val1"
|