def test_start_with_kwargs(self):
f = cloudtail_factory.CloudtailFactory('/foo', None)
f.start('log', mock.Mock(), cwd='bar')
self.assertEqual(1, self.mock_popen.call_count)
kwargs = self.mock_popen.call_args[1]
self.assertEqual('bar', kwargs['cwd'])
def test_start_with_credentials(self):
f = cloudtail_factory.CloudtailFactory('/foo', '/bar')
f.start('log', mock.Mock())
self.assertEqual(1, self.mock_popen.call_count)
self.assertEqual([
'/foo', 'pipe', '--log-id', 'log', '--local-log-level', 'info',
'--ts-mon-credentials', '/bar'
], self.mock_popen.call_args[0][0])
def test_start(self):
fh = mock.Mock()
f = cloudtail_factory.CloudtailFactory('/foo', None)
f.start('log', fh)
self.assertEqual(1, self.mock_popen.call_count)
self.assertEqual(
['/foo', 'pipe', '--log-id', 'log', '--local-log-level', 'info'],
self.mock_popen.call_args[0][0])
kwargs = self.mock_popen.call_args[1]
self.assertEqual(fh, kwargs['stdin'])
self.assertIn('stdout', kwargs)
self.assertIn('stderr', kwargs)
def main(self, opts):
if opts.cloudtail_path:
cloudtail = cloudtail_factory.CloudtailFactory(
opts.cloudtail_path,
opts.cloudtail_ts_mon_credentials)
else:
cloudtail = cloudtail_factory.DummyCloudtailFactory()
watcher = config_watcher.ConfigWatcher(
opts.config_directory,
opts.config_poll_interval,
opts.service_poll_interval,
opts.state_directory,
opts.cipd_version_file,
cloudtail)
def sigint_handler(_signal, _frame):
watcher.stop()
previous_sigint_handler = signal.signal(signal.SIGINT, sigint_handler)
watcher.run()
signal.signal(signal.SIGINT, previous_sigint_handler)