style: Apply Black formatting to test files

- Applied Black formatter to all test files
- Fixed line spacing and string formatting issues
- No functional changes, only code style updates

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: christensen143

tests/test_aws_client.py

@@ -28,9 +28,7 @@ def test_list_claude_models_success(self, mock_run, mock_aws_response):
         """Test successful listing of Claude models."""
         # Arrange
         client = BedrockClient("us-west-2")
-        mock_run.return_value = MagicMock(
-            returncode=0, stdout=json.dumps(mock_aws_response)
-        )
+        mock_run.return_value = MagicMock(returncode=0, stdout=json.dumps(mock_aws_response))
 
         # Act
         result = client.list_claude_models()
@@ -100,9 +98,7 @@ def test_list_claude_models_no_claude_models(self, mock_run):
                 }
             ]
         }
-        mock_run.return_value = MagicMock(
-            returncode=0, stdout=json.dumps(non_claude_response)
-        )
+        mock_run.return_value = MagicMock(returncode=0, stdout=json.dumps(non_claude_response))
 
         # Act
         result = client.list_claude_models()
@@ -129,9 +125,7 @@ def test_list_claude_models_missing_profile_name(self, mock_run):
                 }
             ]
         }
-        mock_run.return_value = MagicMock(
-            returncode=0, stdout=json.dumps(response_without_name)
-        )
+        mock_run.return_value = MagicMock(returncode=0, stdout=json.dumps(response_without_name))
 
         # Act
         result = client.list_claude_models()
@@ -155,10 +149,7 @@ def test_list_claude_models_access_denied_error(self, mock_run):
         # Act & Assert
         with pytest.raises(
             Exception,
-            match=(
-                "Access denied. Please check your AWS "
-                "permissions for Amazon Bedrock."
-            ),
+            match=("Access denied. Please check your AWS " "permissions for Amazon Bedrock."),
         ):
             client.list_claude_models()
 
@@ -172,10 +163,7 @@ def test_list_claude_models_not_authorized_error(self, mock_run):
         error = subprocess.CalledProcessError(
             1,
             "aws",
-            stderr=(
-                "The security token included in the request is invalid. "
-                "not authorized"
-            ),
+            stderr=("The security token included in the request is invalid. " "not authorized"),
         )
         mock_run.side_effect = error
 
@@ -227,9 +215,7 @@ def test_list_claude_models_unexpected_error(self, mock_run):
         mock_run.side_effect = RuntimeError("Unexpected runtime error")
 
         # Act & Assert
-        with pytest.raises(
-            Exception, match="Unexpected error: Unexpected runtime error"
-        ):
+        with pytest.raises(Exception, match="Unexpected error: Unexpected runtime error"):
             client.list_claude_models()
 
         mock_run.assert_called_once()
@@ -240,9 +226,7 @@ def test_list_claude_models_custom_region(self, mock_run, mock_aws_response):
         # Arrange
         custom_region = "eu-west-1"
         client = BedrockClient(custom_region)
-        mock_run.return_value = MagicMock(
-            returncode=0, stdout=json.dumps(mock_aws_response)
-        )
+        mock_run.return_value = MagicMock(returncode=0, stdout=json.dumps(mock_aws_response))
 
         # Act
         result = client.list_claude_models()
@@ -275,9 +259,7 @@ def test_list_claude_models_partial_data(self, mock_run):
                 }
             ]
         }
-        mock_run.return_value = MagicMock(
-            returncode=0, stdout=json.dumps(partial_response)
-        )
+        mock_run.return_value = MagicMock(returncode=0, stdout=json.dumps(partial_response))
 
         # Act
         result = client.list_claude_models()

tests/test_cli.py

