@@ -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+
2333def 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
2839def 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
3344def 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
3849def 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
4355def 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
4862def 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
5568def 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
6274def 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
6780def 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
7286def 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