runWithMockContext

Mock Context를 사용하여 테스트를 격리하고, 인증 상태를 제어하는 방법을 알아봅니다.

runWithMockContext란?

runWithMockContext는 Sonamu의 테스트 환경에서 격리된 Context를 제공하는 함수입니다. Sonamu는 AsyncLocalStorage를 사용하여 각 요청마다 독립된 Context를 유지하는데, 테스트에서도 동일한 메커니즘을 사용합니다.

Context의 구성 요소

Mock Context는 다음과 같은 속성을 포함합니다:

type Context = {
  request: FastifyRequest;
  reply: FastifyReply;
  headers: IncomingHttpHeaders;
  createSSE: <T extends ZodObject>(events: T) => ReturnType<typeof createSSEFactory<T>>;
  naiteStore: NaiteStore;  // Naite 로그 저장소
  locale: string;          // 요청 locale
  user: User | null;       // 인증된 사용자 정보 (better-auth)
  session: Session | null; // 세션 정보 (better-auth)
  // ... ContextExtend 속성
};

테스트에서는 간소화된 Mock Context가 제공됩니다:

{
  session: null,
  user: null,
  naiteStore: Naite.createStore(),
  locale: "",
}

기본 사용법

일반 테스트 (비인증)

Sonamu에서 제공하는 test() 함수는 자동으로 runWithMockContext를 실행합니다.

import { test } from "sonamu/test";
import { Sonamu } from "sonamu";
import { expect } from "vitest";

test("Context 접근 테스트", async () => {
  // test() 함수가 자동으로 runWithMockContext를 호출하므로
  // 바로 Context에 접근할 수 있습니다
  const context = Sonamu.getContext();

  expect(context.user).toBeNull();
  expect(context.session).toBeNull();
  expect(context.locale).toBe("");
});

직접 사용하기

runWithMockContext를 직접 호출할 수도 있습니다:

import { runWithMockContext } from "sonamu/test";
import { Sonamu } from "sonamu";

await runWithMockContext(async () => {
  const context = Sonamu.getContext();

  // 이 블록 내에서는 Mock Context가 활성화됩니다
  console.log(context.user); // null (미인증)
  console.log(context.session); // null
});

인증된 사용자로 테스트하기

testAs() 사용

특정 사용자로 로그인한 상태를 시뮬레이션하려면 testAs()를 사용합니다:

import { testAs } from "sonamu/test";
import { Sonamu } from "sonamu";
import { expect } from "vitest";

testAs(
  { id: 1, username: "john" },  // 사용자 정보
  "인증된 사용자 테스트",
  async () => {
    const context = Sonamu.getContext();
    
    expect(context.user).toEqual({ id: 1, username: "john" });
  }
);

타입 안전성

testAs()는 제네릭을 지원하여 타입 안전성을 보장합니다:

type MyUser = {
  id: number;
  username: string;
  role: "admin" | "user";
};

testAs<MyUser>(
  { id: 1, username: "admin", role: "admin" },
  "관리자 권한 테스트",
  async () => {
    const context = Sonamu.getContext();
    
    // context.user는 MyUser | null 타입으로 추론됩니다
    expect(context.user?.role).toBe("admin");
  }
);

실전 예제

Model 메서드 테스트

import { test, testAs } from "sonamu/test";
import { userModel } from "../application/user/user.model";
import { expect } from "vitest";

test("사용자 목록 조회 (비인증)", async () => {
  // Mock Context가 자동으로 제공됨
  const users = await userModel.findMany({});
  
  expect(Array.isArray(users)).toBe(true);
});

testAs(
  { id: 1, username: "john" },
  "내 프로필 조회 (인증)",
  async () => {
    // context.user = { id: 1, username: "john" }
    const profile = await userModel.getMyProfile();
    
    expect(profile.username).toBe("john");
  }
);

권한 검증 테스트

import { testAs } from "sonamu/test";
import { postModel } from "../application/post/post.model";
import { expect } from "vitest";

testAs(
  { id: 1, role: "admin" },
  "관리자는 모든 게시글 삭제 가능",
  async () => {
    const result = await postModel.delete({ id: 999 });
    expect(result.success).toBe(true);
  }
);

testAs(
  { id: 2, role: "user" },
  "일반 사용자는 타인 게시글 삭제 불가",
  async () => {
    await expect(
      postModel.delete({ id: 999 })
    ).rejects.toThrow("권한이 없습니다");
  }
);

Context 속성 조작

특정 시나리오를 테스트하기 위해 Context 속성을 수정할 수 있습니다:

