mirror of
https://github.com/codex-storage/deluge.git
synced 2025-02-22 08:08:17 +00:00
cae8a18437
Add __str__ to WrappedException so that the stacktrace is printed when a unit test raises a WrappedException. Change the log output from error to warning in DelugeRPCProtocol.dispatch when sending back a raised exception on an RPC request.
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
import warnings
|
|
|
|
from twisted.internet.defer import maybeDeferred
|
|
from twisted.trial import unittest
|
|
|
|
import deluge.component as component
|
|
|
|
|
|
class BaseTestCase(unittest.TestCase):
|
|
"""This is the base class that should be used for all test classes
|
|
that create classes that inherit from deluge.component.Component. It
|
|
ensures that the component registry has been cleaned up when tests
|
|
have finished.
|
|
|
|
"""
|
|
def setUp(self): # NOQA
|
|
|
|
if len(component._ComponentRegistry.components) != 0:
|
|
warnings.warn("The component._ComponentRegistry.components is not empty on test setup.\n"
|
|
"This is probably caused by another test that didn't clean up after finishing!: %s" %
|
|
component._ComponentRegistry.components)
|
|
d = maybeDeferred(self.set_up)
|
|
|
|
def on_setup_error(error):
|
|
warnings.warn("Error caught in test setup!\n%s" % error.getTraceback())
|
|
self.fail()
|
|
|
|
return d.addErrback(on_setup_error)
|
|
|
|
def tearDown(self): # NOQA
|
|
d = maybeDeferred(self.tear_down)
|
|
|
|
def on_teardown_failed(error):
|
|
warnings.warn("Error caught in test teardown!\n%s" % error.getTraceback())
|
|
self.fail()
|
|
|
|
def on_teardown_complete(result):
|
|
component._ComponentRegistry.components.clear()
|
|
component._ComponentRegistry.dependents.clear()
|
|
|
|
return d.addCallbacks(on_teardown_complete, on_teardown_failed)
|
|
|
|
def set_up(self):
|
|
pass
|
|
|
|
def tear_down(self):
|
|
pass
|