mirror of
https://github.com/psycopg/psycopg2.git
synced 2024-11-23 01:16:34 +03:00
fcc083dc12
There is no need to import testutils.unittest instead of simply unittest. They are simple aliases. Use system unittest to be more regular, consistent as well as idiomatic with the wider Python community.
107 lines
3.4 KiB
Python
Executable File
107 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# psycopg2 test suite
|
|
#
|
|
# Copyright (C) 2007-2011 Federico Di Gregorio <fog@debian.org>
|
|
#
|
|
# psycopg2 is free software: you can redistribute it and/or modify it
|
|
# under the terms of the GNU Lesser General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# In addition, as a special exception, the copyright holders give
|
|
# permission to link this program with the OpenSSL library (or with
|
|
# modified versions of OpenSSL that use the same license as OpenSSL),
|
|
# and distribute linked combinations including the two.
|
|
#
|
|
# You must obey the GNU Lesser General Public License in all respects for
|
|
# all of the code used other than OpenSSL.
|
|
#
|
|
# psycopg2 is distributed in the hope that it will be useful, but WITHOUT
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
|
|
# License for more details.
|
|
|
|
# Convert warnings into errors here. We can't do it with -W because on
|
|
# Travis importing site raises a warning.
|
|
import warnings
|
|
warnings.simplefilter('error') # noqa
|
|
|
|
import sys
|
|
from testconfig import dsn
|
|
import unittest
|
|
|
|
import test_async
|
|
import test_bugX000
|
|
import test_bug_gc
|
|
import test_cancel
|
|
import test_connection
|
|
import test_copy
|
|
import test_cursor
|
|
import test_dates
|
|
import test_errcodes
|
|
import test_extras_dictcursor
|
|
import test_fast_executemany
|
|
import test_green
|
|
import test_ipaddress
|
|
import test_lobject
|
|
import test_module
|
|
import test_notify
|
|
import test_psycopg2_dbapi20
|
|
import test_quote
|
|
import test_replication
|
|
import test_sql
|
|
import test_transaction
|
|
import test_types_basic
|
|
import test_types_extras
|
|
import test_with
|
|
|
|
if sys.version_info[:2] < (3, 6):
|
|
import test_async_keyword
|
|
|
|
|
|
def test_suite():
|
|
# If connection to test db fails, bail out early.
|
|
import psycopg2
|
|
try:
|
|
cnn = psycopg2.connect(dsn)
|
|
except Exception as e:
|
|
print "Failed connection to test db:", e.__class__.__name__, e
|
|
print "Please set env vars 'PSYCOPG2_TESTDB*' to valid values."
|
|
sys.exit(1)
|
|
else:
|
|
cnn.close()
|
|
|
|
suite = unittest.TestSuite()
|
|
suite.addTest(test_async.test_suite())
|
|
if sys.version_info[:2] < (3, 6):
|
|
suite.addTest(test_async_keyword.test_suite())
|
|
suite.addTest(test_bugX000.test_suite())
|
|
suite.addTest(test_bug_gc.test_suite())
|
|
suite.addTest(test_cancel.test_suite())
|
|
suite.addTest(test_connection.test_suite())
|
|
suite.addTest(test_copy.test_suite())
|
|
suite.addTest(test_cursor.test_suite())
|
|
suite.addTest(test_dates.test_suite())
|
|
suite.addTest(test_errcodes.test_suite())
|
|
suite.addTest(test_extras_dictcursor.test_suite())
|
|
suite.addTest(test_fast_executemany.test_suite())
|
|
suite.addTest(test_green.test_suite())
|
|
suite.addTest(test_ipaddress.test_suite())
|
|
suite.addTest(test_lobject.test_suite())
|
|
suite.addTest(test_module.test_suite())
|
|
suite.addTest(test_notify.test_suite())
|
|
suite.addTest(test_psycopg2_dbapi20.test_suite())
|
|
suite.addTest(test_quote.test_suite())
|
|
suite.addTest(test_replication.test_suite())
|
|
suite.addTest(test_sql.test_suite())
|
|
suite.addTest(test_transaction.test_suite())
|
|
suite.addTest(test_types_basic.test_suite())
|
|
suite.addTest(test_types_extras.test_suite())
|
|
suite.addTest(test_with.test_suite())
|
|
return suite
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='test_suite')
|