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
})
