mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
75 lines
3.2 KiB
Python
75 lines
3.2 KiB
Python
# Copyright 2012 Citrix Systems, Inc. Licensed under the
|
|
# Apache License, Version 2.0 (the "License"); you may not use this
|
|
# file except in compliance with the License. Citrix Systems, Inc.
|
|
# reserves all rights not expressly granted by the License.
|
|
# You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
# Automatically generated by addcopyright.py at 04/03/2012
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
|
|
from functools import partial
|
|
import os
|
|
import sys
|
|
import logging
|
|
|
|
def testCaseLogger(message, logger=None):
|
|
if logger is not None:
|
|
logger.debug(message)
|
|
|
|
class TestCaseExecuteEngine(object):
|
|
def __init__(self, testclient, testCaseFolder, testcaseLogFile=None, testResultLogFile=None):
|
|
self.testclient = testclient
|
|
self.testCaseFolder = testCaseFolder
|
|
self.logformat = logging.Formatter("%(asctime)s - %(levelname)s - %(name)s - %(message)s")
|
|
|
|
if testcaseLogFile is not None:
|
|
self.logfile = testcaseLogFile
|
|
self.logger = logging.getLogger("TestCaseExecuteEngine")
|
|
fh = logging.FileHandler(self.logfile)
|
|
fh.setFormatter(self.logformat)
|
|
self.logger.addHandler(fh)
|
|
self.logger.setLevel(logging.DEBUG)
|
|
if testResultLogFile is not None:
|
|
ch = logging.StreamHandler()
|
|
ch.setLevel(logging.ERROR)
|
|
ch.setFormatter(self.logformat)
|
|
self.logger.addHandler(ch)
|
|
fp = open(testResultLogFile, "w")
|
|
self.testResultLogFile = fp
|
|
else:
|
|
self.testResultLogFile = sys.stdout
|
|
|
|
def injectTestCase(self, testSuites):
|
|
for test in testSuites:
|
|
if isinstance(test, unittest.BaseTestSuite):
|
|
self.injectTestCase(test)
|
|
else:
|
|
#logger bears the name of the test class
|
|
testcaselogger = logging.getLogger("testclient.testcase.%s"%test.__class__.__name__)
|
|
fh = logging.FileHandler(self.logfile)
|
|
fh.setFormatter(self.logformat)
|
|
testcaselogger.addHandler(fh)
|
|
testcaselogger.setLevel(logging.DEBUG)
|
|
|
|
#inject testclient and logger into each unittest
|
|
setattr(test, "testClient", self.testclient)
|
|
setattr(test, "debug", partial(testCaseLogger, logger=testcaselogger))
|
|
setattr(test.__class__, "clstestclient", self.testclient)
|
|
if hasattr(test, "UserName"):
|
|
self.testclient.createNewApiClient(test.UserName, test.DomainName, test.AcctType)
|
|
|
|
def run(self):
|
|
loader = unittest.loader.TestLoader()
|
|
suite = loader.discover(self.testCaseFolder)
|
|
self.injectTestCase(suite)
|
|
|
|
unittest.TextTestRunner(stream=self.testResultLogFile, verbosity=2).run(suite)
|