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
|
"""
Test parameters substitution.
"""
import datetime
import decimal
import pytest
from pymssql._mssql import quote_simple_value, datetime2
def test_none():
res = quote_simple_value(None)
assert res == b'NULL'
@pytest.mark.parametrize('val', (0, 13))
def test_int(val):
res = quote_simple_value(val)
assert res == f"{val}".encode('utf-8')
@pytest.mark.parametrize('val', (0.0, 3.14))
def test_float(val):
res = quote_simple_value(val)
assert res == f"{val}".encode('utf-8')
@pytest.mark.parametrize('val', (True, False))
def test_bool(val):
res = quote_simple_value(val)
assert res == f"{int(val)}".encode('utf-8')
@pytest.mark.parametrize('val', (0.123, 13.789))
def test_decimal(val):
res = quote_simple_value(decimal.Decimal(val))
assert res == f"{decimal.Decimal(val)}".encode('utf-8')
def test_uuid():
u = '3cde0c92-2674-481f-8fd4-feacf98d8504'
res = quote_simple_value(u)
assert res == f"N'{u}'".encode('utf-8')
@pytest.mark.parametrize('val', ('Фрязино', 'Здравствуй Мир'))
def test_str(val):
res = quote_simple_value(val)
assert res == f"N'{val}'".encode('utf-8')
@pytest.mark.parametrize('val', (b"Hello", b"Hello 'World!'"))
def test_bytes(val):
res = quote_simple_value(val)
assert res == b"'%s'"%(val.replace(b"'", b"''"))
@pytest.mark.parametrize('val', (
datetime.date(1, 2, 3),
datetime.date(2024, 3, 24)))
def test_date(val):
res = quote_simple_value(val)
assert res == f"'{val.year:04}-{val.month:02d}-{val.day:02d}'".encode()
@pytest.mark.parametrize('val', (
datetime.time(0, 0, 0, 1),
datetime.time(0, 0, 1),
datetime.time(1, 2, 59),
datetime.time(23, 3, 24)))
def test_time(val):
res = quote_simple_value(val)
assert res == f"'{val.hour:02d}:{val.minute:02d}:{val.second:02d}" \
f".{val.microsecond:06}'".encode()
@pytest.mark.parametrize('val', (
datetime.datetime(1, 2, 3, 23, 3, 59),
datetime.datetime(2024, 3, 24, 0, 0, 1)))
def test_datetime(val):
res = quote_simple_value(val)
assert res == f"'{val.year:04}-{val.month:02d}-{val.day:02d}" \
f" {val.hour:02d}:{val.minute:02d}:{val.second:02d}" \
f".{(val.microsecond // 1000):03d}'".encode()
@pytest.mark.parametrize('val', (
datetime.datetime(1, 2, 3, 23, 3, 59),
datetime.datetime(2024, 3, 24, 0, 0, 1)))
def test_datetime_use_datetime2(val):
res = quote_simple_value(val, use_datetime2=True)
assert res == f"'{val.year:04}-{val.month:02d}-{val.day:02d}" \
f" {val.hour:02d}:{val.minute:02d}:{val.second:02d}" \
f".{val.microsecond:06}'".encode()
@pytest.mark.parametrize('val', (
datetime2(1, 2, 3, 23, 3, 59),
datetime2(2024, 3, 24, 0, 0, 1)))
def test_datetime2(val):
res = quote_simple_value(val)
assert res == f"'{val.year:04}-{val.month:02d}-{val.day:02d}" \
f" {val.hour:02d}:{val.minute:02d}:{val.second:02d}" \
f".{val.microsecond:06}'".encode()
@pytest.mark.parametrize('val', (
datetime.datetime(1, 2, 3, 23, 3, 59,
tzinfo=datetime.timezone.utc),
datetime.datetime(2024, 3, 24, 0, 0, 1,
tzinfo=datetime.timezone.utc)))
def test_datetime_utc(val):
res = quote_simple_value(val)
assert res == f"'{val.year:04}-{val.month:02d}-{val.day:02d}" \
f" {val.hour:02d}:{val.minute:02d}:{val.second:02d}" \
f".{(val.microsecond // 1000):03d}'".encode()
@pytest.mark.parametrize('val', (
datetime.datetime(1, 2, 3, 23, 3, 59,
tzinfo=datetime.timezone(
datetime.timedelta(hours=3))),
datetime.datetime(2024, 3, 24, 0, 0, 1,
tzinfo=datetime.timezone(
datetime.timedelta(hours=3)))))
def test_datetime_tz(val):
res = quote_simple_value(val)
assert res == f"'{val.year:04}-{val.month:02d}-{val.day:02d}" \
f" {val.hour:02d}:{val.minute:02d}:{val.second:02d}" \
f".{val.microsecond:06}{val.strftime('%Z')[3:]}'".encode()
|
