def test_create_destinations(self):
app = App(['--config', self.config_filename])
destinations = app.create_destinations()
self.assertEqual(1, len(destinations))
self.assertEqual([Stdout()], destinations)
app_flush_graphite = App(['--flush-graphite', 'localhost'])
destinations = app_flush_graphite.create_destinations()
self.assertEqual(1, len(destinations))
self.assertEqual([Graphite('localhost', 2003)], destinations)
app_multi_flush = App(
['--config', self.config_filename, '--flush-graphite',
'example.org:2006,localhost'])
destinations = app_multi_flush.create_destinations()
self.assertEqual(3, len(destinations))
expected = [
Stdout(), Graphite('example.org', 2006),
Graphite ('localhost', 2003)
]
self.assertEqual(expected, destinations)
def test_create_destinations(self):
app = App(['--config', self.config_filename])
destinations = app.create_destinations()
self.assertEqual(1, len(destinations))
self.assertEqual([Stdout()], destinations)
app_flush_graphite = App(['--flush-graphite', 'localhost'])
destinations = app_flush_graphite.create_destinations()
self.assertEqual(1, len(destinations))
self.assertEqual([Graphite('localhost', 2003)], destinations)
app_multi_flush = App([
'--config', self.config_filename, '--flush-graphite',
'example.org:2006,localhost'
])
destinations = app_multi_flush.create_destinations()
self.assertEqual(3, len(destinations))
expected = [
Stdout(),
Graphite('example.org', 2006),
Graphite('localhost', 2003)
]
self.assertEqual(expected, destinations)
def test_create_csv_file_destinations(self):
try:
_, temp_file_name = mkstemp()
other_temp_file_name = temp_file_name + '_2'
app = App([
'--flush-file-csv', '{}|{}'.format(temp_file_name,
other_temp_file_name)
])
destinations = app.create_destinations()
self.assertEqual(2, len(destinations))
self.assertEqual(
[CsvFile(temp_file_name),
CsvFile(other_temp_file_name)], destinations)
finally:
if os.path.exists(temp_file_name):
os.remove(temp_file_name)
if os.path.exists(other_temp_file_name):
os.remove(other_temp_file_name)
