2015-02-25 21:28:34 -07:00

77 lines
2.1 KiB
Python

"""Runs all defined unit tests."""
import cx_Oracle
import imp
import os
import sys
import unittest
inSetup = (os.path.basename(sys.argv[0]).lower() == "setup.py")
print("Running tests for cx_Oracle version", cx_Oracle.version,
cx_Oracle.buildtime)
import TestEnv
if len(sys.argv) > 1 and not inSetup:
moduleNames = [os.path.splitext(v)[0] for v in sys.argv[1:]]
else:
moduleNames = [
"Connection",
"Cursor",
"CursorVar",
"DateTimeVar",
"LobVar",
"LongVar",
"3kNumberVar",
"ObjectVar",
"SessionPool",
"3kStringVar",
"TimestampVar"
]
class BaseTestCase(unittest.TestCase):
def setUp(self):
import cx_Oracle
import TestEnv
self.connection = cx_Oracle.connect(TestEnv.USERNAME,
TestEnv.PASSWORD, TestEnv.TNSENTRY)
self.cursor = self.connection.cursor()
self.cursor.arraysize = TestEnv.ARRAY_SIZE
def tearDown(self):
del self.cursor
del self.connection
loader = unittest.TestLoader()
runner = unittest.TextTestRunner(verbosity = 2)
failures = []
for name in moduleNames:
fileName = name + ".py"
print()
print("Running tests in", fileName)
if inSetup:
fileName = os.path.join("test", fileName)
module = imp.new_module(name)
import cx_Oracle
setattr(module, "USERNAME", TestEnv.USERNAME)
setattr(module, "PASSWORD", TestEnv.PASSWORD)
setattr(module, "TNSENTRY", TestEnv.TNSENTRY)
setattr(module, "ARRAY_SIZE", TestEnv.ARRAY_SIZE)
setattr(module, "TestCase", unittest.TestCase)
setattr(module, "BaseTestCase", BaseTestCase)
setattr(module, "cx_Oracle", cx_Oracle)
exec(open(fileName).read(), module.__dict__)
tests = loader.loadTestsFromModule(module)
result = runner.run(tests)
if not result.wasSuccessful():
failures.append(name)
if failures:
print("***** Some tests in the following modules failed. *****")
for name in failures:
print(" %s" % name)
sys.exit(1)