photo-live/services/api/test/auth.e2e-spec.ts

// services/api/test/auth.e2e-spec.ts —— Task 13 登录/刷新/注销 e2e（先写后实现）。
// 真实 app + MySQL（reset.ts 隔离，用户经 Prisma 直种）。覆盖：
//   成功登录（令牌形状 / TTL / audit_logs / sessions.tokenHash）、
//   凭据错误的三种 401 归并（错误密码 / 账号不存在 / 状态不合格——同码防枚举）、
//   滑动续期（同令牌不轮换、过期时间前移）、垃圾/缺失令牌 401、
//   注销 204 + 令牌即刻失效 + sessions.revokedAt、
//   账号被停用后既有会话被守卫撤销、
//   登录失败限速（第 6 次 429 AUTH_RATE_LIMITED；成功登录清零计数）、
//   DTO 校验 400 VALIDATION_FAILED（全局校验管道）。
// 限速计数在 app 实例内存中跨用例存留——各限速用例使用独立账号，避免相互污染；
// alice 在首个用例失败 1 次后由成功登录清零，后续用例不受影响。
import { createHash } from 'node:crypto';
import { type INestApplication } from '@nestjs/common';
import { Test } from '@nestjs/testing';
import request from 'supertest';
import { PrismaClient } from '@photo/db-cn';
import { ErrorCodes } from '@photo/contracts';
import { AppCnModule } from '../src/app-cn.module';
import { hashPassword } from '../src/auth/password';
import { API_PREFIX } from '../src/shared/constants';
import { resetDb } from './reset';

const TTL_SECONDS = Number.parseInt(
  process.env.AUTH_SESSION_TTL_SECONDS ?? '2592000',
  10,
);
/** 断言容差：登录/续期到响应到达之间的真实耗时（本机 e2e 足够）。 */
const TTL_TOLERANCE_MS = 120_000;

function sha256Hex(value: string): string {
  return createHash('sha256').update(value).digest('hex');
}

