@@ -10,78 +10,78 @@ class TestRedTeamUserAgent:
1010 def test_default_redteam_context (self ):
1111 """Test that the default red team context works as expected."""
1212 base_user_agent = UserAgentSingleton ().value
13-
13+
1414 with UserAgentSingleton .redteam_context ():
1515 redteam_user_agent = UserAgentSingleton ().value
1616 expected = f"{ base_user_agent } (type=redteam; subtype=RedTeam)"
1717 assert redteam_user_agent == expected
18-
18+
1919 # Should restore to original after context
2020 assert UserAgentSingleton ().value == base_user_agent
2121
2222 def test_custom_subtype_redteam_context (self ):
2323 """Test red team context with custom subtype."""
2424 base_user_agent = UserAgentSingleton ().value
2525 custom_subtype = "CustomTest"
26-
26+
2727 with UserAgentSingleton .redteam_context (subtype = custom_subtype ):
2828 redteam_user_agent = UserAgentSingleton ().value
2929 expected = f"{ base_user_agent } (type=redteam; subtype={ custom_subtype } )"
3030 assert redteam_user_agent == expected
31-
31+
3232 # Should restore to original after context
3333 assert UserAgentSingleton ().value == base_user_agent
3434
3535 def test_nested_contexts (self ):
3636 """Test nested user agent contexts."""
3737 base_user_agent = UserAgentSingleton ().value
38-
38+
3939 with UserAgentSingleton .add_useragent_product ("outer/1.0" ):
4040 outer_user_agent = UserAgentSingleton ().value
4141 expected_outer = f"{ base_user_agent } outer/1.0"
4242 assert outer_user_agent == expected_outer
43-
43+
4444 with UserAgentSingleton .redteam_context (subtype = "NestedTest" ):
4545 nested_user_agent = UserAgentSingleton ().value
4646 expected_nested = f"{ expected_outer } (type=redteam; subtype=NestedTest)"
4747 assert nested_user_agent == expected_nested
48-
48+
4949 # Should restore to outer context
5050 assert UserAgentSingleton ().value == expected_outer
51-
51+
5252 # Should restore to original after all contexts
5353 assert UserAgentSingleton ().value == base_user_agent
5454
5555 def test_multiple_redteam_contexts (self ):
5656 """Test multiple red team contexts in sequence."""
5757 base_user_agent = UserAgentSingleton ().value
58-
58+
5959 # First context
6060 with UserAgentSingleton .redteam_context (subtype = "First" ):
6161 first_user_agent = UserAgentSingleton ().value
6262 expected_first = f"{ base_user_agent } (type=redteam; subtype=First)"
6363 assert first_user_agent == expected_first
64-
64+
6565 # Should restore to original
6666 assert UserAgentSingleton ().value == base_user_agent
67-
68- # Second context
67+
68+ # Second context
6969 with UserAgentSingleton .redteam_context (subtype = "Second" ):
7070 second_user_agent = UserAgentSingleton ().value
7171 expected_second = f"{ base_user_agent } (type=redteam; subtype=Second)"
7272 assert second_user_agent == expected_second
73-
73+
7474 # Should restore to original
7575 assert UserAgentSingleton ().value == base_user_agent
7676
7777 def test_redteam_context_kwargs_ignored (self ):
7878 """Test that additional kwargs are ignored gracefully."""
7979 base_user_agent = UserAgentSingleton ().value
80-
80+
8181 with UserAgentSingleton .redteam_context (subtype = "Test" , ignored_param = "should_be_ignored" ):
8282 redteam_user_agent = UserAgentSingleton ().value
8383 expected = f"{ base_user_agent } (type=redteam; subtype=Test)"
8484 assert redteam_user_agent == expected
85-
85+
8686 # Should restore to original after context
87- assert UserAgentSingleton ().value == base_user_agent
87+ assert UserAgentSingleton ().value == base_user_agent
0 commit comments