deluge/tests/test_alertmanager.py

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"], [])