import os
import sys
from dotenv import load_dotenv

# Add parent dir to path so we can import backend packages
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from backend.mega_api import MegaMarketAPI
from backend.mega_api.exceptions import MegaAPIError

def test_mega_api():
    # Load environment variables
    env_path = os.path.join(os.path.dirname(__file__), '.env')
    load_dotenv(env_path)
    
    # Check if key is loaded
    api_token = os.getenv('MEGA_API_TOKEN')
    
    if not api_token:
        print("Error: MEGA_API_TOKEN must be in backend/.env")
        return
        
    print("Initializing MegaMarketAPI...")

    api = MegaMarketAPI()
    
    try:
        # Test 1: Get Orders
        print("\n--- Testing Orders API ---")
        orders_response = api.orders.get_new_orders(limit=5)
        print("Success! Response keys:", orders_response.keys())
        
        # Usually wrapped in data
        data = orders_response.get("data", {})
        orders = data.get("orders", [])
        print(f"Found {len(orders)} new orders.")
            
    except MegaAPIError as e:
        print(f"API Error: {e}")
    except Exception as e:
        print(f"Unexpected Error: {e}")

if __name__ == '__main__':
    test_mega_api()