1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
import pytest
from psqlextra.fields import HStoreField
def test_hstore_field_deconstruct():
"""Tests whether the :see:HStoreField's deconstruct() method works
properly."""
original_kwargs = dict(uniqueness=["beer", "other"], required=[])
_, _, _, new_kwargs = HStoreField(**original_kwargs).deconstruct()
for key, value in original_kwargs.items():
assert new_kwargs[key] == value
@pytest.mark.parametrize(
"input,output",
[
(dict(key1=1, key2=2), dict(key1="1", key2="2")),
(dict(key1="1", key2="2"), dict(key1="1", key2="2")),
(
dict(key1=1, key2=None, key3="3"),
dict(key1="1", key2=None, key3="3"),
),
([1, 2, 3], ["1", "2", "3"]),
(["1", "2", "3"], ["1", "2", "3"]),
],
)
def test_hstore_field_get_prep_value(input, output):
"""Tests whether the :see:HStoreField's get_prep_value method works
properly."""
assert HStoreField().get_prep_value(input) == output
|
