test_base_module.py

import sys
import os

# Add backend to path
sys.path.append(os.path.join(os.getcwd(), 'backend'))

from services.github_scraper import GitHubScraper

def test_parsing():
    scraper = GitHubScraper()
    print("Testing URL parsing...")
    
    test_cases = [
        ("https://github.com/odoo/odoo/tree/18.0/odoo/addons/base", ("odoo/addons", "base", "")),
        ("https://github.com/odoo/odoo/tree/18.0/addons/account", ("addons", "account", "")),
        ("account", ("addons", "account", "")),
    ]
    
    for url, expected in test_cases:
        result = scraper.parse_github_url(url)
        print(f"URL: {url} -> {result}")
        assert result == expected, f"Expected {expected}, got {result}"
    print("Parsing tests passed!")

def test_base_listing():
    scraper = GitHubScraper()
    print("\nTesting listing of 'base' module...")
    
    # Use parent_path='odoo/addons'
    files, actual_parent = scraper.get_folder_files("odoo/addons", "base")
    print(f"Found {len(files)} files in 'base' using parent 'odoo/addons'")
    assert actual_parent == "odoo/addons"
    assert len(files) > 0
    
    # Test fallback
    print("Testing fallback listing for 'base'...")
    files, actual_parent = scraper.get_folder_files("addons", "base")
    print(f"Found {len(files)} files in 'base' using fallback from 'addons' to '{actual_parent}'")
    assert actual_parent == "odoo/addons"
    assert len(files) > 0
    print("Listing tests passed!")

def test_content_fetch():
    scraper = GitHubScraper()
    print("\nTesting content fetch for a single file in 'base'...")
    # Fetch just one file
    parent_path = "odoo/addons"
    addon_name = "base"
    file_path = "models/res_partner.py"
    
    content = scraper.get_file_content(parent_path, addon_name, file_path)
    print(f"Fetched {len(content)} characters from {file_path}")
    assert len(content) > 0
    assert "class Partner" in content
    print("Content fetch test passed!")

if __name__ == "__main__":
    try:
        test_parsing()
        test_base_listing()
        test_content_fetch()
        print("\nALL TESTS PASSED!")
    except Exception as e:
        print(f"\nTEST FAILED: {e}")
        import traceback
        traceback.print_exc()