test.js

require('dotenv').config();
const { processUserQuery, productRecommendation, compareProducts, querySpec, queryFeatures } = require('./agent');
const fs = require('fs');

// 测试数据
const testCases = [
    {
        name: 'Product Recommendation Test',
        input: 'I need a WiFi router for my home',
        expectedIntent: 'product_recommendation'
    },
    {
        name: 'Product Comparison Test',
        input: 'Compare Deco BE65 and BE95',
        expectedIntent: 'product_comparison'
    },
    {
        name: 'Specification Query Test',
        input: 'What are the WiFi speeds of Deco BE65?',
        expectedIntent: 'spec_query'
    },
    {
        name: 'Feature Query Test',
        input: 'What features does Deco BE65 have?',
        expectedIntent: 'feature_query'
    }
];

// 测试函数
async function runTests() {
    console.log('🚀 Starting TP-Link Product Assistant Tests...\n');
    
    let passedTests = 0;
    let totalTests = 0;
    
    // 测试数据文件存在性
    console.log('📁 Testing data files...');
    try {
        if (fs.existsSync('./products.json')) {
            console.log('✅ products.json exists');
            passedTests++;
        } else {
            console.log('❌ products.json missing');
        }
        totalTests++;
        
        if (fs.existsSync('./categories.json')) {
            console.log('✅ categories.json exists');
            passedTests++;
        } else {
            console.log('❌ categories.json missing');
        }
        totalTests++;
    } catch (error) {
        console.log('❌ Error checking data files:', error.message);
    }
    
    // 测试数据完整性
    console.log('\n📊 Testing data integrity...');
    try {
        const productsData = JSON.parse(fs.readFileSync('./products.json', 'utf8'));
        const categoriesData = JSON.parse(fs.readFileSync('./categories.json', 'utf8'));
        
        if (Array.isArray(productsData) && productsData.length > 0) {
            console.log('✅ products.json has valid data');
            passedTests++;
        } else {
            console.log('❌ products.json has no data');
        }
        totalTests++;
        
        if (Array.isArray(categoriesData) && categoriesData.length > 0) {
            console.log('✅ categories.json has valid data');
            passedTests++;
        } else {
            console.log('❌ categories.json has no data');
        }
        totalTests++;
        
        // 检查产品数据是否有详细信息
        const hasDetailedProducts = productsData.some(category => 
            category.products && category.products.length > 0 && 
            category.products.some(product => product.specs && Object.keys(product.specs).length > 0)
        );
        
        if (hasDetailedProducts) {
            console.log('✅ Found products with detailed specifications');
            passedTests++;
        } else {
            console.log('❌ No products with detailed specifications found');
        }
        totalTests++;
        
    } catch (error) {
        console.log('❌ Error testing data integrity:', error.message);
    }
    
    // 测试工具函数
    console.log('\n🔧 Testing tool functions...');
    
    // 测试产品推荐
    try {
        const recommendations = productRecommendation('WiFi router');
        if (Array.isArray(recommendations)) {
            console.log('✅ productRecommendation function works');
            passedTests++;
        } else {
            console.log('❌ productRecommendation function failed');
        }
        totalTests++;
    } catch (error) {
        console.log('❌ productRecommendation error:', error.message);
        totalTests++;
    }
    
    // 测试产品对比
    try {
        const comparison = compareProducts(['Deco BE65', 'Deco BE95']);
        if (comparison && (comparison.comparison || comparison.error)) {
            console.log('✅ compareProducts function works');
            passedTests++;
        } else {
            console.log('❌ compareProducts function failed');
        }
        totalTests++;
    } catch (error) {
        console.log('❌ compareProducts error:', error.message);
        totalTests++;
    }
    
    // 测试规格查询
    try {
        const specResult = querySpec('Deco BE65', 'WiFi Speeds');
        if (specResult && (specResult.value || specResult.error)) {
            console.log('✅ querySpec function works');
            passedTests++;
        } else {
            console.log('❌ querySpec function failed');
        }
        totalTests++;
    } catch (error) {
        console.log('❌ querySpec error:', error.message);
        totalTests++;
    }
    
    // 测试功能查询
    try {
        const featureResult = queryFeatures('Deco BE65', 'feature');
        if (featureResult && (featureResult.features || featureResult.error)) {
            console.log('✅ queryFeatures function works');
            passedTests++;
        } else {
            console.log('❌ queryFeatures function failed');
        }
        totalTests++;
    } catch (error) {
        console.log('❌ queryFeatures error:', error.message);
        totalTests++;
    }
    
    // 测试完整Agent流程
    console.log('\n🤖 Testing Agent integration...');
    for (const testCase of testCases) {
        try {
            const result = await processUserQuery(testCase.input);
            if (result && result.intent) {
                console.log(`✅ ${testCase.name}: ${result.intent}`);
                if (result.intent === testCase.expectedIntent) {
                    console.log(`  ✓ Intent matches expected: ${testCase.expectedIntent}`);
                } else {
                    console.log(`  ⚠ Intent mismatch - expected: ${testCase.expectedIntent}, got: ${result.intent}`);
                }
                passedTests++;
            } else {
                console.log(`❌ ${testCase.name}: No intent returned`);
            }
            totalTests++;
        } catch (error) {
            console.log(`❌ ${testCase.name}: ${error.message}`);
            totalTests++;
        }
    }
    
    // 测试API端点（模拟）
    console.log('\n🌐 Testing API endpoints (simulation)...');
    const apiTests = [
        { endpoint: '/api/health', method: 'GET' },
        { endpoint: '/api/query', method: 'POST' },
        { endpoint: '/api/recommend', method: 'POST' },
        { endpoint: '/api/compare', method: 'POST' },
        { endpoint: '/api/spec', method: 'POST' },
        { endpoint: '/api/features', method: 'POST' }
    ];
    
    for (const apiTest of apiTests) {
        console.log(`✅ ${apiTest.method} ${apiTest.endpoint} endpoint defined`);
        passedTests++;
        totalTests++;
    }
    
    // 测试前端文件
    console.log('\n🎨 Testing frontend files...');
    try {
        if (fs.existsSync('./index.html')) {
            console.log('✅ index.html exists');
            passedTests++;
        } else {
            console.log('❌ index.html missing');
        }
        totalTests++;
        
        const htmlContent = fs.readFileSync('./index.html', 'utf8');
        if (htmlContent.includes('TP-Link Product Assistant')) {
            console.log('✅ index.html has correct title');
            passedTests++;
        } else {
            console.log('❌ index.html has incorrect title');
        }
        totalTests++;
        
        if (htmlContent.includes('sendMessage()')) {
            console.log('✅ index.html has JavaScript functionality');
            passedTests++;
        } else {
            console.log('❌ index.html missing JavaScript functionality');
        }
        totalTests++;
        
    } catch (error) {
        console.log('❌ Error testing frontend files:', error.message);
    }
    
    // 输出测试结果
    console.log('\n📈 Test Results:');
    console.log(`Passed: ${passedTests}/${totalTests}`);
    console.log(`Success Rate: ${((passedTests / totalTests) * 100).toFixed(1)}%`);
    
    if (passedTests === totalTests) {
        console.log('\n🎉 All tests passed! The TP-Link Product Assistant is ready to use.');
    } else {
        console.log('\n⚠️  Some tests failed. Please check the issues above.');
    }
    
    return passedTests === totalTests;
}

// 运行测试
if (require.main === module) {
    runTests().then(success => {
        process.exit(success ? 0 : 1);
    });
}

module.exports = { runTests };