Skip to main content
LangGraph 에이전트의 프로토타입을 만든 후, 자연스러운 다음 단계는 테스트를 추가하는 것입니다. 이 가이드는 단위 테스트를 작성할 때 사용할 수 있는 유용한 패턴들을 다룹니다. 이 가이드는 LangGraph 전용이며 커스텀 구조를 가진 그래프와 관련된 시나리오를 다룬다는 점에 유의하세요. 만약 막 시작하는 단계라면, LangChain의 내장 @[create_agent]를 사용하는 이 섹션을 확인하세요.

​
사전 요구사항

먼저 vitest가 설치되어 있는지 확인하세요: 
$ npm install -D vitest

​
시작하기

많은 LangGraph 에이전트가 상태에 의존하기 때문에, 유용한 패턴은 사용할 각 테스트 전에 그래프를 생성한 다음, 테스트 내에서 새로운 체크포인터 인스턴스로 컴파일하는 것입니다. 아래 예제는 node1과 node2를 거쳐 진행되는 간단한 선형 그래프에서 이것이 어떻게 작동하는지 보여줍니다. 각 노드는 단일 상태 키 my_key를 업데이트합니다: 
import { test, expect } from 'vitest';
import {
  StateGraph,
  START,
  END,
  MemorySaver,
} from '@langchain/langgraph';
import { z } from "zod/v4";

const State = z.object({
  my_key: z.string(),
});

const createGraph = () => {
  return new StateGraph(State)
    .addNode('node1', (state) => ({ my_key: 'hello from node1' }))
    .addNode('node2', (state) => ({ my_key: 'hello from node2' }))
    .addEdge(START, 'node1')
    .addEdge('node1', 'node2')
    .addEdge('node2', END);
};

test('basic agent execution', async () => {
  const uncompiledGraph = createGraph();
  const checkpointer = new MemorySaver();
  const compiledGraph = uncompiledGraph.compile({ checkpointer });
  const result = await compiledGraph.invoke(
    { my_key: 'initial_value' },
    { configurable: { thread_id: '1' } }
  );
  expect(result.my_key).toBe('hello from node2');
});

​
개별 노드와 엣지 테스트하기

컴파일된 LangGraph 에이전트는 각 개별 노드에 대한 참조를 graph.nodes로 노출합니다. 이를 활용하여 에이전트 내의 개별 노드를 테스트할 수 있습니다. 이 방식은 그래프를 컴파일할 때 전달된 체크포인터를 우회한다는 점에 유의하세요: 
import { test, expect } from 'vitest';
import {
  StateGraph,
  START,
  END,
  MemorySaver,
} from '@langchain/langgraph';
import { z } from "zod/v4";

const State = z.object({
  my_key: z.string(),
});

const createGraph = () => {
  return new StateGraph(State)
    .addNode('node1', (state) => ({ my_key: 'hello from node1' }))
    .addNode('node2', (state) => ({ my_key: 'hello from node2' }))
    .addEdge(START, 'node1')
    .addEdge('node1', 'node2')
    .addEdge('node2', END);
};

test('individual node execution', async () => {
  const uncompiledGraph = createGraph();
  // 이 예제에서는 무시됩니다
  const checkpointer = new MemorySaver();
  const compiledGraph = uncompiledGraph.compile({ checkpointer });
  // node 1만 실행합니다
  const result = await compiledGraph.nodes['node1'].invoke(
    { my_key: 'initial_value' },
  );
  expect(result.my_key).toBe('hello from node1');
});

​
부분 실행

더 큰 그래프로 구성된 에이전트의 경우, 전체 플로우를 처음부터 끝까지 테스트하는 대신 에이전트 내의 부분 실행 경로를 테스트하고 싶을 수 있습니다. 경우에 따라서는 이러한 섹션을 서브그래프로 재구조화하는 것이 의미상 적절할 수 있으며, 이렇게 하면 일반적인 방식으로 독립적으로 호출할 수 있습니다. 하지만 에이전트 그래프의 전체 구조를 변경하고 싶지 않다면, LangGraph의 지속성 메커니즘을 사용하여 원하는 섹션의 시작 직전에서 에이전트가 일시 정지되고 원하는 섹션의 끝에서 다시 일시 정지되는 상태를 시뮬레이션할 수 있습니다. 단계는 다음과 같습니다:
  1. 체크포인터와 함께 에이전트를 컴파일합니다 (테스트용으로는 인메모리 체크포인터 MemorySaver로 충분합니다).
  2. 테스트를 시작하려는 노드의 이전 노드 이름으로 설정된 asNode 파라미터와 함께 에이전트의 update_state 메서드를 호출합니다.
  3. 상태를 업데이트할 때 사용한 것과 동일한 thread_id와 중단하려는 노드 이름으로 설정된 interruptBefore 파라미터로 에이전트를 호출합니다.
다음은 선형 그래프에서 두 번째와 세 번째 노드만 실행하는 예제입니다: 
import { test, expect } from 'vitest';
import {
  StateGraph,
  START,
  END,
  MemorySaver,
} from '@langchain/langgraph';
import { z } from "zod/v4";

const State = z.object({
  my_key: z.string(),
});

const createGraph = () => {
  return new StateGraph(State)
    .addNode('node1', (state) => ({ my_key: 'hello from node1' }))
    .addNode('node2', (state) => ({ my_key: 'hello from node2' }))
    .addNode('node3', (state) => ({ my_key: 'hello from node3' }))
    .addNode('node4', (state) => ({ my_key: 'hello from node4' }))
    .addEdge(START, 'node1')
    .addEdge('node1', 'node2')
    .addEdge('node2', 'node3')
    .addEdge('node3', 'node4')
    .addEdge('node4', END);
};

test('partial execution from node2 to node3', async () => {
  const uncompiledGraph = createGraph();
  const checkpointer = new MemorySaver();
  const compiledGraph = uncompiledGraph.compile({ checkpointer });
  await compiledGraph.updateState(
    { configurable: { thread_id: '1' } },
    // node 2에 전달되는 상태 - node 1의
    // 끝에서의 상태를 시뮬레이션합니다
    { my_key: 'initial_value' },
    // node 1에서 온 것처럼 저장된 상태를 업데이트합니다
    // 실행은 node 2에서 재개됩니다
    'node1',
  );
  const result = await compiledGraph.invoke(
    // None을 전달하여 실행을 재개합니다
    null,
    {
      configurable: { thread_id: '1' },
      // node 3 이후에 중단하여 node 4가 실행되지 않도록 합니다
      interruptAfter: ['node3']
    },
  );
  expect(result.my_key).toBe('hello from node3');
});
Edit the source of this page on GitHub.
Connect these docs programmatically to Claude, VSCode, and more via MCP for real-time answers.