def test_add_watcher_same_lowercase_names(self): controller = "tcp://127.0.0.1:%d" % get_available_port() sub = "tcp://127.0.0.1:%d" % get_available_port() arbiter = Arbiter([], controller, sub, loop=get_ioloop(), check_delay=-1) arbiter.add_watcher('foo', SLEEP % 5) self.assertRaises(AlreadyExist, arbiter.add_watcher, 'Foo', SLEEP % 5)
def test_add_watcher(self): controller = "tcp://127.0.0.1:%d" % get_available_port() sub = "tcp://127.0.0.1:%d" % get_available_port() arbiter = Arbiter([], controller, sub, loop=get_ioloop(), check_delay=-1) arbiter.add_watcher('foo', SLEEP % 5) try: yield arbiter.start() self.assertEqual(arbiter.watchers[0].status(), 'active') finally: yield arbiter.stop()
def test_circushttpd(self): controller = "tcp://127.0.0.1:%d" % get_available_port() sub = "tcp://127.0.0.1:%d" % get_available_port() arbiter = Arbiter([], controller, sub, loop=get_ioloop(), check_delay=-1, httpd=True, debug=True) self.arbiters.append(arbiter) try: yield arbiter.start() poll_for_callable(self.assertDictEqual, arbiter.statuses, {'circushttpd': 'active'}) finally: yield arbiter.stop()
def test_start_with_callback(self): controller = "tcp://127.0.0.1:%d" % get_available_port() sub = "tcp://127.0.0.1:%d" % get_available_port() arbiter = Arbiter([], controller, sub, check_delay=-1) callee = mock.MagicMock() def callback(*args): callee() arbiter.stop() arbiter.start(cb=callback) self.assertEqual(callee.call_count, 1)
def test_start_with_callback_and_given_loop(self): controller = "tcp://127.0.0.1:%d" % get_available_port() sub = "tcp://127.0.0.1:%d" % get_available_port() arbiter = Arbiter([], controller, sub, check_delay=-1, loop=get_ioloop()) callback = mock.MagicMock() try: yield arbiter.start(cb=callback) finally: yield arbiter.stop() self.assertEqual(callback.call_count, 0)
def test_ioloop_has_handle_callback_exception(self): controller = "tcp://127.0.0.1:%d" % get_available_port() sub = "tcp://127.0.0.1:%d" % get_available_port() arbiter = Arbiter([], controller, sub, loop=get_ioloop(), check_delay=-1) self.assertTrue(hasattr(arbiter.loop, 'handle_callback_exception'))