GPT로 테스트 코드 생성하는 법 (Jest, Pytest, JUnit 활용법)

테스트 코드는 소프트웨어 품질을 유지하고, 기능이 올바르게 동작하는지를 검증하는 핵심 도구입니다. 하지만 테스트 코드를 처음부터 작성하는 일은 많은 시간과 기술적 부담을 요구하죠. GPT는 이러한 과정을 자동화해줍니다. 함수나 클래스의 코드를 기반으로, 다양한 테스트 프레임워크에 맞는 테스트 코드를 손쉽게 생성할 수 있습니다. 이 글에서는 GPT를 활용해 Jest, Pytest, JUnit 등 실제 개발 현장에서 널리 사용되는 프레임워크에 맞는 테스트 코드를 생성하는 방법을 소개합니다.

Jest 기반 테스트 코드 생성하기 (JavaScript/TypeScript)

Node.js 환경에서 가장 많이 사용되는 테스트 프레임워크 중 하나는 Jest입니다. GPT는 함수나 모듈 코드를 기반으로 Jest 형식의 테스트 코드를 자동으로 생성할 수 있습니다.

function add(a, b) {
  return a + b;
}

const { add } = require('./math');

describe('add 함수 테스트', () => {
  test('두 수를 더한다', () => {
    expect(add(2, 3)).toBe(5);
  });

  test('음수 더하기도 가능하다', () => {
    expect(add(-2, -3)).toBe(-5);
  });
});

TypeScript를 사용하는 프로젝트라면 타입을 고려한 Jest 코드도 생성 가능합니다.

Pytest 기반 테스트 코드 생성하기 (Python)

Python 개발자들에게 Pytest는 가장 많이 사용되는 테스트 프레임워크입니다. GPT는 Python 함수의 입력과 출력 타입을 분석하여, 자동으로 Pytest 형식에 맞는 테스트 코드를 생성할 수 있습니다.

def multiply(a, b):
    return a * b

from your_module import multiply

def test_multiply_positive_numbers():
    assert multiply(2, 3) == 6

def test_multiply_negative_numbers():
    assert multiply(-2, -3) == 6

def test_multiply_with_zero():
    assert multiply(0, 10) == 0

JUnit 기반 테스트 코드 생성하기 (Java)

public class Calculator {
  public int subtract(int a, int b) {
    return a - b;
  }
}

import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.assertEquals;

public class CalculatorTest {

  Calculator calculator = new Calculator();

  @Test
  void testSubtractPositive() {
    assertEquals(2, calculator.subtract(5, 3));
  }

  @Test
  void testSubtractNegative() {
    assertEquals(-1, calculator.subtract(2, 3));
  }
}

Mockito와 함께 사용할 경우:

@Mock
UserRepository userRepository;

@InjectMocks
UserService userService;

@BeforeEach
void setUp() {
  MockitoAnnotations.openMocks(this);
}

@Test
void testGetUser() {
  when(userRepository.findById(1L)).thenReturn(Optional.of(new User(1L, "John")));
  User user = userService.getUser(1L);
  assertEquals("John", user.getName());
}

 

GPT는 테스트 코드 작성에 있어 막강한 자동화 도구입니다. 함수나 클래스 코드만 입력하면, Jest, Pytest, JUnit 등 다양한 프레임워크에 맞는 테스트 코드를 자동 생성해줍니다. 예외 처리, mock 테스트, 경계 조건 등도 자동 반영되며, 초보자도 고급 테스트 코드를 손쉽게 완성할 수 있습니다. 지금 바로 GPT를 테스트 코드 생성 도구로 활용해보세요. 코드 품질과 생산성이 동시에 향상됩니다.