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
|
from datetime import datetime, timedelta
import pytest
import sqlalchemy as sa
from sqlalchemy_utils import DateRangeType
from sqlalchemy_utils.compat import _select_args
intervals = None
inf = 0
try:
import intervals
from infinity import inf
except ImportError:
pass
@pytest.fixture
def Booking(Base):
class Booking(Base):
__tablename__ = 'booking'
id = sa.Column(sa.Integer, primary_key=True)
during = sa.Column(DateRangeType)
return Booking
@pytest.fixture
def create_booking(session, Booking):
def create_booking(date_range):
booking = Booking(
during=date_range
)
session.add(booking)
session.commit()
return session.query(Booking).first()
return create_booking
@pytest.fixture
def init_models(Booking):
pass
@pytest.mark.skipif('intervals is None')
class DateRangeTestCase:
def test_nullify_range(self, create_booking):
booking = create_booking(None)
assert booking.during is None
@pytest.mark.parametrize(
('date_range'),
(
[datetime(2015, 1, 1).date(), datetime(2015, 1, 3).date()],
[datetime(2015, 1, 1).date(), inf],
[-inf, datetime(2015, 1, 1).date()]
)
)
def test_save_date_range(self, create_booking, date_range):
booking = create_booking(date_range)
assert booking.during.lower == date_range[0]
assert booking.during.upper == date_range[1]
def test_nullify_date_range(self, session, Booking):
booking = Booking(
during=intervals.DateInterval(
[datetime(2015, 1, 1).date(), datetime(2015, 1, 3).date()]
)
)
session.add(booking)
session.commit()
booking = session.query(Booking).first()
booking.during = None
session.commit()
booking = session.query(Booking).first()
assert booking.during is None
def test_integer_coercion(self, Booking):
booking = Booking(during=datetime(2015, 1, 1).date())
assert booking.during.lower == datetime(2015, 1, 1).date()
assert booking.during.upper == datetime(2015, 1, 1).date()
def test_compilation(self, session, Booking):
query = sa.select(*_select_args(Booking.during))
# the type should be cacheable and not throw exception
session.execute(query)
@pytest.mark.usefixtures('postgresql_dsn')
class TestDateRangeOnPostgres(DateRangeTestCase):
@pytest.mark.parametrize(
('date_range', 'length'),
(
(
[datetime(2015, 1, 1).date(), datetime(2015, 1, 3).date()],
timedelta(days=2)
),
(
[datetime(2015, 1, 1).date(), datetime(2015, 1, 1).date()],
timedelta(days=0)
),
([-inf, datetime(2015, 1, 1).date()], None),
([datetime(2015, 1, 1).date(), inf], None),
)
)
def test_length(
self,
session,
Booking,
create_booking,
date_range,
length
):
create_booking(date_range)
query = (
session.query(Booking.during.length)
)
assert query.scalar() == length
def test_literal_param(self, session, Booking):
clause = Booking.during == [
datetime(2015, 1, 1).date(),
datetime(2015, 1, 3).date()
]
compiled = str(clause.compile(compile_kwargs={'literal_binds': True}))
assert compiled == "booking.during = '[2015-01-01, 2015-01-03]'"
|
