arasystem/backend/src/app.setup.spec.ts

import { BadRequestException, HttpStatus } from '@nestjs/common';
import type { Request, Response } from 'express';
import type { IncomingMessage, ServerResponse } from 'http';

import { startCpuTimer } from '@/../test/cpu-time';
import { BusinessException } from '@/common/errors/business.exception';
import { MessageDeletionDto } from '@/modules/mail/dto/bulk-messages.dto';
import { MailFilterUpdateDto } from '@/modules/mail/dto/mail-filter.dto';
import { SendMessageDto } from '@/modules/mail/dto/send-message.dto';
import {
  JSON_SHAPE_LIMITS,
  createValidationPipe,
  isDraftSaveRequest,
  isSendMessageRequest,
  rejectOversizedJsonBody,
  verifyJsonBody,
} from './app.setup';

/** 본문 파서의 읽기 전 확인을 부른다 — 던진 것을 돌려준다 (던지지 않으면 undefined) */
function verifyFailure(text: string, encoding = 'utf-8'): unknown {
  try {
    verifyJsonBody(
      {} as IncomingMessage,
      {} as ServerResponse,
      Buffer.from(text, 'utf8'),
      encoding,
    );
  } catch (error) {
    return error;
  }
  return undefined;
}

/** 읽은 뒤의 확인을 부른다 — next 에 넘긴 것 */
function afterParse(body: unknown): unknown[] {
  const next = jest.fn();
  rejectOversizedJsonBody({ body } as Request, {} as Response, next);
  return next.mock.calls[0] as unknown[];
}

const repeat = (unit: string, count: number) =>
  Array.from({ length: count }, () => unit).join(',');

describe('JSON 본문의 모양 한도 (JSON_SHAPE_LIMITS — DTO 변환 · 검사 전)', () => {
  it('읽기 전 확인 — 넘으면 BusinessException VAL_001(상태 400 — body-parser 가 이 상태를 쓴다), 까닭은 로그용 context 에만', () => {
    const failure = verifyFailure(`{"ids":[${repeat('1', 1_001)}]}`);

    expect(failure).toBeInstanceOf(BusinessException);
    expect((failure as BusinessException).code).toBe('VAL_001');
    expect((failure as BusinessException).getStatus()).toBe(
      HttpStatus.BAD_REQUEST,
    );
    // http-errors 는 감싼 오류의 status 를 상태 코드로 쓴다 (body-parser 의 createError(403, err))
    expect((failure as { status?: unknown }).status).toBe(400);
    expect((failure as BusinessException).context).toEqual({
      reason: 'json-shape-limit',
      problem: 'array-too-long',
      stage: 'before-parse',
    });
    expect(verifyFailure(`{"ids":[${repeat('1', 1_000)}]}`)).toBeUndefined();
  });

  it('JSON 본문은 UTF-8 만 — 다른 글자 인코딩(바이트로 셀 수 없다)은 작은 본문도 400 VAL_001', () => {
    // body-parser 는 `utf-` 로 시작하는 charset 을 모두 받아 소문자로 넘긴다
    for (const encoding of [
      'utf-16',
      'utf-16le',
      'utf-16be',
      'utf-32',
      'utf-7',
    ]) {
      const failure = verifyFailure('{"ids":[]}', encoding);
      expect(failure).toBeInstanceOf(BusinessException);
      expect((failure as { status?: unknown }).status).toBe(400);
      expect((failure as BusinessException).code).toBe('VAL_001');
      expect((failure as BusinessException).context).toEqual({
        reason: 'json-not-utf8',
      });
    }
  });

  it('UTF-8 로 읽히는 다른 이름(utf-_8 등)도 UTF-8 로 보고 모양을 본다 — 이름 글자 그대로 견주면 확인을 건너뛴다', () => {
    for (const encoding of [
      'utf-8',
      'utf-_8',
      'utf--8',
      'utf-8.',
      'utf-8:2026',
    ]) {
      expect(verifyFailure('{"ids":[]}', encoding)).toBeUndefined();
      expect(
        (
          verifyFailure(
            `{"ids":[${repeat('1', 1_001)}]}`,
            encoding,
          ) as BusinessException
        ).context,
      ).toEqual({
        reason: 'json-shape-limit',
        problem: 'array-too-long',
        stage: 'before-parse',
      });
    }
  });

  it('읽은 뒤 확인 — 한도 안이면 그대로 넘기고(next()), 본문이 없는 요청도 지난다. 넘으면 next(VAL_001)', () => {
    expect(afterParse({ rules: [], version: 'v' })).toEqual([]);
    expect(afterParse(undefined)).toEqual([]);

    const [failure] = afterParse({
      ids: Array.from({ length: 5_000 }, () => 1),
    });
    expect(failure).toBeInstanceOf(BusinessException);
    expect((failure as BusinessException).context).toEqual({
      reason: 'json-shape-limit',
      problem: 'array-too-long',
      stage: 'after-parse',
    });
  });

  it('한도는 계약의 어떤 본문보다 넉넉하다 — 배열 1,000 · 값 10,000 · 깊이 32', () => {
    expect(JSON_SHAPE_LIMITS).toEqual({
      maxArrayLength: 1_000,
      maxValues: 10_000,
      maxDepth: 32,
    });
  });

  describe('리뷰에서 잰 모양이 400 으로 0.5초 안에 끝난다 (본문 파서의 읽기 전 확인 — 본문은 재기 전에 만든다)', () => {
    it.each([
      [
        '필터 rules 에 {"id":N} 8만 개 (1MB)',
        `{"version":"v","rules":[${Array.from(
          { length: 80_000 },
          (_, index) => `{"id":${index}}`,
        ).join(',')}]}`,
      ],
      [
        '규칙 하나의 conditions 에 {} 15만 개',
        `{"version":"v","rules":[{"id":"r","conditions":[${repeat('{}', 150_000)}]}]}`,
      ],
      [
        '규칙 하나의 conditions 에 {} 30만 개 (900KB)',
        `{"version":"v","rules":[{"id":"r","conditions":[${repeat('{}', 300_000)}]}]}`,
      ],
      [
        '발송 to 에 {} 139만 개 (4MB 경로의 최대)',
        `{"subject":"x","body":"x","to":[${repeat('{}', 1_390_000)}]}`,
      ],
      [
        '여러 통 ids 에 문자열 6만 개',
        `{"ids":[${Array.from(
          { length: 60_000 },
          (_, index) => `"${index.toString(36)}"`,
        ).join(',')}]}`,
      ],
      [
        '로그인 본문에 10만 단 중첩 (600KB — 전에는 DTO 변환의 재귀가 넘쳐 500)',
        `{"email":"a@b.c","password":"x","z":${'{"a":'.repeat(100_000)}1${'}'.repeat(100_000)}}`,
      ],
    ])('%s', (_label, text) => {
      const cpuMs = startCpuTimer();
      const failure = verifyFailure(text);
      expect(cpuMs()).toBeLessThan(500);
      expect((failure as BusinessException).code).toBe('VAL_001');
    });
  });
});

