test_selection_logic.py

#!/usr/bin/env python3
"""
Test script to verify the updated commit/branch selection logic.
"""

import os
from git_repository_manager import GitRepository, GitCommit, GitBranch

def test_selection_logic():
    """Test the different selection behaviors"""
    
    # Test with the current repository
    current_repo_path = os.getcwd()
    
    print(f"Testing selection logic with repository: {current_repo_path}")
    
    # Create a GitRepository instance
    repo = GitRepository(current_repo_path)
    
    print("\n1. Testing repository selection (should show commits):")
    try:
        current_branch, commits = repo.get_current_branch_commits(5)
        print(f"   Current branch: {current_branch}")
        print(f"   Found {len(commits)} commits:")
        
        for i, commit in enumerate(commits, 1):
            print(f"   {i}. {commit.commit_sha[:8]} - {commit.message[:50]}")
            if commit.branch_names:
                print(f"      Branch heads: {commit.get_branch_names_display()}")
    except Exception as e:
        print(f"   Error: {e}")
    
    print("\n2. Testing local branch selection (should show branches):")
    try:
        repo.load_branches()
        print(f"   Found {len(repo.local_branches)} local branches:")
        
        for i, branch in enumerate(repo.local_branches[:5], 1):
            print(f"   {i}. {branch.name} - {branch.commit_sha[:8]}")
    except Exception as e:
        print(f"   Error: {e}")
    
    print("\n3. Testing remote branch selection (should show branches):")
    try:
        print(f"   Found {len(repo.remote_branches)} remote branches:")
        
        for i, branch in enumerate(repo.remote_branches[:5], 1):
            print(f"   {i}. {branch.name} - {branch.commit_sha[:8]}")
    except Exception as e:
        print(f"   Error: {e}")

if __name__ == "__main__":
    test_selection_logic()