tests/test_alertmanager.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

from twisted.trial import unittest

import common

from deluge.core.alertmanager import AlertManager
from deluge.core.core import Core

class AlertManagerTestCase(unittest.TestCase):
    def setUp(self):
        self.core = Core()

        self.am = AlertManager()
        self.am.start()

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