describe('전역 DTO 검증 (createValidationPipe — stopAtFirstError)', () => {
  /** 모양 한도 안의 가장 무거운 본문들 — 본문 파서 · 모양 확인을 지난 뒤 DTO 변환 · 검사까지 0.5초 안에 400 이다 */
  const filtersWithManyUnknowns = {
    version: 'v',
    rules: Array.from({ length: 50 }, (_, index) => ({
      id: `r${index}`,
      name: '',
      isEnabled: true,
      match: 'all',
      conditions: Array.from({ length: 5 }, () => ({
        field: 'subject',
        operator: 'contains',
        value: 'x',
        extra: Array.from({ length: 32 }, () => 1),
      })),
      actions: {
        moveToMailboxId: null,
        markAsRead: true,
        flag: false,
        forwardTo: null,
      },
    })),
  };

  it.each<[string, new () => object, unknown]>([
    [
      '필터 rules 1,000개 (배열 한도까지)',
      MailFilterUpdateDto,
      {
        version: 'v',
        rules: Array.from({ length: 1_000 }, (_, index) => ({ id: index })),
      },
    ],
    [
      '규칙 하나의 conditions 1,000개',
      MailFilterUpdateDto,
      {
        version: 'v',
        rules: [
          { id: 'r', conditions: Array.from({ length: 1_000 }, () => ({})) },
        ],
      },
    ],
    [
      '필터 50개 × 조건 5개마다 모르는 배열 — 값 약 9,800개 (값 한도 가까이)',
      MailFilterUpdateDto,
      filtersWithManyUnknowns,
    ],
    [
      '발송 to 1,000명',
      SendMessageDto,
      {
        subject: 'x',
        body: 'x',
        to: Array.from({ length: 1_000 }, () => ({})),
      },
    ],
    [
      '여러 통 ids 1,000개',
      MessageDeletionDto,
      { ids: Array.from({ length: 1_000 }, (_, index) => `m${index}`) },
    ],
  ])(
    '%s — 모양 확인을 지나도 DTO 검사가 0.5초 안에 400',
    async (_label, metatype, body) => {
      const text = JSON.stringify(body);
      const pipe = createValidationPipe();
      const cpuMs = startCpuTimer();

      expect(verifyFailure(text)).toBeUndefined();
      const parsed: unknown = JSON.parse(text);
      expect(afterParse(parsed)).toEqual([]);
      await expect(
        pipe.transform(parsed, { type: 'body', metatype }),
      ).rejects.toBeInstanceOf(BadRequestException);
      expect(cpuMs()).toBeLessThan(500);
    },
  );

  it('검사 결과는 그대로다 — 맞는 본문은 지나고, 한도를 넘거나 모르는 칸이 있으면 400', async () => {
    const pipe = createValidationPipe();
    const valid = {
      rules: [],
      version: 'Cbd1Lkq0H3S7mJ5yV0w2qg8qkQvJ8sB7bW2nZ1y3u9E',
    };
    await expect(
      pipe.transform(valid, { type: 'body', metatype: MailFilterUpdateDto }),
    ).resolves.toBeInstanceOf(MailFilterUpdateDto);
    for (const body of [
      { ...valid, isActive: true },
      { ...valid, rules: Array.from({ length: 51 }, () => ({})) },
      { ...valid, version: 'a b' },
    ]) {
      await expect(
        pipe.transform(body, { type: 'body', metatype: MailFilterUpdateDto }),
      ).rejects.toBeInstanceOf(BadRequestException);
    }
  });
});

