risk-engine.test.js

/**
 * Unit Tests for HeatShe Risk Engine
 * Run with: node risk-engine.test.js
 */

// Load the risk engine
const RiskEngine = require('./risk-engine.js');

// Test utilities
let testsPassed = 0;
let testsFailed = 0;

function assert(condition, message) {
    if (condition) {
        testsPassed++;
        console.log(`✓ ${message}`);
    } else {
        testsFailed++;
        console.log(`✗ ${message}`);
    }
}

function assertEquals(actual, expected, message) {
    assert(actual === expected, `${message} (expected ${expected}, got ${actual})`);
}

function assertApprox(actual, expected, margin, message) {
    const diff = Math.abs(actual - expected);
    assert(diff <= margin, `${message} (expected ~${expected}, got ${actual}, margin ${margin})`);
}

// ─── TEST SUITE ───
console.log('\n🧪 HeatShe Risk Engine Test Suite\n');

// Test 1: Validation
console.log('📋 Validation Tests');
let result = RiskEngine.validate({
    district: 50,
    occupation: 75,
    hours: 80,
    peakSun: 90,
    age: 30,
    pregnancy: 0,
    temperature: 5
});
assert(result.valid === true, 'Valid inputs should pass');

result = RiskEngine.validate({
    district: 150, // Invalid
    occupation: 75,
    hours: 80,
    peakSun: 90,
    age: 30,
    pregnancy: 0,
    temperature: 5
});
assert(result.valid === false, 'Invalid district risk (>100) should fail');

result = RiskEngine.validate({
    district: 50,
    // Missing occupation
    hours: 80,
    peakSun: 90,
    age: 30,
    pregnancy: 0,
    temperature: 5
});
assert(result.valid === false, 'Missing required field should fail');

// Test 2: Formula Calculation
console.log('\n🔢 Formula Calculation Tests');
result = RiskEngine.calculate({
    district: 50,
    occupation: 80,
    hours: 60,
    peakSun: 70,
    age: 20,
    pregnancy: 0,
    temperature: 0
});
assert(result.valid === true, 'Calculation should succeed');
// Expected: (0.16*50 + 0.30*80 + 0.20*60 + 0.20*70 + 0.09*20 + 0.05*0) * 10 + 0
// = (8 + 24 + 12 + 14 + 1.8 + 0) * 10 + 0 = 59.8 * 10 + 0 = 598 / 10 ≈ 60
assertApprox(result.finalScore, 61, 2, 'Formula calculation should match expected value');

// Test 3: Risk Categories
console.log('\n🎯 Risk Category Tests');
result = RiskEngine.calculate({
    district: 10, occupation: 20, hours: 15, peakSun: 10, age: 5, pregnancy: 0, temperature: -5
});
assertEquals(result.category.key, 'safe', 'Low score should be Safe');

result = RiskEngine.calculate({
    district: 50, occupation: 50, hours: 50, peakSun: 50, age: 50, pregnancy: 0, temperature: 5
});
assertEquals(result.category.key, 'caution', 'Medium score should be Caution');

result = RiskEngine.calculate({
    district: 70, occupation: 80, hours: 75, peakSun: 85, age: 60, pregnancy: 0, temperature: 10
});
assert(['high', 'critical'].includes(result.category.key), 'High score should be High Risk or Critical');

result = RiskEngine.calculate({
    district: 100, occupation: 100, hours: 100, peakSun: 100, age: 100, pregnancy: 100, temperature: 20
});
assertEquals(result.category.key, 'critical', 'Maximum inputs should be Critical');

// Test 4: Temperature Impact
console.log('\n🌡️ Temperature Impact Tests');
result1 = RiskEngine.calculate({
    district: 50, occupation: 50, hours: 50, peakSun: 50, age: 50, pregnancy: 0, temperature: 0
});
result2 = RiskEngine.calculate({
    district: 50, occupation: 50, hours: 50, peakSun: 50, age: 50, pregnancy: 0, temperature: 10
});
assert(result2.finalScore > result1.finalScore, 'Higher temperature should increase risk');
assertEquals(result2.finalScore - result1.finalScore, 10, 'Temperature should add directly to score');

