Skip to content

Commit 00ce11f

Browse files
committed
PR cleanups
1 parent 0013139 commit 00ce11f

File tree

2 files changed

+19
-16
lines changed

2 files changed

+19
-16
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ class UserAgentInterceptor(Interceptor[Any, None, HTTPRequest, None]):
1717
def read_before_execution(
1818
self, context: InterceptorContext[Any, None, None, None]
1919
) -> None:
20-
context.properties["user_agent"] = UserAgentBuilder.from_environment().build()
20+
context.properties["user_agent"] = _UserAgentBuilder.from_environment().build()
2121

2222
def modify_before_signing(
2323
self, context: InterceptorContext[Any, None, HTTPRequest, None]
@@ -42,7 +42,7 @@ def modify_before_signing(
4242
_USERAGENT_SDK_NAME = "python"
4343

4444

45-
class UserAgentBuilder:
45+
class _UserAgentBuilder:
4646
def __init__(
4747
self,
4848
*,
Lines changed: 17 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,71 +1,74 @@
11
import platform
22

33
import smithy_core
4-
from smithy_http.interceptors.user_agent import UserAgentBuilder
4+
from smithy_http.interceptors.user_agent import _UserAgentBuilder # type: ignore
55

66

77
def test_from_environment(monkeypatch): # type: ignore
88
monkeypatch.setattr(platform, "system", lambda: "Linux") # type: ignore
99
monkeypatch.setattr(platform, "release", lambda: "5.4.228-131.415.AMZN2.X86_64") # type: ignore
10+
monkeypatch.setattr(platform, "machine", lambda: "x86_64") # type: ignore
1011
monkeypatch.setattr(platform, "python_version", lambda: "4.3.2") # type: ignore
1112
monkeypatch.setattr(platform, "python_implementation", lambda: "CPython") # type: ignore
1213
monkeypatch.setattr(smithy_core, "__version__", "1.2.3") # type: ignore
1314

14-
user_agent = str(UserAgentBuilder.from_environment().build())
15+
user_agent = str(_UserAgentBuilder.from_environment().build())
1516
assert "python/1.2.3" in user_agent
1617
assert "os/linux#5.4.228-131.415.AMZN2.X86_64" in user_agent
17-
assert "md/arch#arm64" in user_agent
18+
assert "md/arch#x86_64" in user_agent
1819
assert "lang/python#4.3.2" in user_agent
19-
assert "md/pyimpl#Cpython" in user_agent
20+
assert "md/pyimpl#CPython" in user_agent
2021

2122

2223
def test_build_adds_sdk_metadata():
23-
user_agent = UserAgentBuilder(sdk_version="1.2.3").build()
24+
user_agent = _UserAgentBuilder(sdk_version="1.2.3").build()
2425
assert "python/1.2.3" in str(user_agent)
2526

2627

2728
def test_build_adds_ua_metadata():
28-
user_agent = UserAgentBuilder().build()
29+
user_agent = _UserAgentBuilder().build()
2930
assert "ua/2.1" in str(user_agent)
3031

3132

3233
def test_build_os_defaults_to_other():
33-
user_agent = UserAgentBuilder().build()
34+
user_agent = _UserAgentBuilder().build()
3435
assert "os/other" in str(user_agent)
3536

3637

3738
def test_build_os_lowercases_platform():
38-
user_agent = UserAgentBuilder(platform_name="LINUX").build()
39+
user_agent = _UserAgentBuilder(platform_name="LINUX").build()
3940
assert "os/linux" in str(user_agent)
4041

4142

4243
def test_build_os_maps_platform_names():
43-
user_agent = UserAgentBuilder(platform_name="darwin").build()
44+
user_agent = _UserAgentBuilder(platform_name="darwin").build()
4445
assert "os/macos" in str(user_agent)
4546

4647

4748
def test_build_os_includes_version():
48-
user_agent = UserAgentBuilder(platform_name="linux", platform_version="5.4").build()
49+
user_agent = _UserAgentBuilder(
50+
platform_name="linux", platform_version="5.4"
51+
).build()
4952
assert "os/linux#5.4" in str(user_agent)
5053

5154

5255
def test_build_os_other_platform():
53-
user_agent = UserAgentBuilder(
56+
user_agent = _UserAgentBuilder(
5457
platform_name="myos", platform_version="0.0.1"
5558
).build()
5659
assert "os/other md/myos#0.0.1" in str(user_agent)
5760

5861

5962
def test_build_arch_adds_md():
60-
user_agent = UserAgentBuilder(platform_machine="x86_64").build()
63+
user_agent = _UserAgentBuilder(platform_machine="x86_64").build()
6164
assert "md/arch#x86_64" in str(user_agent)
6265

6366

6467
def test_build_language_version():
65-
user_agent = UserAgentBuilder(python_version="3.12").build()
68+
user_agent = _UserAgentBuilder(python_version="3.12").build()
6669
assert "lang/python#3.12" in str(user_agent)
6770

6871

6972
def test_build_language_implementation():
70-
user_agent = UserAgentBuilder(python_implementation="CPython").build()
73+
user_agent = _UserAgentBuilder(python_implementation="CPython").build()
7174
assert "md/pyimpl#CPython" in str(user_agent)

0 commit comments

Comments
 (0)