test_backend_empty_folder.py

import requests
import json

def test_extraction_empty_folder():
    url = "http://localhost:8000/api/extract"
    # Test case: Empty folder path, but valid file path with addon name
    payload = {
        "folderPath": "",
        "filePaths": ["account/models/account_move.py"]
    }
    
    try:
        print(f"Sending request to {url} with empty folderPath...")
        response = requests.post(url, json=payload, timeout=30)
        print(f"Status Code: {response.status_code}")
        
        if response.status_code == 200:
            data = response.json()
            if data.get('success'):
                print(f"Success! Extracted {len(data['data'])} files.")
                first_file = data['data'][0]
                print(f"Folder Name: {first_file['folderName']}")
                print(f"Path: {first_file['path']}")
                
                if first_file['folderName'] == 'account':
                    print("PASS: Correctly inferred folder name.")
                else:
                    print(f"FAIL: Incorrect folder name. Expected 'account', got '{first_file['folderName']}'")
            else:
                print(f"Failed: {data.get('error')}")
        else:
            print(f"Error: {response.text}")
            
    except Exception as e:
        print(f"Exception: {e}")

if __name__ == "__main__":
    test_extraction_empty_folder()