def _test_watch_create_watcher(path, handler): watcher_obj = watcher.Watcher(handler) watcher_obj._running = True watcher_obj._resources.add(path) watcher_obj._idle[path] = True watcher_obj._watching[path] = None return watcher_obj
def test_remove(self, m_stop_watch): path = '/test' m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler) watcher_obj._resources.add(path) watcher_obj.remove(path) self.assertEqual(set(), watcher_obj._resources) m_stop_watch.assert_not_called()
def test_add(self, m_start_watch): paths = ['/test%s' % i for i in range(3)] m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler) for path in paths: watcher_obj.add(path) self.assertEqual(set(paths), watcher_obj._resources) m_start_watch.assert_not_called()
def test_start_watch(self, m_watch): path = '/test' m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler) watcher_obj._start_watch(path) m_watch.assert_called_once_with(path) self.assertTrue(watcher_obj._idle.get(path)) self.assertIn(path, watcher_obj._watching)
def test_stop(self, m_stop_watch): paths = ['/test%s' % i for i in range(3)] m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler) watcher_obj._resources.update(paths) watcher_obj.stop() self.assertFalse(watcher_obj._running) m_stop_watch.assert_not_called()
def test_start(self, m_start_watch): paths = ['/test%s' % i for i in range(3)] m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler) watcher_obj._resources.update(paths) watcher_obj.start() self.assertTrue(watcher_obj._running) m_start_watch.assert_has_calls([mock.call(path) for path in paths], any_order=True)
def test_remove_watching(self, m_stop_watch): path = '/test' m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler) watcher_obj._resources.add(path) m_watching = watcher_obj._watching = mock.MagicMock() m_watching.__contains__.return_value = True watcher_obj.remove(path) self.assertEqual(set(), watcher_obj._resources) m_stop_watch.assert_called_once_with(path)
def test_add_running(self, m_start_watch): paths = ['/test%s' % i for i in range(3)] m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler) watcher_obj._running = True for path in paths: watcher_obj.add(path) self.assertEqual(set(paths), watcher_obj._resources) m_start_watch.assert_has_calls([mock.call(path) for path in paths], any_order=True)
def test_stop_watch_idle(self): path = '/test' m_tg = mock.Mock() m_th = mock.Mock() m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler, m_tg) watcher_obj._idle[path] = False watcher_obj._watching[path] = m_th watcher_obj._stop_watch(path) m_th.kill.assert_not_called()
def test_stop_watch_threaded(self): path = '/test' m_tg = mock.Mock() m_th = mock.Mock() m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler, m_tg) watcher_obj._idle[path] = True watcher_obj._watching[path] = m_th watcher_obj._stop_watch(path) m_th.stop.assert_called()
def test_start_already_watching(self, m_start_watch): paths = ['/test%s' % i for i in range(3)] m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler) watcher_obj._resources.update(paths) m_watching = watcher_obj._watching = mock.MagicMock() m_watching.__iter__.return_value = paths watcher_obj.start() self.assertTrue(watcher_obj._running) m_start_watch.assert_not_called()
def test_add_watching(self, m_start_watch): paths = ['/test%s' % i for i in range(3)] m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler) watcher_obj._running = True m_watching = watcher_obj._watching = mock.MagicMock() m_watching.__contains__.return_value = True for path in paths: watcher_obj.add(path) self.assertEqual(set(paths), watcher_obj._resources) m_start_watch.assert_not_called()
def test_start_watch_threaded(self): path = '/test' m_tg = mock.Mock() m_tg.add_thread.return_value = mock.sentinel.watch_thread m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler, m_tg) watcher_obj._start_watch(path) m_tg.add_thread.assert_called_once_with(watcher_obj._watch, path) self.assertTrue(watcher_obj._idle.get(path)) self.assertEqual(mock.sentinel.watch_thread, watcher_obj._watching.get(path))
def test_stop_watching(self, m_stop_watch): paths = ['/test%s' % i for i in range(3)] m_handler = mock.Mock() watcher_obj = watcher.Watcher(m_handler) watcher_obj._resources.update(paths) m_watching = watcher_obj._watching = mock.MagicMock() m_watching.__iter__.return_value = paths watcher_obj.stop() self.assertFalse(watcher_obj._running) m_stop_watch.assert_has_calls([mock.call(path) for path in paths], any_order=True)
def _setup(self, params): args = ['--config-file', params.config.kuryr_conf] try: if params.config.debug: args.append('-d') except AttributeError: pass config.init(args) config.setup_logging() os_vif.initialize() clients.setup_kubernetes_client() self._pipeline = h_cni.CNIPipeline() self._watcher = k_watcher.Watcher(self._pipeline) self._watcher.add( "%(base)s/namespaces/%(namespace)s/pods" "?fieldSelector=metadata.name=%(pod)s" % { 'base': k_const.K8S_API_BASE, 'namespace': params.args.K8S_POD_NAMESPACE, 'pod': params.args.K8S_POD_NAME })