github/sprockets.mixins.metrics
1
0
Fork 0
mirror of https://github.com/sprockets/sprockets.mixins.metrics.git synced 2024-11-21 19:28:34 +00:00
Code Issues Projects Releases Packages Wiki Activity

Set format strings ass attributes, update tests

Browse source
This commit is contained in:
Dan g 2018-07-27 13:51:02 -04:00
parent 6fd2cf4035
commit 5599e9f73b
2 changed files with 12 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
sprockets/mixins/metrics/statsd.py
View file

@ -123,9 +123,11 @@ class StatsDCollector(object):
if protocol == 'tcp':
self._tcp = True
self._msg_format = '{path}:{value}|{metric_type}\n'
self._sock = self._tcp_socket()
elif protocol == 'udp':
self._tcp = False
self._msg_format = '{path}:{value}|{metric_type}'
self._sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, 0)
else:
raise ValueError('Invalid protocol: {}'.format(protocol))
@ -160,9 +162,10 @@ class StatsDCollector(object):
:param str metric_type: The metric type
"""
msg = '{0}:{1}|{2}'.format(
self._build_path(path, metric_type), value, metric_type)
msg = self._build_udp_or_tcp_message(msg)
msg = self._msg_format.format(
path=self._build_path(path, metric_type),
value=value,
metric_type=metric_type)
LOGGER.debug('Sending %s to %s:%s', msg.encode('ascii'),
self._host, self._port)
@ -175,12 +178,6 @@ class StatsDCollector(object):
except (OSError, socket.error) as error: # pragma: nocover
LOGGER.exception('Error sending statsd metric: %s', error)
def _build_udp_or_tcp_message(self, msg):
if self._tcp is False:
return msg
return msg + "\n"
def _build_path(self, path, metric_type):
"""Return a normalized path.

17
tests.py
View file

@ -63,12 +63,9 @@ class TCPStatsdMetricCollectionTests(testing.AsyncHTTPTestCase):
'protocol': 'tcp',
'prepend_metric_type': True})
def test_that_tcp_message_appends_a_newline(self):
orig = 'testing.timers.SimpleHandler.GET.204'
expected = 'testing.timers.SimpleHandler.GET.204\n'
msg = self.application.statsd._build_udp_or_tcp_message(orig)
self.assertEqual(msg, expected)
def test_tcp_message_format(self):
expected = '{path}:{value}|{metric_type}\n'
self.assertEqual(self.application.statsd._msg_format, expected)
def test_that_http_method_call_is_recorded(self):
response = self.fetch('/')
@ -178,11 +175,9 @@ class UDPStatsdMetricCollectionTests(testing.AsyncHTTPTestCase):
self.statsd.close()
super(UDPStatsdMetricCollectionTests, self).tearDown()
def test_that_udp_message_is_unchanged(self):
expected = 'testing.timers.SimpleHandler.GET.204'
msg = self.application.statsd._build_udp_or_tcp_message(expected)
self.assertEqual(msg, expected)
def test_udp_message_format(self):
expected = '{path}:{value}|{metric_type}'
self.assertEqual(self.application.statsd._msg_format, expected)
def test_that_http_method_call_is_recorded(self):
response = self.fetch('/')