mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
a114bd3e16
The first addition, extend_query, can be used to take an existing query parameter dictionary and inject an *additions as replacement key/value pairs. The second, to_qs, converts a query parameter dictionary to a query string in the form "a=b&c=d". These two functions simplify and make dedupe_qs and quote_plus more efficient in terms of constructing custom query string overrides. Signed-off-by: Kevin Morris <kevr@0cost.org>
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
from collections import OrderedDict
|
|
from datetime import datetime
|
|
from zoneinfo import ZoneInfo
|
|
|
|
from aurweb import util
|
|
|
|
|
|
def test_timestamp_to_datetime():
|
|
ts = datetime.utcnow().timestamp()
|
|
dt = datetime.utcfromtimestamp(int(ts))
|
|
assert util.timestamp_to_datetime(ts) == dt
|
|
|
|
|
|
def test_as_timezone():
|
|
ts = datetime.utcnow().timestamp()
|
|
dt = util.timestamp_to_datetime(ts)
|
|
assert util.as_timezone(dt, "UTC") == dt.astimezone(tz=ZoneInfo("UTC"))
|
|
|
|
|
|
def test_dedupe_qs():
|
|
items = OrderedDict()
|
|
items["key1"] = "test"
|
|
items["key2"] = "blah"
|
|
items["key3"] = 1
|
|
|
|
# Construct and test our query string.
|
|
query_string = '&'.join([f"{k}={v}" for k, v in items.items()])
|
|
assert query_string == "key1=test&key2=blah&key3=1"
|
|
|
|
# Add key1=changed and key2=changed to the query and dedupe it.
|
|
deduped = util.dedupe_qs(query_string, "key1=changed", "key3=changed")
|
|
assert deduped == "key2=blah&key1=changed&key3=changed"
|
|
|
|
|
|
def test_number_format():
|
|
assert util.number_format(0.222, 2) == "0.22"
|
|
assert util.number_format(0.226, 2) == "0.23"
|
|
|
|
|
|
def test_extend_query():
|
|
""" Test extension of a query via extend_query. """
|
|
query = {"a": "b"}
|
|
extended = util.extend_query(query, ("a", "c"), ("b", "d"))
|
|
assert extended.get("a") == "c"
|
|
assert extended.get("b") == "d"
|
|
|
|
|
|
def test_to_qs():
|
|
""" Test conversion from a query dictionary to a query string. """
|
|
query = {"a": "b", "c": [1, 2, 3]}
|
|
qs = util.to_qs(query)
|
|
assert qs == "a=b&c=1&c=2&c=3"
|