def __init__(self, methodName='runTest', webdriver_provider=None, screenshot_util=None):
super(WTFBaseTest, self).__init__(methodName)
self._register_watcher(CaptureScreenShotOnErrorTestWatcher(webdriver_provider, screenshot_util))
# Note this watcher should be registered after all other watchers that use
# on_test_passed() event.
self._delayed_test_watcher = DelayedTestFailTestWatcher()
self._register_watcher(self._delayed_test_watcher)
def __init__(self,
methodName='runTest',
webdriver_provider=None,
screenshot_util=None):
"""
Constructor matches that of UnitTest2 test case, but modified to allow passing in
a ScreenShot utility and register delayed test watchers.
Kwargs:
methodName (str) : Test method name.
webdriver_provider (WebdriverManager) : Default webdriver provider.
screenshot_util (CaptureScreenShotOnErrorTestWatcher) : Screenshot capture utility.
"""
super(WTFBaseTest, self).__init__(methodName)
self._register_watcher(
CaptureScreenShotOnErrorTestWatcher(webdriver_provider,
screenshot_util))
# Note this watcher should be registered after all other watchers that use
# on_test_passed() event.
self._delayed_test_watcher = DelayedTestFailTestWatcher()
self._register_watcher(self._delayed_test_watcher)
def __init__(self, methodName='runTest', webdriver_provider=None, screenshot_util=None):
"""
Constructor matches that of UnitTest2 test case, but modified to allow passing in
a ScreenShot utility and register delayed test watchers.
Kwargs:
methodName (str) : Test method name.
webdriver_provider (WebdriverManager) : Default webdriver provider.
screenshot_util (CaptureScreenShotOnErrorTestWatcher) : Screenshot capture utility.
"""
super(WTFBaseTest, self).__init__(methodName)
self._register_watcher(CaptureScreenShotOnErrorTestWatcher(webdriver_provider, screenshot_util))
# Note this watcher should be registered after all other watchers that use
# on_test_passed() event.
self._delayed_test_watcher = DelayedTestFailTestWatcher()
self._register_watcher(self._delayed_test_watcher)
class WTFBaseTest(WatchedTestCase):
'''
Test Cases can extend this test to additional unit test functionality such as
take screenshot on failure.
'''
def __init__(self, methodName='runTest', webdriver_provider=None, screenshot_util=None):
super(WTFBaseTest, self).__init__(methodName)
self._register_watcher(CaptureScreenShotOnErrorTestWatcher(webdriver_provider, screenshot_util))
# Note this watcher should be registered after all other watchers that use
# on_test_passed() event.
self._delayed_test_watcher = DelayedTestFailTestWatcher()
self._register_watcher(self._delayed_test_watcher)
def assertWithDelayedFailure(self, assert_method, *args, **kwargs):
"""
Cause an assertion failure to be delayed till the end of the test.
Usage:
self.assertWithDelayedFailure(self.AssertEquals, 100, percent)
@param assert_method: Reference to assert method.
@param *params: parameters to pass into assert method.
"""
frame = None
try:
#attempt to get parent frames
frame = inspect.getouterframes(inspect.currentframe())[1]
except:
pass #oh well, we couldn't get it.
assert_func = lambda: assert_method(*args, **kwargs)
generated_exception = self._delayed_test_watcher.delay_failure(assert_func, frame)
if generated_exception:
# Call our on_fail for our test watchers. So we can trigger our screen
# capture at moment of failure.
for test_watcher in self.__wtf_test_watchers__:
test_watcher.on_test_failure(self, self._resultForDoCleanups, generated_exception)
class WTFBaseTest(WatchedTestCase):
'''
Test can extend this basetest to add additional unit test functionality such as
take screenshot on failure.
Example::
from wtframework.wtf.testobjects.basetests import WTFBaseTest
from wtframework.wtf.web.webdriver import WTF_WEBDRIVER_MANAGER
class TestScreenCaptureOnFail(WTFBaseTest):
""""
These test cases are expected to fail. They are here to test
the screen capture on failure.
"""
# Comment out decorator to manually test the screen capture.
@unittest.expectedFailure
def test_fail(self):
driver = WTF_WEBDRIVER_MANAGER.new_driver()
driver.get('http://www.google.com')
self.fail()
#Check your /screenshots folder for a screenshot of Google Homepage.
For the screen capture to work, you need to make sure you use WTF_WEBDRIVER_MANAGER for
getting your webdriver instance. This is used for getting the current instance of webdriver
when a test fails in order to take a screenshot.
WTFBaseTest is also an instance of WatchedTestCase, which you can use to add additional
call backs you wish to use for handling errors or other test events.
'''
def __init__(self,
methodName='runTest',
webdriver_provider=None,
screenshot_util=None):
"""
Constructor matches that of UnitTest2 test case, but modified to allow passing in
a ScreenShot utility and register delayed test watchers.
Kwargs:
methodName (str) : Test method name.
webdriver_provider (WebdriverManager) : Default webdriver provider.
screenshot_util (CaptureScreenShotOnErrorTestWatcher) : Screenshot capture utility.
"""
super(WTFBaseTest, self).__init__(methodName)
self._register_watcher(
CaptureScreenShotOnErrorTestWatcher(webdriver_provider,
screenshot_util))
# Note this watcher should be registered after all other watchers that use
# on_test_passed() event.
self._delayed_test_watcher = DelayedTestFailTestWatcher()
self._register_watcher(self._delayed_test_watcher)
def assertWithDelayedFailure(self, assert_method, *args, **kwargs):
"""
Cause an assertion failure to be delayed till the end of the test.
This is good to use if you want the test to continue after an assertion
fails, and do addtional assertions. At the end of the test, it will
pool all the test failures into 1 failed assert with a summary of
all the test failures that occurred during the test.
Args:
assert_method (function) - Assert method to run.
args - arguments to pass into the assert method.
Kwargs:
kwargs - additional kwargs to pass into the assert method.
Will assert if percent == 100 at the end of the test.::
self.assertWithDelayedFailure(self.AssertEquals, 100, percent)
"""
frame = None
try:
# attempt to get parent frames
frame = inspect.getouterframes(inspect.currentframe())[1]
except:
pass # oh well, we couldn't get it.
assert_func = lambda: assert_method(*args, **kwargs)
generated_exception = self._delayed_test_watcher.delay_failure(
assert_func, frame)
if generated_exception:
# Call our on_fail for our test watchers. So we can trigger our screen
# capture at moment of failure.
for test_watcher in self.__wtf_test_watchers__:
test_watcher.on_test_failure(self, self._resultForDoCleanups,
generated_exception)
class WTFBaseTest(WatchedTestCase):
'''
Test can extend this basetest to add additional unit test functionality such as
take screenshot on failure.
Example::
from wtframework.wtf.testobjects.basetests import WTFBaseTest
from wtframework.wtf.web.webdriver import WTF_WEBDRIVER_MANAGER
class TestScreenCaptureOnFail(WTFBaseTest):
""""
These test cases are expected to fail. They are here to test
the screen capture on failure.
"""
# Comment out decorator to manually test the screen capture.
@unittest.expectedFailure
def test_fail(self):
driver = WTF_WEBDRIVER_MANAGER.new_driver()
driver.get('http://www.google.com')
self.fail()
#Check your /screenshots folder for a screenshot of Google Homepage.
For the screen capture to work, you need to make sure you use WTF_WEBDRIVER_MANAGER for
getting your webdriver instance. This is used for getting the current instance of webdriver
when a test fails in order to take a screenshot.
WTFBaseTest is also an instance of WatchedTestCase, which you can use to add additional
call backs you wish to use for handling errors or other test events.
'''
def __init__(self, methodName='runTest', webdriver_provider=None, screenshot_util=None):
"""
Constructor matches that of UnitTest2 test case, but modified to allow passing in
a ScreenShot utility and register delayed test watchers.
Kwargs:
methodName (str) : Test method name.
webdriver_provider (WebdriverManager) : Default webdriver provider.
screenshot_util (CaptureScreenShotOnErrorTestWatcher) : Screenshot capture utility.
"""
super(WTFBaseTest, self).__init__(methodName)
self._register_watcher(CaptureScreenShotOnErrorTestWatcher(webdriver_provider, screenshot_util))
# Note this watcher should be registered after all other watchers that use
# on_test_passed() event.
self._delayed_test_watcher = DelayedTestFailTestWatcher()
self._register_watcher(self._delayed_test_watcher)
def assertWithDelayedFailure(self, assert_method, *args, **kwargs):
"""
Cause an assertion failure to be delayed till the end of the test.
This is good to use if you want the test to continue after an assertion
fails, and do addtional assertions. At the end of the test, it will
pool all the test failures into 1 failed assert with a summary of
all the test failures that occurred during the test.
Args:
assert_method (function) - Assert method to run.
args - arguments to pass into the assert method.
Kwargs:
kwargs - additional kwargs to pass into the assert method.
Will assert if percent == 100 at the end of the test.::
self.assertWithDelayedFailure(self.AssertEquals, 100, percent)
"""
frame = None
try:
#attempt to get parent frames
frame = inspect.getouterframes(inspect.currentframe())[1]
except:
pass #oh well, we couldn't get it.
assert_func = lambda: assert_method(*args, **kwargs)
generated_exception = self._delayed_test_watcher.delay_failure(assert_func, frame)
if generated_exception:
# Call our on_fail for our test watchers. So we can trigger our screen
# capture at moment of failure.
for test_watcher in self.__wtf_test_watchers__:
test_watcher.on_test_failure(self, self._resultForDoCleanups, generated_exception)
