39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
from twisted.trial import unittest
|
|
|
|
import common
|
|
|
|
from deluge.core.alertmanager import AlertManager
|
|
from deluge.core.core import Core
|
|
import deluge.component as component
|
|
|
|
class AlertManagerTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.core = Core()
|
|
|
|
self.am = component.get("AlertManager")
|
|
component.start(["AlertManager"])
|
|
|
|
def tearDown(self):
|
|
def on_shutdown(result):
|
|
component._ComponentRegistry.components = {}
|
|
del self.am
|
|
del self.core
|
|
|
|
return component.shutdown().addCallback(on_shutdown)
|
|
|
|
def test_register_handler(self):
|
|
def handler(alert):
|
|
return
|
|
|
|
self.am.register_handler("dummy_alert", handler)
|
|
|
|
self.assertEquals(self.am.handlers["dummy_alert"], [handler])
|
|
|
|
def test_deregister_handler(self):
|
|
def handler(alert):
|
|
return
|
|
|
|
self.am.register_handler("dummy_alert", handler)
|
|
self.am.deregister_handler(handler)
|
|
self.assertEquals(self.am.handlers["dummy_alert"], [])
|