@@ -168,9 +168,7 @@ def test_setup_custom_region(
         mock_config_class.return_value = mock_config
 
         # Act
-        result = self.runner.invoke(
-            setup, ["--region", "eu-west-1", "--non-interactive"]
-        )
+        result = self.runner.invoke(setup, ["--region", "eu-west-1", "--non-interactive"])
 
         # Assert
         assert result.exit_code == 0
@@ -334,9 +332,7 @@ def test_status_with_configuration(self, mock_config_class, mock_settings):
         assert "Current settings:" in result.output
         assert "CLAUDE_CODE_USE_BEDROCK: 1" in result.output
         assert "AWS_REGION: us-west-2" in result.output
-        assert (
-            "anthropic.claude-3-sonnet-20240229-v1:0" in result.output
-        )  # Extracted from ARN
+        assert "anthropic.claude-3-sonnet-20240229-v1:0" in result.output  # Extracted from ARN
         assert "Settings file: /test/.claude/settings.local.json" in result.output
         mock_config.load_settings.assert_called_once()
 
@@ -360,9 +356,7 @@ def test_status_arn_extraction(self, mock_config_class):
 
         # Assert
         assert result.exit_code == 0
-        assert (
-            "ANTHROPIC_MODEL: anthropic.claude-3-haiku-20240307-v1:0" in result.output
-        )
+        assert "ANTHROPIC_MODEL: anthropic.claude-3-haiku-20240307-v1:0" in result.output
 
     @patch.object(sys.modules["claude_setup.cli"], "ConfigManager")
     def test_status_simple_model_id(self, mock_config_class):

tests/test_config_manager.py

@@ -52,7 +52,6 @@ def test_save_settings_new_file(self, mock_file, mock_mkdir):
         # write was called
         assert mock_file().write.called
 
-
     @patch("builtins.open", new_callable=mock_open, read_data='{"key": "value"}')
     @patch("claude_setup.config_manager.Path.exists")
     def test_load_settings_legacy_format(self, mock_exists, mock_file):
@@ -68,7 +67,7 @@ def test_load_settings_legacy_format(self, mock_exists, mock_file):
         assert result == {"key": "value"}
         mock_exists.assert_called_once()
         mock_file.assert_called_once_with(config_manager.settings_path, "r")
-    
+
     @patch("builtins.open", new_callable=mock_open, read_data='{"env": {"key": "value"}}')
     @patch("claude_setup.config_manager.Path.exists")
     def test_load_settings_new_format(self, mock_exists, mock_file):
@@ -213,7 +212,7 @@ def test_load_settings_empty_file(self, mock_exists, mock_file):
         assert result == {}
         mock_exists.assert_called_once()
         mock_file.assert_called_once_with(config_manager.settings_path, "r")
-    
+
     @patch("builtins.open", new_callable=mock_open, read_data='{"env": {}}')
     @patch("claude_setup.config_manager.Path.exists")
     def test_load_settings_empty_env(self, mock_exists, mock_file):

tests/test_gitignore_manager.py

@@ -29,9 +29,9 @@ def test_ensure_gitignore_empty_file(self, mock_exists, mock_file):
         mock_file.assert_any_call(Path(".gitignore"), "w")
 
         # Check the written content
-        write_call = [
-            call for call in mock_file.return_value.write.call_args_list if call[0][0]
-        ][-1]
+        write_call = [call for call in mock_file.return_value.write.call_args_list if call[0][0]][
+            -1
+        ]
         written_content = write_call[0][0]
         assert ".claude/settings.local.json\n" == written_content
 
@@ -56,9 +56,9 @@ def test_ensure_gitignore_existing_content(self, mock_exists, mock_file):
         mock_file.assert_any_call(Path(".gitignore"), "w")
 
         # Check the written content includes existing and new patterns
-        write_call = [
-            call for call in mock_file.return_value.write.call_args_list if call[0][0]
-        ][-1]
+        write_call = [call for call in mock_file.return_value.write.call_args_list if call[0][0]][
+            -1
+        ]
         written_content = write_call[0][0]
         assert "node_modules/\n*.log\n.claude/settings.local.json\n" == written_content
 
@@ -118,9 +118,9 @@ def test_ensure_gitignore_whitespace_only_file(self, mock_exists, mock_file):
         mock_file.assert_any_call(Path(".gitignore"), "w")
 
         # Check the written content
-        write_call = [
-            call for call in mock_file.return_value.write.call_args_list if call[0][0]
-        ][-1]
+        write_call = [call for call in mock_file.return_value.write.call_args_list if call[0][0]][
+            -1
+        ]
         written_content = write_call[0][0]
         assert ".claude/settings.local.json\n" == written_content
 
@@ -130,9 +130,7 @@ def test_ensure_gitignore_whitespace_only_file(self, mock_exists, mock_file):
         read_data="node_modules/\n# Comment\n\n*.log\n",
     )
     @patch("claude_setup.gitignore_manager.Path.exists")
-    def test_ensure_gitignore_with_comments_and_empty_lines(
-        self, mock_exists, mock_file
-    ):
+    def test_ensure_gitignore_with_comments_and_empty_lines(self, mock_exists, mock_file):
         """Test handling .gitignore with comments and empty lines."""
         # Arrange
         mock_exists.return_value = True
@@ -145,9 +143,9 @@ def test_ensure_gitignore_with_comments_and_empty_lines(
         assert mock_file.call_count == 2
 
         # Check the written content preserves structure
-        write_call = [
-            call for call in mock_file.return_value.write.call_args_list if call[0][0]
-        ][-1]
+        write_call = [call for call in mock_file.return_value.write.call_args_list if call[0][0]][
+            -1
+        ]
         written_content = write_call[0][0]
         expected = "node_modules/\n# Comment\n\n*.log\n.claude/settings.local.json\n"
         assert expected == written_content
@@ -209,9 +207,7 @@ def test_ensure_gitignore_similar_pattern_exists(self, mock_exists, mock_file):
         read_data="node_modules/\n.claude/settings.local.json  \n*.log\n",
     )
     @patch("claude_setup.gitignore_manager.Path.exists")