import { test } from "sonamu/test";
import { Sonamu } from "sonamu";
import { expect } from "vitest";

test("locale별 응답 테스트", async () => {
  const context = Sonamu.getContext();

  // Context 속성 수정
  context.locale = "ko";

  const result = await someService.getLocalizedGreeting();
  expect(result).toBe("안녕하세요");
});

test() vs testAs() 비교

test()
testAs()

비인증 테스트

test("제목", async () => {
  // context.user === null
});

사용 시기:

인증이 필요 없는 공개 API 테스트
인증 여부와 무관한 로직 테스트
Model의 기본 CRUD 작업 테스트

인증된 사용자 테스트

testAs(user, "제목", async () => {
  // context.user === user
});

사용 시기:

로그인이 필요한 API 테스트
사용자별 권한 검증 테스트
소유권 확인이 필요한 기능 테스트

내부 구조

getMockContext()

Mock Context는 다음과 같이 생성됩니다:

function getMockContext(): Context {
  return {
    session: null,
    user: null,
    naiteStore: Naite.createStore(),
    locale: "",
  } as unknown as Context;
}

특징:

session: null로 초기화 (미인증 상태)
user: 기본값은 null (미인증 상태)
naiteStore: 각 테스트마다 독립된 로그 저장소
locale: 빈 문자열로 초기화

AsyncLocalStorage 격리

Sonamu는 Node.js의 AsyncLocalStorage를 사용하여 Context를 격리합니다:

export async function runWithContext(
  context: Context | null, 
  fn: () => Promise<void>
) {
  await Sonamu.asyncLocalStorage.run(
    { context: context ?? getMockContext() }, 
    fn
  );
}

export async function runWithMockContext(fn: () => Promise<void>) {
  await runWithContext(getMockContext(), fn);
}

격리의 이점:

테스트 간 Context가 섞이지 않음
병렬 테스트 실행 시에도 안전
각 테스트는 독립된 Naite 로그 저장소를 가짐

test() 래퍼의 작동 원리

Sonamu의 test() 함수는 Vitest의 test()를 래핑하여 자동으로 Mock Context를 제공합니다:

export const test = async (
  title: string, 
  fn: TestFunction<object>, 
  options?: TestOptions
) => {
  return vitestTest(title, options, async (context) => {
    await runWithMockContext(async () => {
      try {
        await fn(context);
        context.task.meta.traces = Naite.getAllTraces();
      } catch (e: unknown) {
        context.task.meta.traces = Naite.getAllTraces();
        throw e;
      }
    });
  });
};

처리 과정:

Vitest의 test() 실행
runWithMockContext()로 Mock Context 생성 및 활성화
테스트 함수 실행
성공/실패 여부와 무관하게 Naite 로그 수집
Context 자동 정리

testAs() 구조

testAs()는 사용자 정보를 추가로 받아 Context를 확장합니다:

export const testAs = async <User extends AuthContext["user"]>(
  user: User,
  title: string,
  fn: TestFunction<object>,
  options?: TestOptions,
) => {
  return vitestTest(title, options, async (context) => {
    await runWithContext(
      {
        ...getMockContext(),
        user,  // 사용자 정보 추가
      },
      async () => {
        try {
          await fn(context);
          context.task.meta.traces = Naite.getAllTraces();
        } catch (e: unknown) {
          context.task.meta.traces = Naite.getAllTraces();
          throw e;
        }
      },
    );
  });
};

주의사항

Context 사용 시 주의사항:

test() 외부에서 Context 접근 불가: Sonamu.getContext()는 테스트 함수 내부에서만 호출해야 합니다. 외부에서 호출하면 undefined를 반환합니다.
Context 수정의 영향 범위: Context를 수정하면 해당 테스트 전체에 영향을 미칩니다. 테스트 종료 시 자동으로 정리됩니다.

testAs() 파라미터 순서: 사용자 정보가 첫 번째 파라미터입니다.

// ✅ 올바른 사용
testAs(user, "제목", async () => {});

// ❌ 잘못된 사용
testAs("제목", user, async () => {});

타입 안전성: testAs()의 사용자 타입은 AuthContext["user"]를 확장해야 합니다.

다음 단계

Database Mocking

트랜잭션으로 DB 테스트 격리하기

API Mocking

외부 API 호출 모킹하기

Naite 로깅

테스트 로그 기록 및 추적

Bootstrap

테스트 환경 초기화

시작하기

핵심 개념

데이터베이스

API 개발

프론트엔드 통합

테스팅

고급 기능

도구 & CLI

설정

API 레퍼런스

문제 해결

자주 묻는 질문