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 __init__(self): super(FuxiK8sService, self).__init__() pipeline = h_pipeline.ControllerPipeline(self.tg) self.watcher = watcher.Watcher(pipeline, self.tg) # TODO(ivc): pluggable resource/handler registration # for resource in ["pods", "persistentvolumeclaims"]: for resource in ["persistentvolumeclaims"]: self.watcher.add("%s/%s" % (constants.K8S_API_BASE, resource)) pipeline.register(h_pvc.PVCHandler())
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_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)