This commit is contained in:
2025-05-20 15:44:48 +00:00
parent b5e985eaf9
commit 4f3359b348
7 changed files with 483 additions and 1725 deletions

View File

@ -0,0 +1,197 @@
import * as plugins from '../ts/plugins.js';
import { SmartProxy } from '../ts/index.js';
import { tap, expect } from '@git.zone/tstest/tapbundle';
import { logger } from '../ts/core/utils/logger.js';
// Store the original logger reference
let originalLogger: any = logger;
let mockLogger: any;
// Create test routes using high ports to avoid permission issues
const createRoute = (id: number, domain: string, port: number = 8443) => ({
name: `test-route-${id}`,
match: {
ports: [port],
domains: [domain]
},
action: {
type: 'forward' as const,
target: {
host: 'localhost',
port: 3000 + id
},
tls: {
mode: 'terminate' as const,
certificate: 'auto' as const,
acme: {
email: 'test@testdomain.test',
useProduction: false
}
}
}
});
let testProxy: SmartProxy;
tap.test('should setup test proxy for logger error handling tests', async () => {
// Create a proxy for testing
testProxy = new SmartProxy({
routes: [createRoute(1, 'test1.error-handling.test', 8443)],
acme: {
email: 'test@testdomain.test',
useProduction: false,
port: 8080
}
});
// Mock the certificate manager to avoid actual ACME initialization
const originalCreateCertManager = (testProxy as any).createCertificateManager;
(testProxy as any).createCertificateManager = async function(routes: any[], certDir: string, acmeOptions: any, initialState?: any) {
const mockCertManager = {
setUpdateRoutesCallback: function(callback: any) {
this.updateRoutesCallback = callback;
},
updateRoutesCallback: null as any,
setHttpProxy: function() {},
setGlobalAcmeDefaults: function() {},
setAcmeStateManager: function() {},
initialize: async function() {},
provisionAllCertificates: async function() {},
stop: async function() {},
getAcmeOptions: function() {
return acmeOptions || { email: 'test@testdomain.test', useProduction: false };
},
getState: function() {
return initialState || { challengeRouteActive: false };
}
};
// Always set up the route update callback for ACME challenges
mockCertManager.setUpdateRoutesCallback(async (routes) => {
await this.updateRoutes(routes);
});
return mockCertManager;
};
// Mock initializeCertificateManager as well
(testProxy as any).initializeCertificateManager = async function() {
// Create mock cert manager using the method above
this.certManager = await this.createCertificateManager(
this.settings.routes,
'./certs',
{ email: 'test@testdomain.test', useProduction: false }
);
};
// Start the proxy with mocked components
await testProxy.start();
expect(testProxy).toBeTruthy();
});
tap.test('should handle logger errors in updateRoutes without failing', async () => {
// Temporarily inject the mock logger that throws errors
const origConsoleLog = console.log;
let consoleLogCalled = false;
// Spy on console.log to verify it's used as fallback
console.log = (...args: any[]) => {
consoleLogCalled = true;
// Call original implementation but mute the output for tests
// origConsoleLog(...args);
};
try {
// Create mock logger that throws
mockLogger = {
log: () => {
throw new Error('Simulated logger error');
}
};
// Override the logger in the imported module
// This is a hack but necessary for testing
(global as any).logger = mockLogger;
// Access the internal logger used by SmartProxy
const smartProxyImport = await import('../ts/proxies/smart-proxy/smart-proxy.js');
// @ts-ignore
smartProxyImport.logger = mockLogger;
// Update routes - this should not fail even with logger errors
const newRoutes = [
createRoute(1, 'test1.error-handling.test', 8443),
createRoute(2, 'test2.error-handling.test', 8444)
];
await testProxy.updateRoutes(newRoutes);
// Verify that the update was successful
expect((testProxy as any).settings.routes.length).toEqual(2);
expect(consoleLogCalled).toEqual(true);
} finally {
// Always restore console.log and logger
console.log = origConsoleLog;
(global as any).logger = originalLogger;
}
});
tap.test('should handle logger errors in certificate manager callbacks', async () => {
// Temporarily inject the mock logger that throws errors
const origConsoleLog = console.log;
let consoleLogCalled = false;
// Spy on console.log to verify it's used as fallback
console.log = (...args: any[]) => {
consoleLogCalled = true;
// Call original implementation but mute the output for tests
// origConsoleLog(...args);
};
try {
// Create mock logger that throws
mockLogger = {
log: () => {
throw new Error('Simulated logger error');
}
};
// Override the logger in the imported module
// This is a hack but necessary for testing
(global as any).logger = mockLogger;
// Access the cert manager and trigger the updateRoutesCallback
const certManager = (testProxy as any).certManager;
expect(certManager).toBeTruthy();
expect(certManager.updateRoutesCallback).toBeTruthy();
// Call the certificate manager's updateRoutesCallback directly
const challengeRoute = {
name: 'acme-challenge',
match: {
ports: [8080],
path: '/.well-known/acme-challenge/*'
},
action: {
type: 'static' as const,
content: 'mock-challenge-content'
}
};
// This should not throw, despite logger errors
await certManager.updateRoutesCallback([...testProxy.settings.routes, challengeRoute]);
// Verify console.log was used as fallback
expect(consoleLogCalled).toEqual(true);
} finally {
// Always restore console.log and logger
console.log = origConsoleLog;
(global as any).logger = originalLogger;
}
});
tap.test('should clean up properly', async () => {
await testProxy.stop();
});
tap.start();

