def test_prefix(self):
client = statsd.StatsdClient('localhost', 8125, prefix='main.bucket', sample_rate=None)
client._send(b'subname', b'100|c')
self.assertEqual(client._socket.data, b'main.bucket.subname:100|c')
client = statsd.StatsdClient('localhost', 8125, prefix='main', sample_rate=None)
client._send(b'subname', b'100|c')
self.assertEqual(client._socket.data, b'main.subname:100|c')
client._send(b'subname.subsubname', b'100|c')
self.assertEqual(client._socket.data, b'main.subname.subsubname:100|c')
def test_timing(self):
client = statsd.StatsdClient('localhost',
8125,
prefix='',
sample_rate=None)
client.timing('buck.timing', 100)
self.assertEqual(client._socket.data, b'buck.timing:100|ms')
def test_startstop(self):
timer = statsd.StatsdTimer('timeit', statsd.StatsdClient('localhost', 8125, prefix='', sample_rate=None))
timer.start()
time.sleep(0.25)
timer.stop()
self.assertTrue(timer._client._socket.data.startswith(b'timeit.total:2'))
self.assertTrue(timer._client._socket.data.endswith(b'|ms'))
def test_couter_factory(self):
client = statsd.StatsdClient('localhost', 8125, prefix='', sample_rate=0.999)
counter = client.counter('counted')
counter += 1
self.assertEqual(counter._client._socket.data, b'counted:1|c|@0.999')
counter += 5
self.assertEqual(counter._client._socket.data, b'counted:5|c|@0.999')
def test_send(self):
client = statsd.StatsdClient('localhost',
8125,
prefix='',
sample_rate=None)
client._send(b'buck', b'50|c')
self.assertEqual(client._socket.data, b'buck:50|c')
def test_incr(self):
client = statsd.StatsdClient('localhost',
8125,
prefix='',
sample_rate=None)
client.incr('buck.counter', 5)
self.assertEqual(client._socket.data, b'buck.counter:5|c')
def test_timer_factory(self):
client = statsd.StatsdClient('localhost', 8125, prefix='', sample_rate=0.999)
timer = client.timer('timeit')
timer.start()
time.sleep(0.25)
timer.stop()
self.assertTrue(timer._client._socket.data.startswith(b'timeit.total:2'))
self.assertTrue(timer._client._socket.data.endswith(b'|ms|@0.999'))
def test_timing_sample_rate(self):
client = statsd.StatsdClient('localhost',
8125,
prefix='',
sample_rate=0.999)
client.timing('buck.timing', 100)
self.assertEqual(client._socket.data, b'buck.timing:100|ms|@0.999')
if client._socket.data != '':
self.assertTrue(client._socket.data.endswith(b'|@0.999'))
def test_send_sample_rate(self):
client = statsd.StatsdClient('localhost',
8125,
prefix='',
sample_rate=0.999)
client._send(b'buck', b'50|c')
self.assertEqual(client._socket.data, b'buck:50|c|@0.999')
if client._socket.data != 'buck:50|c':
self.assertTrue(client._socket.data.endswith(b'|@0.999'))
def test_incr_sample_rate(self):
client = statsd.StatsdClient('localhost',
8125,
prefix='',
sample_rate=0.999)
client.incr('buck.counter', 5)
self.assertEqual(client._socket.data, b'buck.counter:5|c|@0.999')
if client._socket.data != 'buck.counter:5|c':
self.assertTrue(client._socket.data.endswith(b'|@0.999'))
def __init__(
self,
*args,
host=DEFAULT_HOST,
port=DEFAULT_PORT,
namespace='routemaster',
tags=None,
):
self.statsd = statsd.StatsdClient(
host=host,
port=int(port),
namespace=namespace,
tags=tags or {},
)
super().__init__(*args)
def test_sub(self):
counter = statsd.StatsdCounter('counted', statsd.StatsdClient('localhost', 8125, prefix='', sample_rate=None))
counter -= 1
self.assertEqual(counter._client._socket.data, b'counted:-1|c')
counter -= 5
self.assertEqual(counter._client._socket.data, b'counted:-5|c')
