def test_name_argument_deprecated(self, recwarn): warnings.simplefilter("always") StatsD('test', name='statsd') assert len(recwarn) == 1 warn = recwarn.pop() assert str(warn.message) == ( "The `name` argument to `StatsD` is no longer needed and has been" " deprecated.")
class DummyServiceTCP(ServiceBase): """Fake Service to test the `StatsD.timer` decorator. """ name = 'dummy_service' statsd = StatsD('test-tcp') @dummy @statsd.timer('nice-tcp-stat', rate=3) def method(self, *args, **kwargs): sentinel = Mock() sentinel(*args, **kwargs) return sentinel
class DummyServiceDisabled(ServiceBase): """Fake Service to test the `StatsD.timer` decorator when disabled. """ name = 'dummy_service' statsd = StatsD('test-disabled') @dummy @statsd.timer('disabled-nice-stat') def method(self, *args, **kwargs): sentinel = Mock() sentinel(*args, **kwargs) return sentinel
class DummyServiceManual(object): """Fake Service to test the `StatsD.timer` decorator without metaclass. """ name = 'dummy_service' statsd = StatsD('test', name='statsd') @dummy @statsd.timer('nice-stat', rate=3) def method(self, *args, **kwargs): sentinel = Mock() sentinel(*args, **kwargs) return sentinel
class RPCDummyService(object): """Fake Service to test the `StatsD.timer` decorator. """ name = 'dummy_service' statsd = StatsD('test') @rpc @statsd.timer('nice-stat', rate=3) def method(self, *args, **kwargs): sentinel = Mock() sentinel(*args, **kwargs) return sentinel
class DummyServiceMeta(ServiceBase): """ Fake Service to test the `StatsD.timer` decorator with the deprecated metaclass """ name = 'dummy_service' statsd = StatsD('test') @dummy @statsd.timer('nice-stat', rate=3) def method(self, *args, **kwargs): sentinel = Mock() sentinel(*args, **kwargs) return sentinel
def test_get_dependency_tcp(self, lazy_client_cls, stats_config): statsd = StatsD('test-tcp') statsd.container = Mock() statsd.container.config = stats_config statsd.setup() worker_ctx = Mock() dep = statsd.get_dependency(worker_ctx) assert lazy_client_cls.call_args_list == [ call(host='tcp.statsd.host', port=4321, prefix='tcp.statsd.prefix', timeout=5, enabled=True, protocol='tcp') ] assert dep == lazy_client_cls.return_value
def test_get_dependency(self, lazy_client_cls, stats_config): statsd = StatsD('test') statsd.container = Mock() statsd.container.config = stats_config statsd.setup() worker_ctx = Mock() dep = statsd.get_dependency(worker_ctx) assert lazy_client_cls.call_args_list == [ call( host='statsd.host', port=1234, prefix='statsd.prefix', maxudpsize=1024, enabled=True, ) ] assert dep == lazy_client_cls.return_value