Class TestMainFunction

claudecode/test_main_function.py:15–512

Test main function execution flow.

class TestMainFunction:
    """Test main function execution flow."""
    
    def test_main_missing_environment_vars(self, capsys):
        """Test main with missing environment variables."""
        with patch.dict(os.environ, {}, clear=True):
            with pytest.raises(SystemExit) as exc_info:
                main()
            
            assert exc_info.value.code == 2  # EXIT_CONFIGURATION_ERROR
            
            captured = capsys.readouterr()
            output = json.loads(captured.out)
            assert 'error' in output
            assert 'GITHUB_REPOSITORY' in output['error']
    
    def test_main_missing_pr_number(self, capsys):
        """Test main with missing PR number."""
        with patch.dict(os.environ, {'GITHUB_REPOSITORY': 'owner/repo'}, clear=True):
            with pytest.raises(SystemExit) as exc_info:
                main()
            
            assert exc_info.value.code == 2  # EXIT_CONFIGURATION_ERROR
            
            captured = capsys.readouterr()
            output = json.loads(captured.out)
            assert 'PR_NUMBER' in output['error']
    
    def test_main_invalid_pr_number(self, capsys):
        """Test main with invalid PR number."""
        with patch.dict(os.environ, {
            'GITHUB_REPOSITORY': 'owner/repo',
            'PR_NUMBER': 'not-a-number'
        }, clear=True):
            with pytest.raises(SystemExit) as exc_info:
                main()
            
            assert exc_info.value.code == 2  # EXIT_CONFIGURATION_ERROR
            
            captured = capsys.readouterr()
            output = json.loads(captured.out)
            assert 'Invalid PR_NUMBER' in output['error']
    
    @patch('claudecode.github_action_audit.GitHubActionClient')
    def test_main_github_client_init_failure(self, mock_client_class, capsys):
        """Test main when GitHub client initialization fails."""
        mock_client_class.side_effect = Exception("Token invalid")
        
        with patch.dict(os.environ, {
            'GITHUB_REPOSITORY': 'owner/repo',
            'PR_NUMBER': '123',
            'GITHUB_TOKEN': 'invalid'
        }):
            with pytest.raises(SystemExit) as exc_info:
                main()
            
            assert exc_info.value.code == 2  # EXIT_CONFIGURATION_ERROR