-    def test_ensure_gitignore_pattern_with_trailing_spaces(
-        self, mock_exists, mock_file
-    ):
+    def test_ensure_gitignore_pattern_with_trailing_spaces(self, mock_exists, mock_file):
         """Test that trailing spaces in existing pattern don't match."""
         # Arrange
         mock_exists.return_value = True
@@ -227,9 +223,9 @@ def test_ensure_gitignore_pattern_with_trailing_spaces(
 
         # Check that the pattern was added despite similar line
         # with trailing spaces
-        write_call = [
-            call for call in mock_file.return_value.write.call_args_list if call[0][0]
-        ][-1]
+        write_call = [call for call in mock_file.return_value.write.call_args_list if call[0][0]][
+            -1
+        ]
         written_content = write_call[0][0]
         assert ".claude/settings.local.json" in written_content
         # Should have both the original line with spaces and the new clean line
@@ -270,8 +266,8 @@ def test_ensure_gitignore_no_trailing_newline(self, mock_exists, mock_file):
         assert mock_file.call_count == 2
 
         # Check the written content adds pattern properly
-        write_call = [
-            call for call in mock_file.return_value.write.call_args_list if call[0][0]
-        ][-1]
+        write_call = [call for call in mock_file.return_value.write.call_args_list if call[0][0]][
+            -1
+        ]
         written_content = write_call[0][0]
         assert "line1\nline2\nline3\n.claude/settings.local.json\n" == written_content

tests/test_integration.py

@@ -36,9 +36,7 @@ def test_complete_setup_workflow(
         """Test complete setup workflow from start to finish."""
         # Arrange
         mock_auth.return_value = True
-        mock_subprocess.return_value = MagicMock(
-            returncode=0, stdout=json.dumps(mock_aws_response)
-        )
+        mock_subprocess.return_value = MagicMock(returncode=0, stdout=json.dumps(mock_aws_response))
 
         with self.runner.isolated_filesystem():
             # Act - Run setup
@@ -78,30 +76,22 @@ def test_complete_setup_workflow(
 
     @patch.object(sys.modules["claude_setup.auth_checker"], "check_aws_auth")
     @patch("claude_setup.aws_client.subprocess.run")
-    def test_setup_with_different_regions(
-        self, mock_subprocess, mock_auth, mock_aws_response
-    ):
+    def test_setup_with_different_regions(self, mock_subprocess, mock_auth, mock_aws_response):
         """Test setup workflow with different AWS regions."""
         # Arrange
         mock_auth.return_value = True
-        mock_subprocess.return_value = MagicMock(
-            returncode=0, stdout=json.dumps(mock_aws_response)
-        )
+        mock_subprocess.return_value = MagicMock(returncode=0, stdout=json.dumps(mock_aws_response))
 
         regions = ["us-east-1", "eu-west-1", "ap-southeast-1"]
 
         for region in regions:
             with self.runner.isolated_filesystem():
                 # Act
-                result = self.runner.invoke(
-                    cli, ["setup", "--region", region, "--non-interactive"]
-                )
+                result = self.runner.invoke(cli, ["setup", "--region", region, "--non-interactive"])
 
                 # Assert
                 assert result.exit_code == 0
-                assert (
-                    f"Fetching available Claude models from {region}" in result.output
-                )
+                assert f"Fetching available Claude models from {region}" in result.output
 
                 # Verify region in configuration
                 config_manager = ConfigManager()
@@ -275,10 +265,7 @@ def test_bedrock_api_error_workflow(self, mock_auth, mock_subprocess):
                 # Assert - The exception from BedrockClient causes the CLI to exit
                 assert result.exit_code != 0
                 # The exception message should appear in the result
-                assert (
-                    "Access denied" in str(result.exception)
-                    or "Access denied" in result.output
-                )
+                assert "Access denied" in str(result.exception) or "Access denied" in result.output
 
                 # Verify no configuration was created
                 config_manager = ConfigManager()

tests/test_utils.py

@@ -150,9 +150,7 @@ class CLITestHelper:
     """Helper class for CLI testing."""
 
     @staticmethod
-    def run_cli_command(
-        runner, command, args=None, input_data=None, catch_exceptions=True
-    ):
+    def run_cli_command(runner, command, args=None, input_data=None, catch_exceptions=True):
         """Run a CLI command with standard error handling.
 
         Args: