def setUp(self):
config = get_collector_config("TracerouteCollector", {})
self.collector = TracerouteCollector(config, None)
self.collector.config["bin"] = "dummy"
self.collector.config["protocol"] = "icmp"
def setUp(self):
config = get_collector_config('TracerouteCollector', {})
self.collector = TracerouteCollector(config, None)
self.collector.config['bin'] = 'dummy'
self.collector.config['protocol'] = 'icmp'
class TestTracerouteCollector(CollectorTestCase):
def setUp(self):
config = get_collector_config("TracerouteCollector", {})
self.collector = TracerouteCollector(config, None)
self.collector.config["bin"] = "dummy"
self.collector.config["protocol"] = "icmp"
def test_import(self):
self.assertTrue(TracerouteCollector)
@patch.object(Collector, "publish")
def test_should_work_with_real_data(self, publish_mock):
with patch("traceroute.Popen") as process_mock:
with patch.object(process_mock.return_value, "stdout") as out_mock:
with patch.object(process_mock.return_value, "stderr") as err_mock:
out_mock.readline = self.getFixture("traceroute").readline
err_mock.readline.return_value = None
self.collector.collect()
rtts = self.getFixture("rtts").getvalue().split("\n")
for idx, call in enumerate(publish_mock.mock_calls):
name, args, kwargs = call
metric_name, metric_value = args
self.assertEquals(metric_name, "yelp.RoundTripTime")
self.assertEquals(metric_value, float(rtts[idx]))
@patch.object(Collector, "publish_metric")
def test_sent_dimensions(self, publish_metric_mock):
with patch("traceroute.Popen") as process_mock:
with patch.object(process_mock.return_value, "stdout") as out_mock:
with patch.object(process_mock.return_value, "stderr") as err_mock:
out_mock.readline = self.getFixture("traceroute").readline
err_mock.readline.return_value = None
self.collector.collect()
hops = self.getFixture("hops").getvalue().split("\n")
for idx, call in enumerate(publish_metric_mock.mock_calls):
name, args, kwargs = call
metric = args[0]
hop, ip = hops[idx].strip().split("|")
if "*" not in ip:
self.assertEquals(metric.dimensions, {"hop": hop, "ip": ip})
else:
self.assertEquals(metric.dimensions, {"hop": hop})
@patch.object(Collector, "publish")
def test_should_fail_gracefully(self, publish_mock):
with patch("traceroute.Popen") as process_mock:
with patch.object(process_mock.return_value, "stdout") as out_mock:
with patch.object(process_mock.return_value, "stderr") as err_mock:
with patch.object(self.collector.log, "error") as error_logger:
err_mock.readline.return_value = "Failed to run collector"
self.collector.collect()
error_logger.assert_called_once_with("Error running traceroute process: Failed to run collector")
self.assertPublishedMany(publish_mock, {})
class TestTracerouteCollector(CollectorTestCase):
def setUp(self):
config = get_collector_config('TracerouteCollector', {})
self.collector = TracerouteCollector(config, None)
self.collector.config['bin'] = 'dummy'
self.collector.config['protocol'] = 'icmp'
def test_import(self):
self.assertTrue(TracerouteCollector)
@patch.object(Collector, 'publish')
def test_should_work_with_real_data(self, publish_mock):
with patch('traceroute.Popen') as process_mock:
with patch.object(process_mock.return_value, 'stdout') as out_mock:
with patch.object(process_mock.return_value,
'stderr') as err_mock:
out_mock.readline = self.getFixture('traceroute').readline
err_mock.readline.return_value = None
self.collector.collect()
rtts = self.getFixture('rtts').getvalue().split('\n')
for idx, call in enumerate(publish_mock.mock_calls):
name, args, kwargs = call
metric_name, metric_value = args
self.assertEquals(metric_name, 'yelp.RoundTripTime')
self.assertEquals(metric_value, float(rtts[idx]))
@patch.object(Collector, 'publish_metric')
def test_sent_dimensions(self, publish_metric_mock):
with patch('traceroute.Popen') as process_mock:
with patch.object(process_mock.return_value, 'stdout') as out_mock:
with patch.object(process_mock.return_value,
'stderr') as err_mock:
out_mock.readline = self.getFixture('traceroute').readline
err_mock.readline.return_value = None
self.collector.collect()
hops = self.getFixture('hops').getvalue().split('\n')
for idx, call in enumerate(publish_metric_mock.mock_calls):
name, args, kwargs = call
metric = args[0]
hop, ip = hops[idx].strip().split('|')
if '*' not in ip:
self.assertEquals(metric.dimensions, {
'hop': hop,
'ip': ip,
})
else:
self.assertEquals(metric.dimensions, {
'hop': hop,
})
@patch.object(Collector, 'publish')
def test_should_fail_gracefully(self, publish_mock):
with patch('traceroute.Popen') as process_mock:
with patch.object(process_mock.return_value, 'stdout') as out_mock:
with patch.object(process_mock.return_value,
'stderr') as err_mock:
with patch.object(self.collector.log,
'error') as error_logger:
err_mock.readline.return_value = 'Failed to run collector'
self.collector.collect()
error_logger.assert_called_once_with(
'Error running traceroute process: Failed to run collector')
self.assertPublishedMany(publish_mock, {})
