Skip to content

Commit d7d2bf6

Browse files
committed
use config_store directly in TCP keepalive tests
1 parent 88c50a7 commit d7d2bf6

1 file changed

Lines changed: 15 additions & 26 deletions

File tree

tests/unit/test_args.py

Lines changed: 15 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -116,15 +116,6 @@ def call_get_client_args(self, **override_kwargs):
116116
}
117117
call_kwargs.update(**override_kwargs)
118118

119-
# Bridge test's scoped_config to config_store for new _compute_socket_options implementation
120-
# The method now reads from config_store instead of receiving scoped_config as parameter
121-
scoped_config = call_kwargs.get('scoped_config', {})
122-
if scoped_config:
123-
for key, value in scoped_config.items():
124-
# Convert 'true'/'false' strings to boolean for tcp_keepalive
125-
if key == 'tcp_keepalive' and isinstance(value, str):
126-
value = value.lower() == 'true'
127-
self.config_store.set_config_variable(key, value)
128119

129120
return self.args_create.get_client_args(**call_kwargs)
130121

@@ -251,48 +242,46 @@ def test_region_does_not_resolve_if_not_s3_and_endpoint_url_provided(self):
251242
)
252243
self.assertEqual(client_args['client_config'].region_name, None)
253244

254-
def test_tcp_keepalive_enabled_scoped_config(self):
255-
scoped_config = {'tcp_keepalive': 'true'}
245+
def test_tcp_keepalive_enabled_config_store(self):
246+
self.config_store.set_config_variable('tcp_keepalive', True)
256247
with mock.patch('botocore.args.EndpointCreator') as m:
257-
self.call_get_client_args(scoped_config=scoped_config)
248+
self.call_get_client_args()
258249
self.assert_create_endpoint_call(
259250
m,
260251
socket_options=self.default_socket_options
261-
+ [(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)],
252+
+ [(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)],
262253
)
263254

264255
def test_tcp_keepalive_not_specified(self):
265256
with mock.patch('botocore.args.EndpointCreator') as m:
266-
self.call_get_client_args(scoped_config={}, client_config=None)
257+
self.call_get_client_args(client_config=None)
267258
self.assert_create_endpoint_call(
268259
m, socket_options=self.default_socket_options
269260
)
270-
self.call_get_client_args(
271-
scoped_config=None, client_config=Config()
272-
)
261+
self.call_get_client_args(client_config=Config())
273262
self.assert_create_endpoint_call(
274263
m, socket_options=self.default_socket_options
275264
)
276265

277266
def test_tcp_keepalive_enabled_if_set_anywhere(self):
278267
with mock.patch('botocore.args.EndpointCreator') as m:
268+
self.config_store.set_config_variable('tcp_keepalive', True)
279269
self.call_get_client_args(
280-
scoped_config={'tcp_keepalive': 'true'},
281270
client_config=Config(tcp_keepalive=False),
282271
)
283272
self.assert_create_endpoint_call(
284273
m,
285274
socket_options=self.default_socket_options
286-
+ [(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)],
275+
+ [(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)],
287276
)
277+
self.config_store.set_config_variable('tcp_keepalive', False)
288278
self.call_get_client_args(
289-
scoped_config={'tcp_keepalive': 'false'},
290279
client_config=Config(tcp_keepalive=True),
291280
)
292281
self.assert_create_endpoint_call(
293282
m,
294283
socket_options=self.default_socket_options
295-
+ [(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)],
284+
+ [(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)],
296285
)
297286

298287
def test_tcp_keepalive_enabled_environment_variable(self):
@@ -306,21 +295,21 @@ def test_tcp_keepalive_enabled_environment_variable(self):
306295
)
307296

308297
def test_tcp_keepalive_explicitly_disabled(self):
309-
scoped_config = {'tcp_keepalive': 'false'}
298+
self.config_store.set_config_variable('tcp_keepalive', False)
310299
with mock.patch('botocore.args.EndpointCreator') as m:
311-
self.call_get_client_args(scoped_config=scoped_config)
300+
self.call_get_client_args()
312301
self.assert_create_endpoint_call(
313302
m, socket_options=self.default_socket_options
314303
)
315304

316305
def test_tcp_keepalive_enabled_case_insensitive(self):
317-
scoped_config = {'tcp_keepalive': 'True'}
306+
self.config_store.set_config_variable('tcp_keepalive', True)
318307
with mock.patch('botocore.args.EndpointCreator') as m:
319-
self.call_get_client_args(scoped_config=scoped_config)
308+
self.call_get_client_args()
320309
self.assert_create_endpoint_call(
321310
m,
322311
socket_options=self.default_socket_options
323-
+ [(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)],
312+
+ [(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)],
324313
)
325314

326315
def test_client_config_has_use_dualstack_endpoint_flag(self):

0 commit comments

Comments
 (0)