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)
