feat: add Jest testing framework and implement unit tests for configuration and service logic

2026-06-17 08:15:52 +00:00 · 2025-10-21 11:08:26 -05:00
parent 526298bea4
commit 4ee1442589
12 changed files with 4517 additions and 3 deletions
@@ -0,0 +1,191 @@
+import { ConfigStorage, GuildConfig, MonitorGroup } from '../src/config/storage';
+import { existsSync, unlinkSync, mkdirSync, rmdirSync } from 'fs';
+import { join } from 'path';
+
+describe('ConfigStorage', () => {
+  let storage: ConfigStorage;
+  let testDataDir: string;
+  let testConfigPath: string;
+
+  beforeEach(() => {
+    // Create a temporary test directory
+    testDataDir = join(__dirname, 'temp-data');
+    testConfigPath = join(testDataDir, 'bot-config.json');
+    
+    // Set test data directory
+    process.env.DATA_DIR = testDataDir;
+    
+    // Clean up any existing test files
+    if (existsSync(testConfigPath)) {
+      unlinkSync(testConfigPath);
+    }
+    if (existsSync(testDataDir)) {
+      rmdirSync(testDataDir);
+    }
+    
+    // Create fresh storage instance
+    storage = new ConfigStorage();
+  });
+
+  afterEach(() => {
+    // Clean up test files
+    if (existsSync(testConfigPath)) {
+      unlinkSync(testConfigPath);
+    }
+    if (existsSync(testDataDir)) {
+      rmdirSync(testDataDir);
+    }
+    
+    // Reset environment
+    delete process.env.DATA_DIR;
+  });
+
+  test('should create default configuration for new guild', () => {
+    const guildId = '123456789';
+    const config = storage.getConfig(guildId);
+
+    expect(config.channelId).toBeNull();
+    expect(config.messageIds).toEqual([]);
+    expect(config.monitorIds).toEqual([]);
+    expect(config.groups).toEqual([]);
+    expect(config.updateInterval).toBe(60000);
+    expect(config.embedColor).toBe(5814783);
+    expect(config.statusMessage).toBe('Service Status');
+  });
+
+  test('should save and load guild configuration', () => {
+    const guildId = '123456789';
+    
+    // Set individual config values
+    storage.setChannelId(guildId, '987654321');
+    storage.setMessageIds(guildId, ['msg1', 'msg2']);
+    storage.setMonitorIds(guildId, [1, 2, 3]);
+    storage.setUpdateInterval(guildId, 30000);
+    storage.setEmbedColor(guildId, 16711680);
+    storage.setStatusMessage(guildId, 'My Custom Status');
+    
+    const loadedConfig = storage.getConfig(guildId);
+
+    expect(loadedConfig.channelId).toBe('987654321');
+    expect(loadedConfig.messageIds).toEqual(['msg1', 'msg2']);
+    expect(loadedConfig.monitorIds).toEqual([1, 2, 3]);
+    expect(loadedConfig.updateInterval).toBe(30000);
+    expect(loadedConfig.embedColor).toBe(16711680);
+    expect(loadedConfig.statusMessage).toBe('My Custom Status');
+  });
+
+  test('should update specific fields in guild configuration', () => {
+    const guildId = '123456789';
+    
+    // Set initial config
+    storage.setChannelId(guildId, 'channel123');
+    storage.setMessageIds(guildId, ['msg1', 'msg2']);
+    storage.setMonitorIds(guildId, [1, 2, 3]);
+    
+    const config = storage.getConfig(guildId);
+    expect(config.channelId).toBe('channel123');
+    expect(config.messageIds).toEqual(['msg1', 'msg2']);
+    expect(config.monitorIds).toEqual([1, 2, 3]);
+  });
+
+  test('should manage monitor groups', () => {
+    const guildId = '123456789';
+    
+    // Create groups
+    storage.addGroup(guildId, 'Media Servers');
+    storage.addGroup(guildId, 'Gaming');
+    
+    const groups = storage.getGroups(guildId);
+    expect(groups).toHaveLength(2);
+    expect(groups[0].name).toBe('Media Servers');
+    expect(groups[1].name).toBe('Gaming');
+    
+    // Add monitors to group
+    storage.addMonitorToGroup(guildId, 'Media Servers', 1);
+    storage.addMonitorToGroup(guildId, 'Media Servers', 2);
+    storage.addMonitorToGroup(guildId, 'Gaming', 3);
+    
+    const updatedGroups = storage.getGroups(guildId);
+    expect(updatedGroups[0].monitorIds).toEqual([1, 2]);
+    expect(updatedGroups[1].monitorIds).toEqual([3]);
+  });
+
+  test('should remove monitors from groups', () => {
+    const guildId = '123456789';
+    
+    storage.addGroup(guildId, 'Media Servers');
+    storage.addMonitorToGroup(guildId, 'Media Servers', 1);
+    storage.addMonitorToGroup(guildId, 'Media Servers', 2);
+    
+    storage.removeMonitorFromGroup(guildId, 1);
+    
+    const groups = storage.getGroups(guildId);
+    expect(groups[0].monitorIds).toEqual([2]);
+  });
+
+  test('should delete groups', () => {
+    const guildId = '123456789';
+    
+    storage.addGroup(guildId, 'Media Servers');
+    storage.addGroup(guildId, 'Gaming');
+    
+    expect(storage.getGroups(guildId)).toHaveLength(2);
+    
+    storage.removeGroup(guildId, 'Media Servers');
+    
+    const groups = storage.getGroups(guildId);
+    expect(groups).toHaveLength(1);
+    expect(groups[0].name).toBe('Gaming');
+  });
+
+  test('should get all guild IDs', () => {
+    storage.setChannelId('guild1', 'channel1');
+    storage.setChannelId('guild2', 'channel2');
+    storage.setChannelId('guild3', 'channel3');
+    
+    const guildIds = storage.getAllGuildIds();
+    expect(guildIds).toHaveLength(3);
+    expect(guildIds).toContain('guild1');
+    expect(guildIds).toContain('guild2');
+    expect(guildIds).toContain('guild3');
+  });
+
+  test('should handle non-existent groups gracefully', () => {
+    const guildId = '123456789';
+    
+    // Try to add monitor to non-existent group
+    expect(() => storage.addMonitorToGroup(guildId, 'Non-existent', 1)).not.toThrow();
+    
+    // Try to remove monitor from non-existent group
+    expect(() => storage.removeMonitorFromGroup(guildId, 1)).not.toThrow();
+    
+    // Try to delete non-existent group
+    expect(() => storage.removeGroup(guildId, 'Non-existent')).not.toThrow();
+  });
+
+  test('should persist configuration to file', () => {
+    const guildId = '123456789';
+    storage.setChannelId(guildId, 'channel123');
+    storage.setMonitorIds(guildId, [1, 2, 3]);
+    
+    // Create new storage instance to test persistence
+    const newStorage = new ConfigStorage();
+    const config = newStorage.getConfig(guildId);
+    
+    expect(config.channelId).toBe('channel123');
+    expect(config.monitorIds).toEqual([1, 2, 3]);
+  });
+
+  test('should handle file read errors gracefully', () => {
+    // Create invalid JSON file
+    const fs = require('fs');
+    fs.writeFileSync(testConfigPath, 'invalid json');
+    
+    // Should not throw error, should create default config
+    expect(() => new ConfigStorage()).not.toThrow();
+    
+    const storage = new ConfigStorage();
+    const config = storage.getConfig('test-guild');
+    expect(config.channelId).toBeNull();
+  });
+});