diff options
| -rwxr-xr-x | tests/test_connection.py | 21 | ||||
| -rwxr-xr-x | tests/test_cursor.py | 16 |
2 files changed, 23 insertions, 14 deletions
diff --git a/tests/test_connection.py b/tests/test_connection.py index b4422c6..b562707 100755 --- a/tests/test_connection.py +++ b/tests/test_connection.py @@ -350,14 +350,19 @@ class ConnectionTests(ConnectingTestCase): conn = self.connect() self.assert_(conn.pgconn_ptr is not None) - f = self.libpq.PQserverVersion - f.argtypes = [ctypes.c_void_p] - f.restype = ctypes.c_int - ver = f(conn.pgconn_ptr) - if ver == 0 and sys.platform == 'darwin': - return self.skipTest("I don't know why this func returns 0 on OSX") - - self.assertEqual(ver, conn.server_version) + try: + f = self.libpq.PQserverVersion + except AttributeError: + pass + else: + f.argtypes = [ctypes.c_void_p] + f.restype = ctypes.c_int + ver = f(conn.pgconn_ptr) + if ver == 0 and sys.platform == 'darwin': + return self.skipTest( + "I don't know why this func returns 0 on OSX") + + self.assertEqual(ver, conn.server_version) conn.close() self.assert_(conn.pgconn_ptr is None) diff --git a/tests/test_cursor.py b/tests/test_cursor.py index 47b41e6..ce1b0cb 100755 --- a/tests/test_cursor.py +++ b/tests/test_cursor.py @@ -654,14 +654,18 @@ class CursorTests(ConnectingTestCase): curs = self.conn.cursor() self.assert_(curs.pgresult_ptr is None) - f = self.libpq.PQcmdStatus - f.argtypes = [ctypes.c_void_p] - f.restype = ctypes.c_char_p - curs.execute("select 'x'") self.assert_(curs.pgresult_ptr is not None) - status = f(curs.pgresult_ptr) - self.assertEqual(status, b'SELECT 1') + + try: + f = self.libpq.PQcmdStatus + except AttributeError: + pass + else: + f.argtypes = [ctypes.c_void_p] + f.restype = ctypes.c_char_p + status = f(curs.pgresult_ptr) + self.assertEqual(status, b'SELECT 1') curs.close() self.assert_(curs.pgresult_ptr is None) |