describe('임시 저장 요청 판정 (발송과 같은 JSON 본문 4MB)', () => {
  it('POST /api/drafts 와 PUT /api/drafts/{id} 는 임시 저장 요청이다 — 대소문자·끝의 / ·쿼리는 가리지 않는다', () => {
    for (const [method, url] of [
      ['POST', '/api/drafts'],
      ['POST', '/API/Drafts/'],
      ['PUT', '/api/drafts/faaaaabi'],
      ['PUT', '/api/drafts/faaaaabi/?x=1'],
    ] as const) {
      expect(isDraftSaveRequest({ method, url })).toBe(true);
    }
  });

  it('조회·다른 경로·하위 경로는 1MB 그대로다', () => {
    for (const [method, url] of [
      ['GET', '/api/drafts/faaaaabi'],
      ['PUT', '/api/drafts'],
      ['POST', '/api/drafts/faaaaabi'],
      ['PUT', '/api/drafts/faaaaabi/attachments'],
      ['PATCH', '/api/drafts/faaaaabi'],
      ['POST', '/api/draftsx'],
      ['PUT', '/drafts/faaaaabi'],
    ] as const) {
      expect(isDraftSaveRequest({ method, url })).toBe(false);
    }
    expect(isDraftSaveRequest({ method: 'POST', url: undefined })).toBe(false);
  });
});

describe('발송 요청 판정 (발송만 JSON 본문 4MB)', () => {
  it('POST /api/messages 는 발송 요청이다 — 대소문자·끝의 / ·쿼리는 가리지 않는다 (라우터와 같다)', () => {
    for (const url of [
      '/api/messages',
      '/api/messages/',
      '/API/Messages',
      '/api/messages?draft=1',
    ]) {
      expect(isSendMessageRequest({ method: 'POST', url })).toBe(true);
    }
  });

  it('다른 메서드·다른 경로는 1MB 그대로다 — 메일 수정·첨부 업로드·로그인 포함', () => {
    for (const [method, url] of [
      ['PATCH', '/api/messages/eaaaaab'],
      ['GET', '/api/messages'],
      ['PUT', '/api/messages'],
      ['POST', '/api/messages/eaaaaab'],
      ['POST', '/api/messages/eaaaaab/attachments'],
      ['POST', '/api/mail-attachments'],
      ['POST', '/api/auth/login'],
      ['POST', '/messages'],
      ['POST', '/api/messagesx'],
    ] as const) {
      expect(isSendMessageRequest({ method, url })).toBe(false);
    }
    expect(isSendMessageRequest({ method: 'POST', url: undefined })).toBe(
      false,
    );
  });
});