psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-03-05 16:35:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Keep into account psycopg build in lo64 tests

Browse Source
This commit is contained in:
Daniele Varrazzo 2014-09-11 01:38:45 +01:00
parent 2f862972c9
commit bc5e2aeead
1 changed files with 23 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
tests/test_lobject.py
View File

@ -440,13 +440,23 @@ decorate_all_tests(LargeObjectTruncateTests,
skip_if_no_lo, skip_lo_if_green, skip_if_no_truncate) skip_if_no_lo, skip_lo_if_green, skip_if_no_truncate)
def _has_lo64(conn):
"""Return (bool, msg) about the lo64 support"""
if conn.server_version < 90300:
return (False, "server version %s doesn'ts support the lo64 API"
% conn.server_version)
if 'lo64' not in psycopg2.__version__:
return (False, "this psycopg build doesn't support the lo64 API")
return (True, "this server and build support the lo64 API")
def skip_if_no_lo64(f): def skip_if_no_lo64(f):
@wraps(f) @wraps(f)
def skip_if_no_lo64_(self): def skip_if_no_lo64_(self):
if self.conn.server_version < 90300: lo64, msg = _has_lo64(self.conn)
return self.skipTest("large objects 64bit only supported from PG 9.3") if not lo64: return self.skipTest(msg)
else: else: return f(self)
return f(self)
return skip_if_no_lo64_ return skip_if_no_lo64_
@ -467,10 +477,9 @@ decorate_all_tests(LargeObject64Tests,
def skip_if_lo64(f): def skip_if_lo64(f):
@wraps(f) @wraps(f)
def skip_if_lo64_(self): def skip_if_lo64_(self):
if self.conn.server_version >= 90300: lo64, msg = _has_lo64(self.conn)
return self.skipTest("large objects 64bit only supported from PG 9.3") if lo64: return self.skipTest(msg)
else: else: return f(self)
return f(self)
return skip_if_lo64_ return skip_if_lo64_
@ -478,12 +487,16 @@ class LargeObjectNot64Tests(LargeObjectTestCase):
def test_seek_larger_than_2gb(self): def test_seek_larger_than_2gb(self):
lo = self.conn.lobject() lo = self.conn.lobject()
offset = 1 << 32 # 4gb offset = 1 << 32 # 4gb
self.assertRaises(psycopg2.InterfaceError, lo.seek, offset, 0) self.assertRaises(
(psycopg2.InterfaceError, psycopg2.NotSupportedError),
lo.seek, offset, 0)
def test_truncate_larger_than_2gb(self): def test_truncate_larger_than_2gb(self):
lo = self.conn.lobject() lo = self.conn.lobject()
length = 1 << 32 # 4gb length = 1 << 32 # 4gb
self.assertRaises(psycopg2.InterfaceError, lo.truncate, length) self.assertRaises(
(psycopg2.InterfaceError, psycopg2.NotSupportedError),
lo.truncate, length)
decorate_all_tests(LargeObjectNot64Tests, decorate_all_tests(LargeObjectNot64Tests,
skip_if_no_lo, skip_lo_if_green, skip_if_no_truncate, skip_if_lo64) skip_if_no_lo, skip_lo_if_green, skip_if_no_truncate, skip_if_lo64)