def setUp(self): """ Initialize an empty provider. """ self.provider = Provider() self.provider.server = Server()
class TestProvider(unittest.TestCase): def setUp(self): """ Initialize an empty provider. """ self.provider = Provider() self.provider.server = Server() def test_hooks(self): """ Test hooks for provider events. """ updated = Latch() session_created = Latch() session_destroyed = Latch() self.provider.hooks["updated"].append(updated) self.provider.hooks["session_created"].append(session_created) self.provider.hooks["session_destroyed"].append(session_destroyed) self.provider.update() self.assertTrue(updated.toggled) self.provider.create_session("User-Agent", "127.0.0.1", "1.0") self.assertTrue(session_created.toggled) self.assertEqual(session_created.args[0], 1) self.provider.destroy_session(1) self.assertTrue(session_destroyed.toggled) self.assertEqual(session_destroyed.args[0], 1)