def test_register_retrieve_driver(self): """Register driver and retrieve it again""" driver = self.register_empty_driver("some_type", "some_name") self.assertTrue(drivers.get_driver("some_type", "some_name") is driver)
def test_get_driver_default(self): user = utils.get_test_user() expected_driver = drivers.get_driver('messaging', 'fake') actual_driver = messaging._get_driver(user) self.assertEquals(actual_driver, expected_driver)
def setUp(self): import surveilr.messaging.fake # This does nothing, but pyflakes gets upset if we import it # and never "use" it. surveilr.messaging.fake self.driver = drivers.get_driver('messaging', 'fake')
def test_get_driver_sms(self): user = utils.get_test_user(messaging_driver='sms') expected_driver = drivers.get_driver('messaging', 'sms') actual_driver = messaging._get_driver(user) self.assertEquals(actual_driver, expected_driver)
def _get_driver(recipient): return drivers.get_driver('messaging', getattr(recipient, 'messaging_driver', 'fake'))
def __init__(self): driver_name = config.get_str('sms', 'driver') self.driver = drivers.get_driver('sms', driver_name)