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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
import pytest
from django.db.utils import IntegrityError
from psqlextra.fields import HStoreField
from . import migrations
from .fake_model import get_fake_model
def test_hstore_required_migration_create_drop_model():
"""Tests whether constraints are properly created and dropped when creating
and dropping a model."""
required = ["beer", "cookies"]
test = migrations.create_drop_model(
HStoreField(required=required), ["ADD CONSTRAINT", "DROP CONSTRAINT"]
)
with test as calls:
assert len(calls["ADD CONSTRAINT"]) == len(required)
assert len(calls["DROP CONSTRAINT"]) == len(required)
def test_hstore_required_migration_alter_db_table():
"""Tests whether constraints are renamed properly when renaming the
database table."""
test = migrations.alter_db_table(
HStoreField(required=["beer", "cookie"]),
["RENAME CONSTRAINT", "ADD CONSTRAINT", "DROP CONSTRAINT"],
)
with test as calls:
assert len(calls["RENAME CONSTRAINT"]) == 2
assert len(calls.get("ADD CONSTRAINT", [])) == 0
assert len(calls.get("DROP CONSTRAINT", [])) == 0
def test_hstore_required_add_field():
"""Tests whether adding a field properly creates the constraints."""
test = migrations.add_field(
HStoreField(required=["beer"]), ["ADD CONSTRAINT", "DROP CONSTRAINT"]
)
with test as calls:
assert len(calls.get("ADD CONSTRAINT", [])) == 1
assert len(calls.get("DROP CONSTRAINT", [])) == 0
def test_hstore_required_remove_field():
"""Tests whether removing a field properly removes the constraint."""
test = migrations.remove_field(
HStoreField(required=["beer"]), ["ADD CONSTRAINT", "DROP CONSTRAINT"]
)
with test as calls:
assert len(calls.get("ADD CONSTRAINT", [])) == 0
assert len(calls.get("DROP CONSTRAINT", [])) == 1
def test_hstore_required_alter_field_nothing():
"""Tests whether no constraints are dropped when not changing anything in
the required."""
test = migrations.alter_field(
HStoreField(required=["beer"]),
HStoreField(required=["beer"]),
["ADD CONSTRAINT", "DROP CONSTRAINT"],
)
with test as calls:
assert len(calls.get("ADD CONSTRAINT", [])) == 0
assert len(calls.get("DROP CONSTRAINT", [])) == 0
def test_hstore_required_alter_field_add():
"""Tests whether only one constraint is created when adding another key to
the required."""
test = migrations.alter_field(
HStoreField(required=["beer"]),
HStoreField(required=["beer", "beer1"]),
["ADD CONSTRAINT", "DROP CONSTRAINT"],
)
with test as calls:
assert len(calls.get("ADD CONSTRAINT", [])) == 1
assert len(calls.get("DROP CONSTRAINT", [])) == 0
def test_hstore_required_alter_field_remove():
"""Tests whether one constraint is dropped when removing a key from
required."""
test = migrations.alter_field(
HStoreField(required=["beer"]),
HStoreField(required=[]),
["ADD CONSTRAINT", "DROP CONSTRAINT"],
)
with test as calls:
assert len(calls.get("ADD CONSTRAINT", [])) == 0
assert len(calls.get("DROP CONSTRAINT", [])) == 1
def test_hstore_required_rename_field():
"""Tests whether renaming a field doesn't cause the constraint to be re-
created."""
test = migrations.rename_field(
HStoreField(required=["beer", "cookies"]),
["RENAME CONSTRAINT", "ADD CONSTRAINT", "DROP CONSTRAINT"],
)
with test as calls:
assert len(calls.get("RENAME CONSTRAINT", [])) == 2
assert len(calls.get("ADD CONSTRAINT", [])) == 0
assert len(calls.get("DROP CONSTRAINT", [])) == 0
def test_hstore_required_required_enforcement():
"""Tests whether the constraints are actually properly enforced."""
model = get_fake_model({"title": HStoreField(required=["en"])})
with pytest.raises(IntegrityError):
model.objects.create(title={"ar": "hello"})
def test_hstore_required_no_required():
"""Tests whether setting `required` to False casues no requiredness
constraints to be added."""
model = get_fake_model({"title": HStoreField(required=False)})
model.objects.create(title={"ar": "hello"})
|
