def test_same_statsd_client_for_same_prefix(self) -> None:
with patch.object(current_app, 'config') as mock_config, \
patch.object(StatsClient, '__init__', return_value=None) as mock_statsd_init:
mock_config.return_value.single.return_value = True
statsd_client1 = _get_statsd_client(prefix='test_same')
self.assertIsNotNone(statsd_client1)
statsd_client2 = _get_statsd_client(prefix='test_same')
self.assertIsNotNone(statsd_client2)
self.assertEqual(statsd_client1, statsd_client2)
self.assertEqual(mock_statsd_init.call_count, 1)
def test_get_statsd_client(self) -> None:
with patch.object(current_app, 'config') as mock_config, \
patch.object(StatsClient, '__init__', return_value=None):
mock_config.return_value.single.return_value = True
statsd_client1 = _get_statsd_client(prefix='test')
self.assertIsNotNone(statsd_client1)
def test_no_statsd_client(self) -> None:
with patch.object(StatsClient, '__init__'):
statsd_client = _get_statsd_client(prefix='foo')
self.assertIsNone(statsd_client)
