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
|
#! /usr/bin/env python
import sys, time
import skytools.psycopgwrapper
import skytools._cquoting, skytools._pyquoting
# create a DictCursor row
class fake_cursor:
index = {'id': 0, 'data': 1}
description = ['x', 'x']
dbrow = skytools.psycopgwrapper._CompatRow(fake_cursor())
dbrow[0] = '123'
dbrow[1] = 'value'
def regtest(name, func, cases):
bad = 0
for dat, res in cases:
res2 = func(dat)
if res != res2:
print "failure: %s(%s) = %s (expected %s)" % (name, repr(dat), repr(res2), repr(res))
bad += 1
if bad:
print "%-20s: failed" % name
else:
print "%-20s: OK" % name
sql_literal = [
[None, "null"],
["", "''"],
["a'b", "'a''b'"],
[r"a\'b", r"E'a\\''b'"],
[1, "'1'"],
[True, "'True'"],
]
regtest("quote_literal/c", skytools._cquoting.quote_literal, sql_literal)
regtest("quote_literal/py", skytools._pyquoting.quote_literal, sql_literal)
sql_copy = [
[None, "\\N"],
["", ""],
["a'\tb", "a'\\tb"],
[r"a\'b", r"a\\'b"],
[1, "1"],
[True, "True"],
[u"qwe", "qwe"],
]
regtest("quote_copy/c", skytools._cquoting.quote_copy, sql_copy)
regtest("quote_copy/py", skytools._pyquoting.quote_copy, sql_copy)
sql_bytea_raw = [
[None, None],
["", ""],
["a'\tb", "a'\\011b"],
[r"a\'b", r"a\\'b"],
["\t\344", r"\011\344"],
]
regtest("quote_bytea_raw/c", skytools._cquoting.quote_bytea_raw, sql_bytea_raw)
regtest("quote_bytea_raw/py", skytools._pyquoting.quote_bytea_raw, sql_bytea_raw)
t_urlenc = [
[{}, ""],
[{'a': 1}, "a=1"],
[{'a': None}, "a"],
[{'qwe': 1, u'zz': u"qwe"}, "qwe=1&zz=qwe"],
[{'a': '\000%&'}, "a=%00%25%26"],
[dbrow, 'data=value&id=123'],
]
regtest("db_urlencode/c", skytools._cquoting.db_urlencode, t_urlenc)
regtest("db_urlencode/py", skytools._pyquoting.db_urlencode, t_urlenc)
t_urldec = [
["", {}],
["a=b&c", {'a': 'b', 'c': None}],
["&&b=f&&", {'b': 'f'}],
[u"abc=qwe", {'abc': 'qwe'}],
["b=", {'b': ''}],
["b=%00%45", {'b': '\x00E'}],
]
regtest("db_urldecode/c", skytools._cquoting.db_urldecode, t_urldec)
regtest("db_urldecode/py", skytools._pyquoting.db_urldecode, t_urldec)
t_unesc = [
["", ""],
["\\N", "N"],
["abc", "abc"],
[u"abc", "abc"],
[r"\0\000\001\01\1", "\0\000\001\001\001"],
[r"a\001b\tc\r\n", "a\001b\tc\r\n"],
]
regtest("unescape/c", skytools._cquoting.unescape, t_unesc)
regtest("unescape/py", skytools._pyquoting.unescape, t_unesc)
|
