Skip to content

Commit 8ea3bf2

Browse files
committed
PR updates
1 parent 00ce11f commit 8ea3bf2

File tree

2 files changed

+35
-20
lines changed

2 files changed

+35
-20
lines changed

packages/smithy-http/src/smithy_http/interceptors/user_agent.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -46,12 +46,12 @@ class _UserAgentBuilder:
4646
def __init__(
4747
self,
4848
*,
49-
platform_name: str | None = None,
50-
platform_version: str | None = None,
51-
platform_machine: str | None = None,
52-
python_version: str | None = None,
53-
python_implementation: str | None = None,
54-
sdk_version: str | None = None,
49+
platform_name: str | None,
50+
platform_version: str | None,
51+
platform_machine: str | None,
52+
python_version: str | None,
53+
python_implementation: str | None,
54+
sdk_version: str | None,
5555
) -> None:
5656
self._platform_name = platform_name
5757
self._platform_version = platform_version

packages/smithy-http/tests/unit/interceptors/test_user_agent.py

Lines changed: 29 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -20,55 +20,70 @@ def test_from_environment(monkeypatch): # type: ignore
2020
assert "md/pyimpl#CPython" in user_agent
2121

2222

23+
defaults = {
24+
"platform_name": None,
25+
"platform_version": None,
26+
"platform_machine": None,
27+
"python_version": None,
28+
"python_implementation": None,
29+
"sdk_version": None,
30+
}
31+
32+
2333
def test_build_adds_sdk_metadata():
24-
user_agent = _UserAgentBuilder(sdk_version="1.2.3").build()
34+
args = {"sdk_version": "1.2.3"}
35+
user_agent = _UserAgentBuilder(**{**defaults, **args}).build()
2536
assert "python/1.2.3" in str(user_agent)
2637

2738

2839
def test_build_adds_ua_metadata():
29-
user_agent = _UserAgentBuilder().build()
40+
user_agent = _UserAgentBuilder(**defaults).build()
3041
assert "ua/2.1" in str(user_agent)
3142

3243

3344
def test_build_os_defaults_to_other():
34-
user_agent = _UserAgentBuilder().build()
45+
user_agent = _UserAgentBuilder(**defaults).build()
3546
assert "os/other" in str(user_agent)
3647

3748

3849
def test_build_os_lowercases_platform():
39-
user_agent = _UserAgentBuilder(platform_name="LINUX").build()
50+
args = {"platform_name": "LINUX"}
51+
user_agent = _UserAgentBuilder(**{**defaults, **args}).build()
4052
assert "os/linux" in str(user_agent)
4153

4254

4355
def test_build_os_maps_platform_names():
44-
user_agent = _UserAgentBuilder(platform_name="darwin").build()
56+
args = {"platform_name": "darwin"}
57+
user_agent = _UserAgentBuilder(**{**defaults, **args}).build()
58+
4559
assert "os/macos" in str(user_agent)
4660

4761

4862
def test_build_os_includes_version():
49-
user_agent = _UserAgentBuilder(
50-
platform_name="linux", platform_version="5.4"
51-
).build()
63+
args = {"platform_name": "linux", "platform_version": "5.4"}
64+
user_agent = _UserAgentBuilder(**{**defaults, **args}).build()
5265
assert "os/linux#5.4" in str(user_agent)
5366

5467

5568
def test_build_os_other_platform():
56-
user_agent = _UserAgentBuilder(
57-
platform_name="myos", platform_version="0.0.1"
58-
).build()
69+
args = {"platform_name": "myos", "platform_version": "0.0.1"}
70+
user_agent = _UserAgentBuilder(**{**defaults, **args}).build()
5971
assert "os/other md/myos#0.0.1" in str(user_agent)
6072

6173

6274
def test_build_arch_adds_md():
63-
user_agent = _UserAgentBuilder(platform_machine="x86_64").build()
75+
args = {"platform_machine": "x86_64"}
76+
user_agent = _UserAgentBuilder(**{**defaults, **args}).build()
6477
assert "md/arch#x86_64" in str(user_agent)
6578

6679

6780
def test_build_language_version():
68-
user_agent = _UserAgentBuilder(python_version="3.12").build()
81+
args = {"python_version": "3.12"}
82+
user_agent = _UserAgentBuilder(**{**defaults, **args}).build()
6983
assert "lang/python#3.12" in str(user_agent)
7084

7185

7286
def test_build_language_implementation():
73-
user_agent = _UserAgentBuilder(python_implementation="CPython").build()
87+
args = {"python_implementation": "CPython"}
88+
user_agent = _UserAgentBuilder(**{**defaults, **args}).build()
7489
assert "md/pyimpl#CPython" in str(user_agent)

0 commit comments

Comments
 (0)