// Test 5: Top Factors Identification
console.log('\n🏆 Top Factors Identification Tests');
result = RiskEngine.calculate({
    district: 20, occupation: 95, hours: 85, peakSun: 90, age: 10, pregnancy: 0, temperature: 0
});
assert(result.topFactors.length === 3, 'Should return top 3 factors');
assert(result.topFactors[0].name === 'occupation', 'Occupation should be top factor');
assert(result.topFactors[1].name === 'peakSun', 'Peak sun should be second');
assert(result.topFactors[2].name === 'hours', 'Hours should be third');

// Test 6: Pregnancy Impact
console.log('\n🤰 Pregnancy Status Impact Tests');
result1 = RiskEngine.calculate({
    district: 50, occupation: 50, hours: 50, peakSun: 50, age: 35, pregnancy: 0, temperature: 0
});
result2 = RiskEngine.calculate({
    district: 50, occupation: 50, hours: 50, peakSun: 50, age: 35, pregnancy: 100, temperature: 0
});
assert(result2.finalScore > result1.finalScore, 'Pregnancy should increase risk');

// Test 7: Age Impact
console.log('\n🎂 Age Factor Tests');
result1 = RiskEngine.calculate({
    district: 50, occupation: 50, hours: 50, peakSun: 50, age: 20, pregnancy: 0, temperature: 0
});
result2 = RiskEngine.calculate({
    district: 50, occupation: 50, hours: 50, peakSun: 50, age: 80, pregnancy: 0, temperature: 0
});
assert(result2.finalScore > result1.finalScore, 'Higher age should increase risk');

// Test 8: Explanation Generation
console.log('\n📝 Explanation Generation Tests');
result = RiskEngine.calculate({
    district: 20, occupation: 20, hours: 20, peakSun: 20, age: 20, pregnancy: 0, temperature: 0
});
assert(result.explanation !== '', 'Should generate explanation');
assert(result.explanation.includes('LOW') || result.explanation.includes('Safe'), 'Safe score should mention low risk');

// Test 9: Multilingual Support
console.log('\n🌐 Multilingual Support Tests');
result_en = RiskEngine.calculate({
    district: 50, occupation: 50, hours: 50, peakSun: 50, age: 50, pregnancy: 0, temperature: 0
}, 'en');
result_kn = RiskEngine.calculate({
    district: 50, occupation: 50, hours: 50, peakSun: 50, age: 50, pregnancy: 0, temperature: 0
}, 'kn');
assert(result_en.explanation !== result_kn.explanation, 'English and Kannada explanations should differ');
assert(result_en.category.labelKn !== '', 'Kannada category label should be present');

// Test 10: Recommendations
console.log('\n💡 Recommendations Tests');
result = RiskEngine.calculate({
    district: 100, occupation: 100, hours: 100, peakSun: 100, age: 100, pregnancy: 100, temperature: 20
});
assert(result.recommendations.length > 0, 'Should generate recommendations');
assert(result.recommendations[0].action !== '', 'Recommendations should have action text');

// Test 11: Boundary Cases
console.log('\n🚨 Boundary Case Tests');
result = RiskEngine.calculate({
    district: 0, occupation: 0, hours: 0, peakSun: 0, age: 0, pregnancy: 0, temperature: -10
});
assert(result.finalScore >= 0, 'Score should never be negative');

result = RiskEngine.calculate({
    district: 100, occupation: 100, hours: 100, peakSun: 100, age: 100, pregnancy: 100, temperature: 40
});
assert(result.finalScore <= 100, 'Score should never exceed 100');

// Test 12: Weights
console.log('\n⚖️ Weights Verification Tests');
const weights = RiskEngine.getWeights();
const weightSum = Object.values(weights).reduce((a, b) => a + b, 0);
assertApprox(weightSum, 1.0, 0.01, 'Weights should sum to 1.0');

// ─── TEST SUMMARY ───
console.log(`\n${'═'.repeat(50)}`);
console.log(`✅ Tests Passed: ${testsPassed}`);
console.log(`❌ Tests Failed: ${testsFailed}`);
console.log(`📊 Total: ${testsPassed + testsFailed}`);
console.log(`${'═'.repeat(50)}\n`);

if (testsFailed === 0) {
    console.log('🎉 All tests passed! Risk engine is production-ready.\n');
} else {
    console.log(`⚠️ ${testsFailed} test(s) failed. Please review.\n`);
}