created lots of tests
This commit is contained in:
parent
42baadae38
commit
54f37f9521
21 changed files with 4577 additions and 215 deletions
275
libs/core/handlers/test/auto-register.test.ts
Normal file
275
libs/core/handlers/test/auto-register.test.ts
Normal file
|
|
@ -0,0 +1,275 @@
|
|||
import { describe, expect, it, beforeEach, mock } from 'bun:test';
|
||||
import {
|
||||
autoRegisterHandlers,
|
||||
createAutoHandlerRegistry,
|
||||
findHandlerFiles,
|
||||
extractHandlerClasses,
|
||||
} from '../src/registry/auto-register';
|
||||
import type { HandlerRegistry } from '@stock-bot/handler-registry';
|
||||
import { Handler, Operation } from '../src/decorators/decorators';
|
||||
|
||||
describe('Auto Registration', () => {
|
||||
const mockRegistry: HandlerRegistry = {
|
||||
registerHandler: mock(() => {}),
|
||||
getHandler: mock(() => null),
|
||||
hasHandler: mock(() => false),
|
||||
getAllHandlers: mock(() => []),
|
||||
getHandlersByService: mock(() => []),
|
||||
getHandlerOperations: mock(() => []),
|
||||
hasOperation: mock(() => false),
|
||||
executeOperation: mock(async () => ({ result: 'mocked' })),
|
||||
clear: mock(() => {}),
|
||||
};
|
||||
|
||||
const mockLogger = {
|
||||
info: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
};
|
||||
|
||||
beforeEach(() => {
|
||||
// Reset all mocks
|
||||
Object.values(mockRegistry).forEach(method => {
|
||||
if (typeof method === 'function' && 'mockClear' in method) {
|
||||
(method as any).mockClear();
|
||||
}
|
||||
});
|
||||
Object.values(mockLogger).forEach(method => {
|
||||
if (typeof method === 'function' && 'mockClear' in method) {
|
||||
(method as any).mockClear();
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
describe('findHandlerFiles', () => {
|
||||
it('should find handler files matching default pattern', async () => {
|
||||
// This test would need actual file system or mocking
|
||||
// For now, we'll test the function exists and returns an array
|
||||
const files = await findHandlerFiles();
|
||||
expect(Array.isArray(files)).toBe(true);
|
||||
});
|
||||
|
||||
it('should find handler files with custom pattern', async () => {
|
||||
const files = await findHandlerFiles('**/*.handler.ts');
|
||||
expect(Array.isArray(files)).toBe(true);
|
||||
});
|
||||
|
||||
it('should find handler files in specific directory', async () => {
|
||||
const files = await findHandlerFiles('*.handler.ts', './src/handlers');
|
||||
expect(Array.isArray(files)).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
describe('extractHandlerClasses', () => {
|
||||
it('should extract handler classes from module', () => {
|
||||
@Handler('TestHandler1')
|
||||
class Handler1 {
|
||||
@Operation('op1')
|
||||
async operation1() {}
|
||||
}
|
||||
|
||||
@Handler('TestHandler2')
|
||||
class Handler2 {
|
||||
@Operation('op2')
|
||||
async operation2() {}
|
||||
}
|
||||
|
||||
class NotAHandler {
|
||||
async someMethod() {}
|
||||
}
|
||||
|
||||
const testModule = {
|
||||
Handler1,
|
||||
Handler2,
|
||||
NotAHandler,
|
||||
someFunction: () => {},
|
||||
someValue: 123,
|
||||
};
|
||||
|
||||
const handlers = extractHandlerClasses(testModule);
|
||||
|
||||
expect(handlers).toHaveLength(2);
|
||||
expect(handlers.map(h => h.name)).toContain('Handler1');
|
||||
expect(handlers.map(h => h.name)).toContain('Handler2');
|
||||
});
|
||||
|
||||
it('should handle modules with no handlers', () => {
|
||||
const testModule = {
|
||||
SomeClass: class {},
|
||||
someFunction: () => {},
|
||||
someValue: 'test',
|
||||
};
|
||||
|
||||
const handlers = extractHandlerClasses(testModule);
|
||||
expect(handlers).toHaveLength(0);
|
||||
});
|
||||
|
||||
it('should handle empty modules', () => {
|
||||
const handlers = extractHandlerClasses({});
|
||||
expect(handlers).toHaveLength(0);
|
||||
});
|
||||
|
||||
it('should extract handlers with metadata', () => {
|
||||
@Handler('MetadataHandler')
|
||||
class HandlerWithMetadata {
|
||||
@Operation('process')
|
||||
async process() {}
|
||||
}
|
||||
|
||||
const testModule = { HandlerWithMetadata };
|
||||
const handlers = extractHandlerClasses(testModule);
|
||||
|
||||
expect(handlers).toHaveLength(1);
|
||||
|
||||
const handlerClass = handlers[0];
|
||||
const metadata = Reflect.getMetadata('handler', handlerClass);
|
||||
|
||||
expect(metadata).toEqual({
|
||||
name: 'MetadataHandler',
|
||||
disabled: false,
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('autoRegisterHandlers', () => {
|
||||
it('should auto-register handlers', async () => {
|
||||
// Since this function reads from file system, we'll test its behavior
|
||||
// by mocking the registry calls
|
||||
const options = {
|
||||
pattern: '**/*.handler.ts',
|
||||
directory: './test-handlers',
|
||||
serviceName: 'test-service',
|
||||
};
|
||||
|
||||
// This would normally scan files and register handlers
|
||||
// For unit testing, we'll verify the function executes without errors
|
||||
await expect(
|
||||
autoRegisterHandlers(mockRegistry, options, mockLogger)
|
||||
).resolves.not.toThrow();
|
||||
});
|
||||
|
||||
it('should use default options when not provided', async () => {
|
||||
await expect(
|
||||
autoRegisterHandlers(mockRegistry)
|
||||
).resolves.not.toThrow();
|
||||
});
|
||||
|
||||
it('should handle registration errors gracefully', async () => {
|
||||
mockRegistry.registerHandler = mock(() => {
|
||||
throw new Error('Registration failed');
|
||||
});
|
||||
|
||||
// Should not throw, but log errors
|
||||
await expect(
|
||||
autoRegisterHandlers(mockRegistry, {}, mockLogger)
|
||||
).resolves.not.toThrow();
|
||||
});
|
||||
});
|
||||
|
||||
describe('createAutoHandlerRegistry', () => {
|
||||
it('should create a registry function for a service', () => {
|
||||
const registerFunction = createAutoHandlerRegistry('my-service');
|
||||
|
||||
expect(typeof registerFunction).toBe('function');
|
||||
|
||||
// Test the created function
|
||||
const result = registerFunction(mockRegistry, mockLogger);
|
||||
expect(result).toBeInstanceOf(Promise);
|
||||
});
|
||||
|
||||
it('should pass through custom options', () => {
|
||||
const customOptions = {
|
||||
pattern: '**/*.custom-handler.ts',
|
||||
directory: './custom-handlers',
|
||||
};
|
||||
|
||||
const registerFunction = createAutoHandlerRegistry('my-service', customOptions);
|
||||
|
||||
// The function should be created with merged options
|
||||
expect(typeof registerFunction).toBe('function');
|
||||
});
|
||||
|
||||
it('should use service name in registration', async () => {
|
||||
@Handler('ServiceHandler')
|
||||
class TestServiceHandler {
|
||||
@Operation('serviceOp')
|
||||
async operation() {}
|
||||
}
|
||||
|
||||
// Mock file discovery to return our test handler
|
||||
const mockFindFiles = mock(async () => ['test.handler.ts']);
|
||||
const mockExtract = mock(() => [TestServiceHandler]);
|
||||
|
||||
const registerFunction = createAutoHandlerRegistry('test-service');
|
||||
|
||||
// Execute registration
|
||||
await registerFunction(mockRegistry, mockLogger);
|
||||
|
||||
// Verify service name would be used in actual implementation
|
||||
expect(typeof registerFunction).toBe('function');
|
||||
});
|
||||
});
|
||||
|
||||
describe('integration scenarios', () => {
|
||||
it('should handle complex handler hierarchies', () => {
|
||||
@Handler('BaseHandler')
|
||||
class BaseTestHandler {
|
||||
@Operation('baseOp')
|
||||
async baseOperation() {}
|
||||
}
|
||||
|
||||
@Handler('DerivedHandler')
|
||||
class DerivedTestHandler extends BaseTestHandler {
|
||||
@Operation('derivedOp')
|
||||
async derivedOperation() {}
|
||||
}
|
||||
|
||||
const testModule = {
|
||||
BaseTestHandler,
|
||||
DerivedTestHandler,
|
||||
};
|
||||
|
||||
const handlers = extractHandlerClasses(testModule);
|
||||
|
||||
expect(handlers).toHaveLength(2);
|
||||
|
||||
// Both should be valid handler classes
|
||||
handlers.forEach(handlerClass => {
|
||||
const metadata = Reflect.getMetadata('handler', handlerClass);
|
||||
expect(metadata).toBeDefined();
|
||||
expect(metadata.disabled).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
it('should skip disabled handlers if needed', () => {
|
||||
@Handler('EnabledHandler')
|
||||
class EnabledHandler {
|
||||
@Operation('op1')
|
||||
async operation() {}
|
||||
}
|
||||
|
||||
@Handler('DisabledHandler')
|
||||
class DisabledHandler {
|
||||
@Operation('op2')
|
||||
async operation() {}
|
||||
}
|
||||
|
||||
// Mark as disabled
|
||||
Reflect.defineMetadata('handler', { name: 'DisabledHandler', disabled: true }, DisabledHandler);
|
||||
|
||||
const testModule = {
|
||||
EnabledHandler,
|
||||
DisabledHandler,
|
||||
};
|
||||
|
||||
const handlers = extractHandlerClasses(testModule);
|
||||
|
||||
// Should extract both, filtering happens during registration
|
||||
expect(handlers).toHaveLength(2);
|
||||
|
||||
const disabledMetadata = Reflect.getMetadata('handler', DisabledHandler);
|
||||
expect(disabledMetadata.disabled).toBe(true);
|
||||
});
|
||||
});
|
||||
});
|
||||
Loading…
Add table
Add a link
Reference in a new issue