testModels.js

/**
 * Test script to verify Database Models
 * Run this file to test Department and Employee models
 * 
 * Usage: node testModels.js
 */

require('dotenv').config();
const mongoose = require('mongoose');
const { Department, Employee } = require('./models');

// Connect to MongoDB
const connectDB = async () => {
  try {
    await mongoose.connect(process.env.MONGODB_URI);
    console.log('✅ Connected to MongoDB');
  } catch (error) {
    console.error('❌ MongoDB connection error:', error.message);
    process.exit(1);
  }
};

// Test Department Model
const testDepartmentModel = async () => {
  console.log('\n📊 Testing Department Model...\n');
  
  try {
    // Create a test department
    const department = new Department({
      name: 'IT Department',
      description: 'Information Technology and Software Development'
    });
    
    await department.save();
    console.log('✅ Department created successfully:');
    console.log(JSON.stringify(department, null, 2));
    
    // Find the department
    const foundDept = await Department.findOne({ name: 'IT Department' });
    console.log('\n✅ Department found:', foundDept.name);
    
    // Test static method
    const activeDepts = await Department.findActive();
    console.log(`✅ Active departments count: ${activeDepts.length}`);
    
    return department;
  } catch (error) {
    console.error('❌ Department test error:', error.message);
    throw error;
  }
};

// Test Employee Model
const testEmployeeModel = async (department) => {
  console.log('\n👤 Testing Employee Model...\n');
  
  try {
    // Create a test employee
    const employee = new Employee({
      firstName: 'john',
      lastName: 'doe',
      email: 'john.doe@company.com',
      phone: '123-456-7890',
      jobTitle: 'Software Engineer',
      department: department._id,
      country: 'United States',
      state: 'California',
      city: 'San Francisco',
      address: '123 Tech Street',
      zipCode: '94102',
      salary: 75000,
      gender: 'Male'
    });
    
    await employee.save();
    console.log('✅ Employee created successfully:');
    console.log(JSON.stringify(employee, null, 2));
    
    // Test virtual field
    console.log('\n✅ Full Name (virtual):', employee.fullName);
    
    // Find employee with populated department
    const foundEmployee = await Employee.findById(employee._id)
      .populate('department', 'name description');
    console.log('✅ Employee found with department:', 
      foundEmployee.fullName, 
      '→', 
      foundEmployee.department.name
    );
    
    // Create a supervisor
    const supervisor = new Employee({
      firstName: 'jane',
      lastName: 'smith',
      email: 'jane.smith@company.com',
      phone: '123-456-7891',
      jobTitle: 'Senior Manager',
      department: department._id,
      country: 'United States',
      state: 'California',
      city: 'San Francisco',
      salary: 95000,
      gender: 'Female'
    });
    
    await supervisor.save();
    console.log('\n✅ Supervisor created:', supervisor.fullName);
    
    // Assign supervisor to employee
    employee.supervisor = supervisor._id;
    await employee.save();
    console.log('✅ Supervisor assigned to employee');
    
    // Test populated query
    const employeeWithSupervisor = await Employee.findById(employee._id)
      .populate('department', 'name')
      .populate('supervisor', 'firstName lastName jobTitle');
    
    console.log('\n✅ Employee with supervisor:');
    console.log(`   Employee: ${employeeWithSupervisor.fullName}`);
    console.log(`   Supervisor: ${employeeWithSupervisor.supervisor.firstName} ${employeeWithSupervisor.supervisor.lastName}`);
    console.log(`   Department: ${employeeWithSupervisor.department.name}`);
    
    // Test static methods
    const potentialSupervisors = await Employee.getPotentialSupervisors(employee._id);
    console.log(`\n✅ Potential supervisors count: ${potentialSupervisors.length}`);
    
    return { employee, supervisor };
  } catch (error) {
    console.error('❌ Employee test error:', error.message);
    throw error;
  }
};

// Test Relationships
const testRelationships = async (department) => {
  console.log('\n🔗 Testing Relationships...\n');
  
  try {
    // Get employees in department
    const deptEmployees = await Employee.findByDepartment(department._id);
    console.log(`✅ Employees in ${department.name}: ${deptEmployees.length}`);
    
    deptEmployees.forEach(emp => {
      console.log(`   - ${emp.fullName} (${emp.jobTitle})`);
      if (emp.supervisor) {
        console.log(`     Supervisor: ${emp.supervisor.firstName} ${emp.supervisor.lastName}`);
      }
    });
    
    // Update department employee count
    department.employeeCount = deptEmployees.length;
    await department.save();
    console.log(`\n✅ Department employee count updated: ${department.employeeCount}`);
    
  } catch (error) {
    console.error('❌ Relationship test error:', error.message);
    throw error;
  }
};

// Test Validations
const testValidations = async () => {
  console.log('\n✅ Testing Validations...\n');
  
  try {
    // Test invalid email
    const invalidEmployee = new Employee({
      firstName: 'Test',
      lastName: 'User',
      email: 'invalid-email', // Invalid email
      phone: '123-456-7890',
      jobTitle: 'Tester',
      country: 'USA',
      state: 'CA',
      city: 'SF'
    });
    
    await invalidEmployee.save();
  } catch (error) {
    console.log('✅ Email validation working:', error.message);
  }
  
  try {
    // Test missing required field
    const incompleteEmployee = new Employee({
      firstName: 'Test'
      // Missing required fields
    });
    
    await incompleteEmployee.save();
  } catch (error) {
    console.log('✅ Required field validation working');
  }
};

// Cleanup test data
const cleanup = async () => {
  console.log('\n🧹 Cleaning up test data...\n');
  
  try {
    await Employee.deleteMany({ 
      email: { $in: ['john.doe@company.com', 'jane.smith@company.com'] } 
    });
    await Department.deleteMany({ name: 'IT Department' });
    console.log('✅ Test data cleaned up');
  } catch (error) {
    console.error('❌ Cleanup error:', error.message);
  }
};

// Main test function
const runTests = async () => {
  console.log('🧪 Starting Model Tests...');
  console.log('================================\n');
  
  try {
    await connectDB();
    
    // Run tests
    const department = await testDepartmentModel();
    const { employee, supervisor } = await testEmployeeModel(department);
    await testRelationships(department);
    await testValidations();
    
    console.log('\n================================');
    console.log('✅ All tests completed successfully!');
    console.log('================================\n');
    
    // Cleanup
    await cleanup();
    
    // Close connection
    await mongoose.connection.close();
    console.log('✅ MongoDB connection closed');
    
  } catch (error) {
    console.error('\n❌ Test failed:', error);
    await cleanup();
    await mongoose.connection.close();
    process.exit(1);
  }
};

// Run tests
runTests();