View File

@ -0,0 +1,97 @@
import * as plugins from '../ts/plugins.js';
import { SmartProxy } from '../ts/index.js';
import { SmartCertManager } from '../ts/proxies/smart-proxy/certificate-manager.js';
import { tap, expect } from '@git.zone/tstest/tapbundle';
// Create test routes using high ports to avoid permission issues
const createRoute = (id: number, domain: string, port: number = 8443) => ({
name: `test-route-${id}`,
match: {
ports: [port],
domains: [domain]
},
action: {
type: 'forward' as const,
target: {
host: 'localhost',
port: 3000 + id
}
}
});
// Test function to check if error handling is applied to logger calls
tap.test('should have error handling around logger calls in route update callbacks', async () => {
// Create a simple cert manager instance for testing
const certManager = new SmartCertManager(
[createRoute(1, 'test.example.com', 8443)],
'./certs',
{ email: 'test@example.com', useProduction: false }
);
// Create a mock update routes callback that tracks if it was called
let callbackCalled = false;
const mockCallback = async (routes: any[]) => {
callbackCalled = true;
// Just return without doing anything
return Promise.resolve();
};
// Set the callback
certManager.setUpdateRoutesCallback(mockCallback);
// Verify the callback was successfully set
expect(callbackCalled).toEqual(false);
// Create a test route
const testRoute = createRoute(2, 'test2.example.com', 8444);
// Verify we can add a challenge route without error
// This tests the try/catch we added around addChallengeRoute logger calls
try {
// Accessing private method for testing
// @ts-ignore
await (certManager as any).addChallengeRoute();
// If we got here without error, the error handling works
expect(true).toEqual(true);
} catch (error) {
// This shouldn't happen if our error handling is working
expect(false).toEqual(true, 'Error handling failed in addChallengeRoute');
}
// Verify that we handle errors in removeChallengeRoute
try {
// Set the flag to active so we can test removal logic
// @ts-ignore
certManager.challengeRouteActive = true;
// @ts-ignore
await (certManager as any).removeChallengeRoute();
// If we got here without error, the error handling works
expect(true).toEqual(true);
} catch (error) {
// This shouldn't happen if our error handling is working
expect(false).toEqual(true, 'Error handling failed in removeChallengeRoute');
}
});
// Test verifyChallengeRouteRemoved error handling
tap.test('should have error handling in verifyChallengeRouteRemoved', async () => {
// Create a SmartProxy for testing
const testProxy = new SmartProxy({
routes: [createRoute(1, 'test1.domain.test')]
});
// Verify that verifyChallengeRouteRemoved has error handling
try {
// @ts-ignore - Access private method for testing
await (testProxy as any).verifyChallengeRouteRemoved();
// If we got here without error, the try/catch is working
// (This will still throw at the end after max retries, but we're testing that
// the logger calls have try/catch blocks around them)
} catch (error) {
// This error is expected since we don't have a real challenge route
// But we're testing that the logger calls don't throw
expect(error.message).toContain('Failed to verify challenge route removal');
}
});
tap.start();