def test_max_retries_config_error(self, new_retry_policy): config = { "example.endpoint": "127.0.0.1:1234", "example.max_retries": "5", } with self.assertRaises(Exception): thrift_pool.thrift_pool_from_config(config, prefix="example.")
def test_max_connection_attempts_works_through_config( self, new_retry_policy): config = { "example.endpoint": "127.0.0.1:1234", "example.max_connection_attempts": "5" } thrift_pool.thrift_pool_from_config(config, prefix="example.") new_retry_policy.assert_called_with(attempts=5)
def test_max_retries_still_works_through_config(self, new_retry_policy): config = { "example.endpoint": "127.0.0.1:1234", "example.max_retries": "5" } with pytest.deprecated_call(): thrift_pool.thrift_pool_from_config(config, prefix="example.") new_retry_policy.assert_called_with(attempts=5)
def test_dont_configure_both(self, new_retry_policy): config = { "example.endpoint": "127.0.0.1:1234", "example.max_retries": "5", "example.max_connection_attempts": "5", } with self.assertRaises(Exception): thrift_pool.thrift_pool_from_config(config, prefix="example.")
def parse(self, key_path: str, raw_config: config.RawConfig) -> ContextFactory: pool = thrift_pool_from_config(raw_config, prefix=f"{key_path}.", **self.kwargs) return ThriftContextFactory(pool, self.client_cls)
def test_default_through_parser(self, new_retry_policy): config = {"example.endpoint": "127.0.0.1:1234"} thrift_pool.thrift_pool_from_config(config, prefix="example.") new_retry_policy.assert_called_with(attempts=3)