running prettier for cleanup
This commit is contained in:
parent
fe7733aeb5
commit
d85cd58acd
151 changed files with 29158 additions and 27966 deletions
|
|
@ -1,284 +1,280 @@
|
|||
/**
|
||||
* QuestDB Client Test Setup
|
||||
*
|
||||
* Setup file specific to QuestDB client library tests.
|
||||
* Provides utilities and mocks for testing database operations.
|
||||
*/
|
||||
|
||||
import { newDb } from 'pg-mem';
|
||||
import { mock, spyOn, beforeAll, beforeEach } from 'bun:test';
|
||||
|
||||
// Mock PostgreSQL database for unit tests
|
||||
let pgMem: any;
|
||||
|
||||
beforeAll(() => {
|
||||
// Create in-memory PostgreSQL database
|
||||
pgMem = newDb();
|
||||
|
||||
// Register QuestDB-specific functions
|
||||
pgMem.public.registerFunction({
|
||||
name: 'now',
|
||||
implementation: () => new Date().toISOString()
|
||||
});
|
||||
|
||||
pgMem.public.registerFunction({
|
||||
name: 'dateadd',
|
||||
args: [{ type: 'text' }, { type: 'int' }, { type: 'timestamp' }],
|
||||
returns: 'timestamp',
|
||||
implementation: (unit: string, amount: number, date: Date) => {
|
||||
const result = new Date(date);
|
||||
switch (unit) {
|
||||
case 'd':
|
||||
case 'day':
|
||||
result.setDate(result.getDate() + amount);
|
||||
break;
|
||||
case 'h':
|
||||
case 'hour':
|
||||
result.setHours(result.getHours() + amount);
|
||||
break;
|
||||
case 'm':
|
||||
case 'minute':
|
||||
result.setMinutes(result.getMinutes() + amount);
|
||||
break;
|
||||
default:
|
||||
throw new Error(`Unsupported date unit: ${unit}`);
|
||||
}
|
||||
return result;
|
||||
} }); // Mock QuestDB HTTP client
|
||||
// Mock fetch using Bun's built-in mock
|
||||
(global as any).fetch = mock(() => {});
|
||||
|
||||
// Mock the logger module to avoid Pino configuration conflicts
|
||||
mock.module('@stock-bot/logger', () => ({
|
||||
Logger: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
child: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
}))
|
||||
})),
|
||||
getLogger: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
child: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
}))
|
||||
}))
|
||||
}));
|
||||
|
||||
// Mock Pino and its transports to avoid configuration conflicts
|
||||
mock.module('pino', () => ({
|
||||
default: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
child: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
}))
|
||||
}))
|
||||
}));
|
||||
|
||||
mock.module('pino-pretty', () => ({
|
||||
default: mock(() => ({}))
|
||||
}));
|
||||
|
||||
mock.module('pino-loki', () => ({
|
||||
default: mock(() => ({}))
|
||||
}));
|
||||
});
|
||||
|
||||
beforeEach(() => {
|
||||
// Reset database state
|
||||
if (pgMem) {
|
||||
try {
|
||||
pgMem.public.none('DROP TABLE IF EXISTS ohlcv CASCADE');
|
||||
pgMem.public.none('DROP TABLE IF EXISTS trades CASCADE');
|
||||
pgMem.public.none('DROP TABLE IF EXISTS quotes CASCADE');
|
||||
pgMem.public.none('DROP TABLE IF EXISTS indicators CASCADE');
|
||||
pgMem.public.none('DROP TABLE IF EXISTS performance CASCADE');
|
||||
pgMem.public.none('DROP TABLE IF EXISTS risk_metrics CASCADE');
|
||||
} catch (error) {
|
||||
// Tables might not exist, ignore errors
|
||||
}
|
||||
} // Reset fetch mock
|
||||
if ((global as any).fetch) {
|
||||
((global as any).fetch as any).mockClear?.();
|
||||
}
|
||||
});
|
||||
|
||||
/**
|
||||
* QuestDB-specific test utilities
|
||||
*/
|
||||
export const questdbTestHelpers = {
|
||||
/**
|
||||
* Get mock PostgreSQL adapter
|
||||
*/
|
||||
getMockPgAdapter: () => pgMem?.adapters?.createPg?.(),
|
||||
|
||||
/**
|
||||
* Execute SQL in mock database
|
||||
*/
|
||||
executeMockSQL: (sql: string, params?: any[]) => {
|
||||
return pgMem?.public?.query(sql, params);
|
||||
},
|
||||
/**
|
||||
* Mock successful QuestDB HTTP response
|
||||
*/ mockQuestDBHttpSuccess: (data: any) => {
|
||||
((global as any).fetch as any).mockResolvedValue?.({
|
||||
ok: true,
|
||||
status: 200,
|
||||
json: async () => data,
|
||||
text: async () => JSON.stringify(data)
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* Mock QuestDB HTTP error
|
||||
*/ mockQuestDBHttpError: (status: number, message: string) => {
|
||||
((global as any).fetch as any).mockResolvedValue?.({
|
||||
ok: false,
|
||||
status,
|
||||
json: async () => ({ error: message }),
|
||||
text: async () => message
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* Mock InfluxDB line protocol response
|
||||
*/ mockInfluxDBSuccess: () => {
|
||||
((global as any).fetch as any).mockResolvedValue?.({
|
||||
ok: true,
|
||||
status: 204,
|
||||
text: async () => ''
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* Create test OHLCV table
|
||||
*/
|
||||
createTestOHLCVTable: () => {
|
||||
const sql = `
|
||||
CREATE TABLE ohlcv (
|
||||
symbol VARCHAR(10),
|
||||
timestamp TIMESTAMP,
|
||||
open DECIMAL(10,2),
|
||||
high DECIMAL(10,2),
|
||||
low DECIMAL(10,2),
|
||||
close DECIMAL(10,2),
|
||||
volume BIGINT,
|
||||
source VARCHAR(50)
|
||||
)
|
||||
`;
|
||||
return pgMem?.public?.none(sql);
|
||||
},
|
||||
|
||||
/**
|
||||
* Insert test OHLCV data
|
||||
*/
|
||||
insertTestOHLCVData: (data: any[]) => {
|
||||
const sql = `
|
||||
INSERT INTO ohlcv (symbol, timestamp, open, high, low, close, volume, source)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
|
||||
`;
|
||||
|
||||
return Promise.all(
|
||||
data.map(row =>
|
||||
pgMem?.public?.none(sql, [
|
||||
row.symbol,
|
||||
row.timestamp,
|
||||
row.open,
|
||||
row.high,
|
||||
row.low,
|
||||
row.close,
|
||||
row.volume,
|
||||
row.source || 'test'
|
||||
])
|
||||
)
|
||||
);
|
||||
},
|
||||
|
||||
/**
|
||||
* Generate InfluxDB line protocol test data
|
||||
*/
|
||||
generateInfluxDBLines: (count: number = 5) => {
|
||||
const lines: string[] = [];
|
||||
const baseTime = Date.now() * 1000000; // Convert to nanoseconds
|
||||
|
||||
for (let i = 0; i < count; i++) {
|
||||
const time = baseTime + (i * 60000000000); // 1 minute intervals
|
||||
const price = 150 + Math.random() * 10;
|
||||
|
||||
lines.push(
|
||||
`ohlcv,symbol=TEST open=${price},high=${price + 1},low=${price - 1},close=${price + 0.5},volume=1000i ${time}`
|
||||
);
|
||||
}
|
||||
|
||||
return lines;
|
||||
},
|
||||
|
||||
/**
|
||||
* Validate QuestDB query syntax
|
||||
*/
|
||||
validateQuestDBQuery: (query: string): boolean => {
|
||||
// Basic validation for QuestDB-specific syntax
|
||||
const questdbKeywords = [
|
||||
'SAMPLE BY',
|
||||
'LATEST BY',
|
||||
'ASOF JOIN',
|
||||
'SPLICE JOIN',
|
||||
'LT JOIN'
|
||||
];
|
||||
|
||||
// Check for valid SQL structure
|
||||
const hasSelect = /SELECT\s+/i.test(query);
|
||||
const hasFrom = /FROM\s+/i.test(query);
|
||||
|
||||
return hasSelect && hasFrom;
|
||||
},
|
||||
|
||||
/**
|
||||
* Mock connection pool
|
||||
*/ createMockPool: () => {
|
||||
const mockQuery = () => Promise.resolve({ rows: [], rowCount: 0 });
|
||||
const mockRelease = () => {};
|
||||
const mockConnect = () => Promise.resolve({
|
||||
query: mockQuery,
|
||||
release: mockRelease
|
||||
});
|
||||
const mockEnd = () => Promise.resolve(undefined);
|
||||
|
||||
return {
|
||||
connect: mockConnect,
|
||||
end: mockEnd,
|
||||
totalCount: 0,
|
||||
idleCount: 0,
|
||||
waitingCount: 0
|
||||
};
|
||||
}
|
||||
};
|
||||
/**
|
||||
* QuestDB Client Test Setup
|
||||
*
|
||||
* Setup file specific to QuestDB client library tests.
|
||||
* Provides utilities and mocks for testing database operations.
|
||||
*/
|
||||
|
||||
import { beforeAll, beforeEach, mock, spyOn } from 'bun:test';
|
||||
import { newDb } from 'pg-mem';
|
||||
|
||||
// Mock PostgreSQL database for unit tests
|
||||
let pgMem: any;
|
||||
|
||||
beforeAll(() => {
|
||||
// Create in-memory PostgreSQL database
|
||||
pgMem = newDb();
|
||||
|
||||
// Register QuestDB-specific functions
|
||||
pgMem.public.registerFunction({
|
||||
name: 'now',
|
||||
implementation: () => new Date().toISOString(),
|
||||
});
|
||||
|
||||
pgMem.public.registerFunction({
|
||||
name: 'dateadd',
|
||||
args: [{ type: 'text' }, { type: 'int' }, { type: 'timestamp' }],
|
||||
returns: 'timestamp',
|
||||
implementation: (unit: string, amount: number, date: Date) => {
|
||||
const result = new Date(date);
|
||||
switch (unit) {
|
||||
case 'd':
|
||||
case 'day':
|
||||
result.setDate(result.getDate() + amount);
|
||||
break;
|
||||
case 'h':
|
||||
case 'hour':
|
||||
result.setHours(result.getHours() + amount);
|
||||
break;
|
||||
case 'm':
|
||||
case 'minute':
|
||||
result.setMinutes(result.getMinutes() + amount);
|
||||
break;
|
||||
default:
|
||||
throw new Error(`Unsupported date unit: ${unit}`);
|
||||
}
|
||||
return result;
|
||||
},
|
||||
}); // Mock QuestDB HTTP client
|
||||
// Mock fetch using Bun's built-in mock
|
||||
(global as any).fetch = mock(() => {});
|
||||
|
||||
// Mock the logger module to avoid Pino configuration conflicts
|
||||
mock.module('@stock-bot/logger', () => ({
|
||||
Logger: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
child: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
})),
|
||||
})),
|
||||
getLogger: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
child: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
})),
|
||||
})),
|
||||
}));
|
||||
|
||||
// Mock Pino and its transports to avoid configuration conflicts
|
||||
mock.module('pino', () => ({
|
||||
default: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
child: mock(() => ({
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
debug: mock(() => {}),
|
||||
fatal: mock(() => {}),
|
||||
trace: mock(() => {}),
|
||||
})),
|
||||
})),
|
||||
}));
|
||||
|
||||
mock.module('pino-pretty', () => ({
|
||||
default: mock(() => ({})),
|
||||
}));
|
||||
|
||||
mock.module('pino-loki', () => ({
|
||||
default: mock(() => ({})),
|
||||
}));
|
||||
});
|
||||
|
||||
beforeEach(() => {
|
||||
// Reset database state
|
||||
if (pgMem) {
|
||||
try {
|
||||
pgMem.public.none('DROP TABLE IF EXISTS ohlcv CASCADE');
|
||||
pgMem.public.none('DROP TABLE IF EXISTS trades CASCADE');
|
||||
pgMem.public.none('DROP TABLE IF EXISTS quotes CASCADE');
|
||||
pgMem.public.none('DROP TABLE IF EXISTS indicators CASCADE');
|
||||
pgMem.public.none('DROP TABLE IF EXISTS performance CASCADE');
|
||||
pgMem.public.none('DROP TABLE IF EXISTS risk_metrics CASCADE');
|
||||
} catch (error) {
|
||||
// Tables might not exist, ignore errors
|
||||
}
|
||||
} // Reset fetch mock
|
||||
if ((global as any).fetch) {
|
||||
((global as any).fetch as any).mockClear?.();
|
||||
}
|
||||
});
|
||||
|
||||
/**
|
||||
* QuestDB-specific test utilities
|
||||
*/
|
||||
export const questdbTestHelpers = {
|
||||
/**
|
||||
* Get mock PostgreSQL adapter
|
||||
*/
|
||||
getMockPgAdapter: () => pgMem?.adapters?.createPg?.(),
|
||||
|
||||
/**
|
||||
* Execute SQL in mock database
|
||||
*/
|
||||
executeMockSQL: (sql: string, params?: any[]) => {
|
||||
return pgMem?.public?.query(sql, params);
|
||||
},
|
||||
/**
|
||||
* Mock successful QuestDB HTTP response
|
||||
*/ mockQuestDBHttpSuccess: (data: any) => {
|
||||
((global as any).fetch as any).mockResolvedValue?.({
|
||||
ok: true,
|
||||
status: 200,
|
||||
json: async () => data,
|
||||
text: async () => JSON.stringify(data),
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* Mock QuestDB HTTP error
|
||||
*/ mockQuestDBHttpError: (status: number, message: string) => {
|
||||
((global as any).fetch as any).mockResolvedValue?.({
|
||||
ok: false,
|
||||
status,
|
||||
json: async () => ({ error: message }),
|
||||
text: async () => message,
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* Mock InfluxDB line protocol response
|
||||
*/ mockInfluxDBSuccess: () => {
|
||||
((global as any).fetch as any).mockResolvedValue?.({
|
||||
ok: true,
|
||||
status: 204,
|
||||
text: async () => '',
|
||||
});
|
||||
},
|
||||
|
||||
/**
|
||||
* Create test OHLCV table
|
||||
*/
|
||||
createTestOHLCVTable: () => {
|
||||
const sql = `
|
||||
CREATE TABLE ohlcv (
|
||||
symbol VARCHAR(10),
|
||||
timestamp TIMESTAMP,
|
||||
open DECIMAL(10,2),
|
||||
high DECIMAL(10,2),
|
||||
low DECIMAL(10,2),
|
||||
close DECIMAL(10,2),
|
||||
volume BIGINT,
|
||||
source VARCHAR(50)
|
||||
)
|
||||
`;
|
||||
return pgMem?.public?.none(sql);
|
||||
},
|
||||
|
||||
/**
|
||||
* Insert test OHLCV data
|
||||
*/
|
||||
insertTestOHLCVData: (data: any[]) => {
|
||||
const sql = `
|
||||
INSERT INTO ohlcv (symbol, timestamp, open, high, low, close, volume, source)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
|
||||
`;
|
||||
|
||||
return Promise.all(
|
||||
data.map(row =>
|
||||
pgMem?.public?.none(sql, [
|
||||
row.symbol,
|
||||
row.timestamp,
|
||||
row.open,
|
||||
row.high,
|
||||
row.low,
|
||||
row.close,
|
||||
row.volume,
|
||||
row.source || 'test',
|
||||
])
|
||||
)
|
||||
);
|
||||
},
|
||||
|
||||
/**
|
||||
* Generate InfluxDB line protocol test data
|
||||
*/
|
||||
generateInfluxDBLines: (count: number = 5) => {
|
||||
const lines: string[] = [];
|
||||
const baseTime = Date.now() * 1000000; // Convert to nanoseconds
|
||||
|
||||
for (let i = 0; i < count; i++) {
|
||||
const time = baseTime + i * 60000000000; // 1 minute intervals
|
||||
const price = 150 + Math.random() * 10;
|
||||
|
||||
lines.push(
|
||||
`ohlcv,symbol=TEST open=${price},high=${price + 1},low=${price - 1},close=${price + 0.5},volume=1000i ${time}`
|
||||
);
|
||||
}
|
||||
|
||||
return lines;
|
||||
},
|
||||
|
||||
/**
|
||||
* Validate QuestDB query syntax
|
||||
*/
|
||||
validateQuestDBQuery: (query: string): boolean => {
|
||||
// Basic validation for QuestDB-specific syntax
|
||||
const questdbKeywords = ['SAMPLE BY', 'LATEST BY', 'ASOF JOIN', 'SPLICE JOIN', 'LT JOIN'];
|
||||
|
||||
// Check for valid SQL structure
|
||||
const hasSelect = /SELECT\s+/i.test(query);
|
||||
const hasFrom = /FROM\s+/i.test(query);
|
||||
|
||||
return hasSelect && hasFrom;
|
||||
},
|
||||
|
||||
/**
|
||||
* Mock connection pool
|
||||
*/ createMockPool: () => {
|
||||
const mockQuery = () => Promise.resolve({ rows: [], rowCount: 0 });
|
||||
const mockRelease = () => {};
|
||||
const mockConnect = () =>
|
||||
Promise.resolve({
|
||||
query: mockQuery,
|
||||
release: mockRelease,
|
||||
});
|
||||
const mockEnd = () => Promise.resolve(undefined);
|
||||
|
||||
return {
|
||||
connect: mockConnect,
|
||||
end: mockEnd,
|
||||
totalCount: 0,
|
||||
idleCount: 0,
|
||||
waitingCount: 0,
|
||||
};
|
||||
},
|
||||
};
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue