test-image-generation.js

// Test script for image generation
// Run: node test-image-generation.js

const API_KEY = process.env.OPENROUTER_API_KEY || 'your-api-key-here';
const BACKEND_URL = 'http://localhost:5000';

async function testImageGeneration() {
  console.log('🧪 Testing Image Generation with OpenRouter GPT-5 Image Mini\n');
  
  // Test 1: Health check
  console.log('Test 1: Health Check');
  try {
    const healthResponse = await fetch(`${BACKEND_URL}/api/health`);
    const healthData = await healthResponse.json();
    console.log('✅ Health check:', healthData);
  } catch (error) {
    console.error('❌ Health check failed:', error.message);
    return;
  }
  
  // Test 2: Image generation with valid API key
  console.log('\nTest 2: Image Generation (with API key)');
  try {
    const response = await fetch(`${BACKEND_URL}/api/generate-image`, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
        'X-OpenRouter-API-Key': API_KEY
      },
      body: JSON.stringify({
        prompt: 'A beautiful sunset over mountains',
        aspectRatio: '1:1'
      })
    });
    
    const data = await response.json();
    
    if (!response.ok) {
      console.error('❌ Image generation failed:', JSON.stringify(data, null, 2));
      return;
    }
    
    if (data.generatedImages && data.generatedImages.length > 0) {
      const image = data.generatedImages[0];
      if (image.image?.imageBytes) {
        console.log('✅ Image generated successfully!');
        console.log('   Image size (base64):', image.image.imageBytes.length, 'characters');
      } else if (image.imageUrl) {
        console.log('✅ Image URL received:', image.imageUrl);
      } else {
        console.error('❌ Unexpected response format:', JSON.stringify(data, null, 2));
      }
    } else {
      console.error('❌ No images in response:', JSON.stringify(data, null, 2));
    }
  } catch (error) {
    console.error('❌ Image generation error:', error.message);
  }
  
  // Test 3: Different aspect ratios
  console.log('\nTest 3: Different Aspect Ratios');
  const aspectRatios = ['1:1', '16:9', '9:16'];
  
  for (const aspectRatio of aspectRatios) {
    console.log(`\nTesting aspect ratio: ${aspectRatio}`);
    try {
      const response = await fetch(`${BACKEND_URL}/api/generate-image`, {
        method: 'POST',
        headers: {
          'Content-Type': 'application/json',
          'X-OpenRouter-API-Key': API_KEY
        },
        body: JSON.stringify({
          prompt: `Test image for ${aspectRatio} aspect ratio`,
          aspectRatio: aspectRatio
        })
      });
      
      const data = await response.json();
      
      if (response.ok && data.generatedImages && data.generatedImages.length > 0) {
        console.log(`✅ ${aspectRatio} - Success`);
      } else {
        console.error(`❌ ${aspectRatio} - Failed:`, data.error?.message || 'Unknown error');
      }
    } catch (error) {
      console.error(`❌ ${aspectRatio} - Error:`, error.message);
    }
  }
  
  // Test 4: Error handling (no API key)
  console.log('\nTest 4: Error Handling (no API key)');
  try {
    const response = await fetch(`${BACKEND_URL}/api/generate-image`, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json'
      },
      body: JSON.stringify({
        prompt: 'Test without API key',
        aspectRatio: '1:1'
      })
    });
    
    const data = await response.json();
    
    if (!response.ok && data.error) {
      console.log('✅ Error handling works:', data.error.message);
    } else {
      console.error('❌ Expected error but got:', JSON.stringify(data, null, 2));
    }
  } catch (error) {
    console.error('❌ Error handling test failed:', error.message);
  }
  
  // Test 5: Empty prompt
  console.log('\nTest 5: Empty Prompt Validation');
  try {
    const response = await fetch(`${BACKEND_URL}/api/generate-image`, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
        'X-OpenRouter-API-Key': API_KEY
      },
      body: JSON.stringify({
        prompt: '',
        aspectRatio: '1:1'
      })
    });
    
    const data = await response.json();
    
    if (!response.ok && response.status === 400) {
      console.log('✅ Empty prompt validation works:', data.error?.message);
    } else {
      console.error('❌ Expected validation error but got:', JSON.stringify(data, null, 2));
    }
  } catch (error) {
    console.error('❌ Validation test failed:', error.message);
  }
  
  console.log('\n✅ All tests completed!');
}

// Run tests
testImageGeneration().catch(console.error);