def test_pipeline_contextmanager_disabled(self, stats_client_cls, stats_config): stats_config['enabled'] = False lc = LazyClient(**stats_config) with lc.pipeline() as pipe: pipe.incr('foo') pipe.send() assert isinstance(pipe, Mock) assert lc.client.pipeline.mock_calls == []
def test_pipeline_contextmanager_enabled(self, stats_client_cls, stats_config): lc = LazyClient(**stats_config) with lc.pipeline() as pipe: pipe.incr('foo') pipe.send() assert pipe is lc.client.pipeline.return_value.__enter__.return_value assert lc.client.pipeline.mock_calls == [ call(), call().__enter__(), call().__enter__().incr('foo'), call().__enter__().send(), call().__exit__(None, None, None) ]