def test_timing(self):
"""Test timing wrapper."""
client = flask_statsdclient.StatsDClient(self.app)
timestamp = int((time.time() - 1515191917) * 1000)
with patch('datadog.statsd.timing') as mock_timing:
client.timing('test.counter', timestamp)
mock_timing.assert_called_once_with('test.counter', timestamp)
def test_gauge(self):
"""Test gauge wrapper."""
client = flask_statsdclient.StatsDClient(self.app)
with patch('datadog.statsd.gauge') as mock_gauge:
client.gauge('test', 1)
mock_gauge.assert_called_once_with('test', 1)
client.gauge('foo', 1, delta=True)
mock_gauge.assert_called_with('foo', 1, delta=True)
def test_incr_with_prefix(self):
"""Test incr wrapper with prefix."""
self.app.config['STATSD_PREFIX'] = 'foo'
client = flask_statsdclient.StatsDClient(self.app)
with patch('datadog.statsd.increment') as mock_incr:
client.incr('test.counter')
mock_incr.assert_called_once_with('foo.test.counter')
def test_custom_app_config(self):
"""Test custom configs set on app."""
self.app.config['STATSD_HOST'] = '10.1.1.1'
self.app.config['STATSD_PORT'] = 9999
self.app.config['STATSD_PREFIX'] = 'foo'
client = flask_statsdclient.StatsDClient(self.app)
self.assertEqual(('10.1.1.1', 9999), client.statsd.__dict__['_addr'])
self.assertEqual('foo', client.statsd.__dict__['_prefix'])
def test_incr(self):
"""Test incr wrapper."""
client = flask_statsdclient.StatsDClient(self.app)
with patch('datadog.statsd.increment') as mock_incr:
client.incr('test.counter')
mock_incr.assert_called_once_with('test.counter')
client.incr('test.counter', 10)
mock_incr.assert_called_with('test.counter', 10)
client.incr('test.counter', rate=0.1)
mock_incr.assert_called_with('test.counter', rate=0.1)
def test_custom_kwarg_config(self):
"""Test custom configs passed via kwargs."""
config = {
'STATSD_HOST': '1.2.3.4',
'STATSD_PORT': 12345,
'STATSD_PREFIX': 'foo.bar'
}
client = flask_statsdclient.StatsDClient(self.app, config)
self.assertEqual(('1.2.3.4', 12345), client.statsd.__dict__['_addr'])
self.assertEqual('foo.bar', client.statsd.__dict__['_prefix'])
def test_set(self):
"""Test set wrapper."""
client = flask_statsdclient.StatsDClient(self.app)
with patch('datadog.statsd.set') as mock_set:
client.set('test.counter', 1)
mock_set.assert_called_once_with('test.counter', 1)
def test_decr(self):
"""Test decr wrapper."""
client = flask_statsdclient.StatsDClient(self.app)
with patch('datadog.statsd.decrement') as mock_decr:
client.decr('test.counter')
mock_decr.assert_called_once_with('test.counter')
def create_app():
"""Create a Flask app for context."""
app = Flask(__name__)
client = flask_statsdclient.StatsDClient()
client.init_app(app)
return app
def test_default_config(self):
"""Test the default configs."""
client = flask_statsdclient.StatsDClient(self.app)
self.assertEqual(('127.0.0.1', 8125), client.statsd.__dict__['_addr'])
def test_timer(self):
"""Test timer wrapper."""
client = flask_statsdclient.StatsDClient(self.app)
with patch('statsd.StatsClient.timer') as mock_timer:
client.timer('test.counter')
mock_timer.assert_called_once_with('test.counter')