78 lines
3.1 KiB
TypeScript
78 lines
3.1 KiB
TypeScript
import fs from 'node:fs/promises';
|
||
import os from 'node:os';
|
||
import path from 'node:path';
|
||
import { eq } from 'drizzle-orm';
|
||
import { afterAll, beforeAll, describe, expect, it } from 'vitest';
|
||
import * as schema from '@/lib/db/schema';
|
||
import { getGameDb } from '@/lib/db/client';
|
||
import { FUEL_PRICE_PER_L, PLANE_TYPES } from '@/lib/game/catalog';
|
||
import { POST } from './route';
|
||
|
||
let dir: string;
|
||
|
||
beforeAll(async () => {
|
||
// Same pattern as app/api/state/route.test.ts: point the DB singleton at a
|
||
// throwaway file (fresh module registry per test file → fresh singleton).
|
||
dir = await fs.mkdtemp(path.join(os.tmpdir(), 'flight-refuel-api-'));
|
||
process.env.FLIGHT_DB_PATH = path.join(dir, 'refuel.db');
|
||
});
|
||
|
||
afterAll(async () => {
|
||
delete process.env.FLIGHT_DB_PATH;
|
||
await fs.rm(dir, { recursive: true, force: true });
|
||
});
|
||
|
||
const post = (planeId: string) => POST(new Request(`http://localhost/api/planes/${planeId}/refuel`), { params: Promise.resolve({ planeId }) });
|
||
|
||
describe('POST /api/planes/:planeId/refuel', () => {
|
||
it('rejects an unknown id with 404', async () => {
|
||
const res = await post('ghost');
|
||
expect(res.status).toBe(404);
|
||
});
|
||
|
||
it('fills a partially empty tank and deducts (maxFuel − fuel) × FUEL_PRICE_PER_L from cash', async () => {
|
||
const { db } = await getGameDb();
|
||
await db.update(schema.planes).set({ fuel: 1_000 }).where(eq(schema.planes.id, 'plane-starter'));
|
||
const cashBefore = (await db.select().from(schema.account))[0].cash;
|
||
|
||
const res = await post('plane-starter');
|
||
expect(res.status).toBe(200);
|
||
|
||
const body = (await res.json()) as {
|
||
account: { cash: number };
|
||
planes: Array<{ id: string; fuel: number; maxFuel: number }>;
|
||
};
|
||
const cost = Math.round((PLANE_TYPES.atr72.maxFuel - 1_000) * FUEL_PRICE_PER_L * 100) / 100;
|
||
expect(body.account.cash).toBeCloseTo(cashBefore - cost, 6);
|
||
|
||
const plane = body.planes.find((p) => p.id === 'plane-starter')!;
|
||
expect(plane.fuel).toBe(PLANE_TYPES.atr72.maxFuel); // back to full
|
||
});
|
||
|
||
it('rejects a flying plane with 409', async () => {
|
||
const { db } = await getGameDb();
|
||
await db.update(schema.planes)
|
||
.set({ status: 'flying', departureSimTime: Date.now(), contractId: 'c-x' })
|
||
.where(eq(schema.planes.id, 'plane-starter'));
|
||
|
||
const res = await post('plane-starter');
|
||
expect(res.status).toBe(409);
|
||
expect(((await res.json()) as { error: string }).error).toMatch(/flying/);
|
||
|
||
// Ground it again for later tests.
|
||
await db.update(schema.planes)
|
||
.set({ status: 'available', departureSimTime: null, contractId: null })
|
||
.where(eq(schema.planes.id, 'plane-starter'));
|
||
});
|
||
|
||
it('rejects the fill-up with 409 when cash is below the cost', async () => {
|
||
const { db } = await getGameDb();
|
||
await db.update(schema.planes).set({ fuel: 1_000 }).where(eq(schema.planes.id, 'plane-starter'));
|
||
await db.update(schema.account).set({ cash: 10 }); // cost is ~4960
|
||
|
||
const res = await post('plane-starter');
|
||
expect(res.status).toBe(409);
|
||
expect(((await res.json()) as { error: string }).error).toMatch(/insufficient funds/i);
|
||
});
|
||
});
|