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
140
141
142
143
144
145
146
147
148
149
|
import pytest
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import ARRAY
from sqlalchemy_utils import (
assert_max_length,
assert_max_value,
assert_min_value,
assert_non_nullable,
assert_nullable
)
from tests import TestCase
class AssertionTestCase(TestCase):
dns = 'postgres://postgres@localhost/sqlalchemy_utils_test'
def create_models(self):
class User(self.Base):
__tablename__ = 'user'
id = sa.Column(sa.Integer, primary_key=True)
name = sa.Column(sa.String(20))
age = sa.Column(sa.Integer, nullable=False)
email = sa.Column(sa.String(200), nullable=False, unique=True)
fav_numbers = sa.Column(ARRAY(sa.Integer))
__table_args__ = (
sa.CheckConstraint(sa.and_(age >= 0, age <= 150)),
sa.CheckConstraint(
sa.and_(
sa.func.array_length(fav_numbers, 1) <= 8
)
)
)
self.User = User
def setup_method(self, method):
TestCase.setup_method(self, method)
user = self.User(
name='Someone',
email='someone@example.com',
age=15,
fav_numbers=[1, 2, 3]
)
self.session.add(user)
self.session.commit()
self.user = user
class TestAssertMaxLengthWithArray(AssertionTestCase):
def test_with_max_length(self):
assert_max_length(self.user, 'fav_numbers', 8)
assert_max_length(self.user, 'fav_numbers', 8)
def test_smaller_than_max_length(self):
with pytest.raises(AssertionError):
assert_max_length(self.user, 'fav_numbers', 7)
with pytest.raises(AssertionError):
assert_max_length(self.user, 'fav_numbers', 7)
def test_bigger_than_max_length(self):
with pytest.raises(AssertionError):
assert_max_length(self.user, 'fav_numbers', 9)
with pytest.raises(AssertionError):
assert_max_length(self.user, 'fav_numbers', 9)
class TestAssertNonNullable(AssertionTestCase):
def test_non_nullable_column(self):
# Test everything twice so that session gets rolled back properly
assert_non_nullable(self.user, 'age')
assert_non_nullable(self.user, 'age')
def test_nullable_column(self):
with pytest.raises(AssertionError):
assert_non_nullable(self.user, 'name')
with pytest.raises(AssertionError):
assert_non_nullable(self.user, 'name')
class TestAssertNullable(AssertionTestCase):
def test_nullable_column(self):
assert_nullable(self.user, 'name')
assert_nullable(self.user, 'name')
def test_non_nullable_column(self):
with pytest.raises(AssertionError):
assert_nullable(self.user, 'age')
with pytest.raises(AssertionError):
assert_nullable(self.user, 'age')
class TestAssertMaxLength(AssertionTestCase):
def test_with_max_length(self):
assert_max_length(self.user, 'name', 20)
assert_max_length(self.user, 'name', 20)
def test_with_non_nullable_column(self):
assert_max_length(self.user, 'email', 200)
assert_max_length(self.user, 'email', 200)
def test_smaller_than_max_length(self):
with pytest.raises(AssertionError):
assert_max_length(self.user, 'name', 19)
with pytest.raises(AssertionError):
assert_max_length(self.user, 'name', 19)
def test_bigger_than_max_length(self):
with pytest.raises(AssertionError):
assert_max_length(self.user, 'name', 21)
with pytest.raises(AssertionError):
assert_max_length(self.user, 'name', 21)
class TestAssertMinValue(AssertionTestCase):
def test_with_min_value(self):
assert_min_value(self.user, 'age', 0)
assert_min_value(self.user, 'age', 0)
def test_smaller_than_min_value(self):
with pytest.raises(AssertionError):
assert_min_value(self.user, 'age', -1)
with pytest.raises(AssertionError):
assert_min_value(self.user, 'age', -1)
def test_bigger_than_min_value(self):
with pytest.raises(AssertionError):
assert_min_value(self.user, 'age', 1)
with pytest.raises(AssertionError):
assert_min_value(self.user, 'age', 1)
class TestAssertMaxValue(AssertionTestCase):
def test_with_min_value(self):
assert_max_value(self.user, 'age', 150)
assert_max_value(self.user, 'age', 150)
def test_smaller_than_max_value(self):
with pytest.raises(AssertionError):
assert_max_value(self.user, 'age', 149)
with pytest.raises(AssertionError):
assert_max_value(self.user, 'age', 149)
def test_bigger_than_max_value(self):
with pytest.raises(AssertionError):
assert_max_value(self.user, 'age', 151)
with pytest.raises(AssertionError):
assert_max_value(self.user, 'age', 151)
|
