diff options
Diffstat (limited to 'deluge/tests/test_alertmanager.py')
-rw-r--r-- | deluge/tests/test_alertmanager.py | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/deluge/tests/test_alertmanager.py b/deluge/tests/test_alertmanager.py new file mode 100644 index 0000000..f197882 --- /dev/null +++ b/deluge/tests/test_alertmanager.py @@ -0,0 +1,39 @@ +# -*- coding: utf-8 -*- +# +# This file is part of Deluge and is licensed under GNU General Public License 3.0, or later, with +# the additional special exception to link portions of this program with the OpenSSL library. +# See LICENSE for more details. +# + +from __future__ import unicode_literals + +import deluge.component as component +from deluge.core.core import Core + +from .basetest import BaseTestCase + + +class AlertManagerTestCase(BaseTestCase): + def set_up(self): + self.core = Core() + self.core.config.config['lsd'] = False + self.am = component.get('AlertManager') + return component.start(['AlertManager']) + + def tear_down(self): + return component.shutdown() + + def test_register_handler(self): + def handler(alert): + return + + self.am.register_handler('dummy_alert', handler) + self.assertEqual(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.assertEqual(self.am.handlers['dummy_alert'], []) |