Closes #55
This commit is contained in:
Jing Wang 2016-06-07 20:30:03 -07:00
Родитель 5b846ed921
Коммит 276e272821
2 изменённых файлов: 8 добавлений и 1 удалений

Просмотреть файл

@ -238,7 +238,9 @@ class ParamEscaper(object):
return "'{}'".format(item.replace("'", "''"))
def escape_item(self, item):
if isinstance(item, (int, float)):
if item is None:
return 'NULL'
elif isinstance(item, (int, float)):
return self.escape_number(item)
elif isinstance(item, basestring):
return self.escape_string(item)

Просмотреть файл

@ -45,6 +45,11 @@ class DBAPITestCase(with_metaclass(abc.ABCMeta, object)):
cursor.execute('SELECT a FROM many_rows ORDER BY a')
self.assertEqual(cursor.fetchall(), [(i,) for i in range(10000)])
@with_cursor
def test_null_param(self, cursor):
cursor.execute('SELECT %s FROM one_row', (None,))
self.assertEqual(cursor.fetchall(), [(None,)])
@with_cursor
def test_iterator(self, cursor):
cursor.execute('SELECT * FROM one_row')