describe('auth (e2e)', () => {
  let app: INestApplication;
  let prisma: PrismaClient;
  let aliceId: string;

  const PASSWORD = 'correct-pass-1';

  beforeAll(async () => {
    await resetDb();
    const moduleRef = await Test.createTestingModule({
      imports: [AppCnModule],
    }).compile();
    app = moduleRef.createNestApplication();
    app.setGlobalPrefix(API_PREFIX);
    await app.init();

    prisma = new PrismaClient();
    const hash = await hashPassword(PASSWORD);
    aliceId = (
      await prisma.users.create({
        data: { account: 'alice', passwordHash: hash, displayName: 'Alice' },
        select: { id: true },
      })
    ).id;
    // 状态不合格三态：停用 / 本地锁定 / 凭据待激活（A01 未激活无法进入）。
    await prisma.users.create({
      data: {
        account: 'bob',
        passwordHash: hash,
        displayName: 'Bob',
        sourceStatus: 'disabled',
      },
    });
    await prisma.users.create({
      data: {
        account: 'carol',
        passwordHash: hash,
        displayName: 'Carol',
        localLocked: true,
      },
    });
    await prisma.users.create({
      data: {
        account: 'dave',
        passwordHash: hash,
        displayName: 'Dave',
        credentialStatus: 'pending',
      },
    });
    // 限速专用账号（与其它用例隔离，避免内存计数交叉污染）。
    await prisma.users.create({
      data: { account: 'erin', passwordHash: hash, displayName: 'Erin' },
    });
    await prisma.users.create({
      data: { account: 'frank', passwordHash: hash, displayName: 'Frank' },
    });
    // 会话撤销语义专用账号（测试内会被置为 disabled，不复用其它账号）。
    await prisma.users.create({
      data: { account: 'gina', passwordHash: hash, displayName: 'Gina' },
    });
    // password-change 专用账号（P3 Task 3；改密用例放在最后，成功后旧密码失效）。
    await prisma.users.create({
      data: { account: 'hank', passwordHash: hash, displayName: 'Hank' },
    });
  });

  afterAll(async () => {
    await app.close();
    await prisma.$disconnect();
  });

  function login(account: string, password: string) {
    return request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/password`)
      .send({ account, password });
  }

  it('正确账号密码 → 200 {sessionToken, expiresAt≈now+TTL}，tokenHash=sha256(token)，且写 audit_logs(auth.login)', async () => {
    const before = Date.now();
    const res = await login('alice', PASSWORD);
    expect(res.status).toBe(200);
    expect(Object.keys(res.body).sort()).toEqual(['expiresAt', 'sessionToken']);

    const { sessionToken, expiresAt } = res.body as {
      sessionToken: string;
      expiresAt: string;
    };
    // 32B 随机数的 hex 编码。
    expect(sessionToken).toMatch(/^[0-9a-f]{64}$/);

    const expiresMs = Date.parse(expiresAt);
    const lower = before + TTL_SECONDS * 1000 - TTL_TOLERANCE_MS;
    const upper = before + TTL_SECONDS * 1000 + TTL_TOLERANCE_MS;
    expect(expiresMs).toBeGreaterThanOrEqual(lower);
    expect(expiresMs).toBeLessThanOrEqual(upper);

    // 库内只存哈希；明文令牌不落库。
    const session = await prisma.sessions.findUnique({
      where: { tokenHash: sha256Hex(sessionToken) },
    });
    expect(session).not.toBeNull();
    expect(session?.userId).toBe(aliceId);
    expect(session?.revokedAt).toBeNull();

    const audit = await prisma.audit_logs.findFirst({
      where: { action: 'auth.login', actorId: aliceId },
    });
    expect(audit).not.toBeNull();
  });

  it('错误密码 → 401 AUTH_INVALID_CREDENTIALS', async () => {
    const res = await login('alice', 'wrong-password');
    expect(res.status).toBe(401);
    expect(res.body.code).toBe(ErrorCodes.AUTH_INVALID_CREDENTIALS);
    expect(typeof res.body.requestId).toBe('string');
  });

  it('不存在的账号 → 401 同一错误码（防账号枚举）', async () => {
    const res = await login('ghost-account', 'whatever');
    expect(res.status).toBe(401);
    expect(res.body.code).toBe(ErrorCodes.AUTH_INVALID_CREDENTIALS);
  });

  it.each([
    ['sourceStatus=disabled', 'bob'],
    ['localLocked=true', 'carol'],
    ['credentialStatus=pending（A01 未激活）', 'dave'],
  ])('%s → 401 同一错误码（正确密码也不放行）', async (_label, account) => {
    const res = await login(account as string, PASSWORD);
    expect(res.status).toBe(401);
    expect(res.body.code).toBe(ErrorCodes.AUTH_INVALID_CREDENTIALS);
  });

  it('请求体缺 password → 400 VALIDATION_FAILED（全局校验管道）', async () => {
    const res = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/password`)
      .send({ account: 'alice' });
    expect(res.status).toBe(400);
    expect(res.body.code).toBe(ErrorCodes.VALIDATION_FAILED);
  });

  it('滑动续期：有效令牌 → 200 {expiresAt} 前移，且旧令牌仍可用（不轮换）', async () => {
    const loginRes = await login('alice', PASSWORD);
    const token: string = loginRes.body.sessionToken;
    const originalExpires: number = Date.parse(loginRes.body.expiresAt);

    // 把会话过期时间手工缩短到即将过期，续期后应回到 ~now+TTL（前移可观测）。
    await prisma.sessions.update({
      where: { tokenHash: sha256Hex(token) },
      data: { expiresAt: new Date(Date.now() + 60_000) },
    });

    const before = Date.now();
    const res = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/refresh`)
      .set('Authorization', `Bearer ${token}`);
    expect(res.status).toBe(200);
    expect(Object.keys(res.body)).toEqual(['expiresAt']);

    const renewedMs = Date.parse(res.body.expiresAt);
    expect(renewedMs).toBeGreaterThan(originalExpires);
    expect(renewedMs).toBeGreaterThanOrEqual(
      before + TTL_SECONDS * 1000 - TTL_TOLERANCE_MS,
    );
    expect(renewedMs).toBeLessThanOrEqual(
      before + TTL_SECONDS * 1000 + TTL_TOLERANCE_MS,
    );

    // 不轮换：同一令牌再次续期仍 200（若实现为轮换，旧令牌应已失效）。
    const again = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/refresh`)
      .set('Authorization', `Bearer ${token}`);
    expect(again.status).toBe(200);
  });

  it('续期携带垃圾令牌 → 401 AUTH_TOKEN_EXPIRED', async () => {
    const res = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/refresh`)
      .set('Authorization', 'Bearer not-a-real-token');
    expect(res.status).toBe(401);
    expect(res.body.code).toBe(ErrorCodes.AUTH_TOKEN_EXPIRED);
  });

  it('续期缺失 Authorization 头 → 401 AUTH_TOKEN_EXPIRED', async () => {
    const res = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/refresh`);
    expect(res.status).toBe(401);
    expect(res.body.code).toBe(ErrorCodes.AUTH_TOKEN_EXPIRED);
  });

  it('注销 → 204 无响应体；同令牌后续受保护请求 401；sessions.revokedAt 落库', async () => {
    const loginRes = await login('alice', PASSWORD);
    const token: string = loginRes.body.sessionToken;

    const logout = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/logout`)
      .set('Authorization', `Bearer ${token}`);
    expect(logout.status).toBe(204);
    expect(logout.body).toEqual({});

    const after = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/refresh`)
      .set('Authorization', `Bearer ${token}`);
    expect(after.status).toBe(401);
    expect(after.body.code).toBe(ErrorCodes.AUTH_TOKEN_EXPIRED);

    const session = await prisma.sessions.findUnique({
      where: { tokenHash: sha256Hex(token) },
      select: { revokedAt: true },
    });
    expect(session?.revokedAt).not.toBeNull();
  });

  it('账号被停用后，既有会话被守卫拒绝（撤销语义）', async () => {
    const res = await login('gina', PASSWORD);
    expect(res.status).toBe(200);
    const token: string = res.body.sessionToken;

    await prisma.users.update({
      where: { account: 'gina' },
      data: { sourceStatus: 'disabled' },
    });

    const refresh = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/refresh`)
      .set('Authorization', `Bearer ${token}`);
    expect(refresh.status).toBe(401);
    expect(refresh.body.code).toBe(ErrorCodes.AUTH_TOKEN_EXPIRED);
  });

  it('限速：同账号连续 5 次失败后，第 6 次（即使密码正确）→ 429 AUTH_RATE_LIMITED', async () => {
    for (let i = 0; i < 5; i++) {
      const res = await login('erin', 'wrong-password');
      expect(res.status).toBe(401);
      expect(res.body.code).toBe(ErrorCodes.AUTH_INVALID_CREDENTIALS);
    }
    const locked = await login('erin', PASSWORD);
    expect(locked.status).toBe(429);
    expect(locked.body.code).toBe(ErrorCodes.AUTH_RATE_LIMITED);
  });

  it('限速：成功登录清零失败计数（4 次失败 + 成功 + 再失败仍是 401 而非 429）', async () => {
    for (let i = 0; i < 4; i++) {
      const res = await login('frank', 'wrong-password');
      expect(res.status).toBe(401);
    }
    const ok = await login('frank', PASSWORD);
    expect(ok.status).toBe(200);

    const res = await login('frank', 'wrong-password');
    expect(res.status).toBe(401);
    expect(res.body.code).toBe(ErrorCodes.AUTH_INVALID_CREDENTIALS);
  });

  // ── 修改密码（P3 Task 3 契约增补）──

  const changePassword = (token: string, oldPassword: string, newPassword: string) =>
    request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/password-change`)
      .set('Authorization', `Bearer ${token}`)
      .send({ oldPassword, newPassword });

  it('修改密码：未认证 → 401 AUTH_TOKEN_EXPIRED', async () => {
    const res = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/password-change`)
      .send({ oldPassword: PASSWORD, newPassword: 'brand-new-pass-9' });
    expect(res.status).toBe(401);
    expect(res.body.code).toBe(ErrorCodes.AUTH_TOKEN_EXPIRED);
  });

  it('修改密码：newPassword 少于 8 位 → 400 VALIDATION_FAILED', async () => {
    const loginRes = await login('hank', PASSWORD);
    const token: string = loginRes.body.sessionToken;

    const res = await changePassword(token, PASSWORD, 'short7');
    expect(res.status).toBe(400);
    expect(res.body.code).toBe(ErrorCodes.VALIDATION_FAILED);

    // 密码未变：旧密码仍可登录（新密码不可）。
    const relogin = await login('hank', PASSWORD);
    expect(relogin.status).toBe(200);
  });

  it('修改密码：旧密码错误 → 401 AUTH_INVALID_CREDENTIALS，当前会话不受影响', async () => {
    const loginRes = await login('hank', PASSWORD);
    const token: string = loginRes.body.sessionToken;

    const res = await changePassword(token, 'wrong-old-password', 'brand-new-pass-9');
    expect(res.status).toBe(401);
    expect(res.body.code).toBe(ErrorCodes.AUTH_INVALID_CREDENTIALS);

    // 会话未被撤销（错误旧密码不产生任何副作用）。
    const refresh = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/refresh`)
      .set('Authorization', `Bearer ${token}`);
    expect(refresh.status).toBe(200);
  });

  it('修改密码：成功 → 204；其他会话撤销、当前会话保留；新密码可登录、旧密码拒绝；写审计', async () => {
    // 双会话：tokenA 为当前会话（发起改密），tokenB 为应被撤销的其他会话。
    const loginA = await login('hank', PASSWORD);
    const tokenA: string = loginA.body.sessionToken;
    const loginB = await login('hank', PASSWORD);
    const tokenB: string = loginB.body.sessionToken;
    const NEW_PASSWORD = 'brand-new-pass-9';

    const res = await changePassword(tokenA, PASSWORD, NEW_PASSWORD);
    expect(res.status).toBe(204);
    expect(res.body).toEqual({});

    // 其他会话：即刻失效（revokedAt 落库）。
    const refreshB = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/refresh`)
      .set('Authorization', `Bearer ${tokenB}`);
    expect(refreshB.status).toBe(401);
    expect(refreshB.body.code).toBe(ErrorCodes.AUTH_TOKEN_EXPIRED);
    const sessionB = await prisma.sessions.findUnique({
      where: { tokenHash: sha256Hex(tokenB) },
      select: { revokedAt: true },
    });
    expect(sessionB?.revokedAt).not.toBeNull();

    // 当前会话：保留（改密不踢本人下线）。
    const refreshA = await request(app.getHttpServer())
      .post(`/${API_PREFIX}/auth/refresh`)
      .set('Authorization', `Bearer ${tokenA}`);
    expect(refreshA.status).toBe(200);

    // 新旧密码登录裁决。
    const reloginNew = await login('hank', NEW_PASSWORD);
    expect(reloginNew.status).toBe(200);
    const reloginOld = await login('hank', PASSWORD);
    expect(reloginOld.status).toBe(401);
    expect(reloginOld.body.code).toBe(ErrorCodes.AUTH_INVALID_CREDENTIALS);

    // 审计：auth.password_change（actor=本人）。
    const hankId = (
      await prisma.users.findUnique({
        where: { account: 'hank' },
        select: { id: true },
      })
    )?.id;
    const audit = await prisma.audit_logs.findFirst({
      where: { action: 'auth.password_change', actorId: hankId },
    });
    expect(audit).not.toBeNull();
  });
});