fbe696b88c
Constants migration: - server-agent.ts: SERVER_CONSULTATION_STATUS, LANGUAGE_CODE - troubleshoot-agent.ts: TROUBLESHOOT_STATUS - notification.ts: NOTIFICATION_TYPE API improvements: - search-tool.ts: Zod schema validation for Brave/Context7 APIs - api-helper.ts: Centralized API call utility with retry/timeout Testing: - kv-cache.test.ts: 38 test cases for cache abstraction Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
120 lines
3.2 KiB
TypeScript
120 lines
3.2 KiB
TypeScript
/**
|
|
* Vitest 테스트 환경 설정
|
|
*
|
|
* Miniflare를 사용하여 D1 Database를 in-memory SQLite로 시뮬레이션
|
|
*/
|
|
import { readFileSync } from 'fs';
|
|
import { join } from 'path';
|
|
import { beforeAll, afterEach } from 'vitest';
|
|
|
|
declare global {
|
|
function getMiniflareBindings(): {
|
|
DB: D1Database;
|
|
RATE_LIMIT_KV: KVNamespace;
|
|
};
|
|
}
|
|
|
|
let db: D1Database | null = null;
|
|
|
|
beforeAll(async () => {
|
|
// Miniflare 바인딩 가져오기 (있을 경우만)
|
|
if (typeof getMiniflareBindings === 'function') {
|
|
try {
|
|
const bindings = getMiniflareBindings();
|
|
db = bindings.DB;
|
|
|
|
// 스키마 초기화
|
|
const schemaPath = join(__dirname, '../schema.sql');
|
|
const schema = readFileSync(schemaPath, 'utf-8');
|
|
|
|
// 각 statement를 개별 실행
|
|
const statements = schema
|
|
.split(';')
|
|
.map(s => s.trim())
|
|
.filter(s => s.length > 0 && !s.startsWith('--'));
|
|
|
|
for (const statement of statements) {
|
|
await db.exec(statement);
|
|
}
|
|
} catch (error) {
|
|
// Miniflare 바인딩이 없는 테스트는 skip
|
|
console.warn('Miniflare bindings not available, skipping DB setup');
|
|
}
|
|
}
|
|
});
|
|
|
|
afterEach(async () => {
|
|
// DB가 있을 경우만 정리
|
|
if (db) {
|
|
// 각 테스트 후 데이터 정리 (스키마는 유지)
|
|
await db.exec('DELETE FROM deposit_transactions');
|
|
await db.exec('DELETE FROM bank_notifications');
|
|
await db.exec('DELETE FROM user_deposits');
|
|
await db.exec('DELETE FROM user_domains');
|
|
await db.exec('DELETE FROM summaries');
|
|
await db.exec('DELETE FROM message_buffer');
|
|
await db.exec('DELETE FROM users');
|
|
}
|
|
});
|
|
|
|
/**
|
|
* 테스트용 헬퍼 함수: 사용자 생성
|
|
*/
|
|
export async function createTestUser(
|
|
telegramId: string,
|
|
username?: string
|
|
): Promise<number> {
|
|
const bindings = getMiniflareBindings();
|
|
const result = await bindings.DB.prepare(
|
|
'INSERT INTO users (telegram_id, username) VALUES (?, ?)'
|
|
).bind(telegramId, username || null).run();
|
|
|
|
return Number(result.meta?.last_row_id || 0);
|
|
}
|
|
|
|
/**
|
|
* 테스트용 헬퍼 함수: 은행 알림 생성
|
|
*/
|
|
export async function createBankNotification(
|
|
depositorName: string,
|
|
amount: number
|
|
): Promise<number> {
|
|
const bindings = getMiniflareBindings();
|
|
const result = await bindings.DB.prepare(
|
|
'INSERT INTO bank_notifications (depositor_name, depositor_name_prefix, amount) VALUES (?, ?, ?)'
|
|
).bind(depositorName, depositorName.slice(0, 7), amount).run();
|
|
|
|
return Number(result.meta?.last_row_id || 0);
|
|
}
|
|
|
|
/**
|
|
* 테스트용 헬퍼 함수: 입금 거래 생성
|
|
*/
|
|
export async function createDepositTransaction(
|
|
userId: number,
|
|
amount: number,
|
|
status: string,
|
|
depositorName?: string
|
|
): Promise<number> {
|
|
const bindings = getMiniflareBindings();
|
|
const result = await bindings.DB.prepare(
|
|
`INSERT INTO deposit_transactions (user_id, type, amount, status, depositor_name, depositor_name_prefix)
|
|
VALUES (?, 'deposit', ?, ?, ?, ?)`
|
|
).bind(
|
|
userId,
|
|
amount,
|
|
status,
|
|
depositorName || null,
|
|
depositorName ? depositorName.slice(0, 7) : null
|
|
).run();
|
|
|
|
return Number(result.meta?.last_row_id || 0);
|
|
}
|
|
|
|
/**
|
|
* 테스트용 헬퍼 함수: DB 바인딩 가져오기
|
|
*/
|
|
export function getTestDB(): D1Database {
|
|
return getMiniflareBindings().DB;
|
|
}
|