[파이썬] 파이썬 애플리케이션의 AWS Lambda 테스트와 디버깅

05 Sep 2023

python

AWS Lambda는 서버리스 아키텍처를 위한 서비스로서, 파이썬 애플리케이션을 효과적으로 실행하고 관리할 수 있는 환경을 제공합니다. Lambda를 사용하면 별도의 서버 설정이나 관리 없이 애플리케이션 코드를 실행할 수 있어 개발자들에게 많은 편의를 제공합니다.

그러나 Lambda를 개발하고 테스트하며 디버깅하는 과정은 종종 도전적일 수 있습니다. 이 글에서는 파이썬 애플리케이션을 AWS Lambda에서 테스트하고 디버깅하는 방법에 대해 알아보겠습니다.

테스트와 디버깅을 위한 환경 설정

Lambda에서 파이썬 애플리케이션을 테스트하려면 몇 가지 환경 설정이 필요합니다.

가상 환경(Virtual Environment) 설정: Lambda에서 사용하는 런타임 버전에 맞춰 파이썬 가상 환경을 설정합니다.
필요한 패키지 설치: 애플리케이션에 필요한 패키지들을 설치합니다. 이때, requirements.txt 파일을 작성하여 필요한 패키지를 명시할 수 있습니다.
이벤트(Mock Event) 생성: Lambda 함수의 이벤트를 테스트하기 위해 이벤트 객체를 생성합니다.
Lambda 함수 테스트 코드 작성: 테스트하려는 Lambda 함수의 테스트 코드를 작성합니다. 이는 함수의 입력 값을 확인하고 예상한 결과 값을 검증하는 용도로 사용됩니다.

테스트 방법

Lambda 함수를 테스트하는 방법은 다양하지만, 여기서는 대표적인 방법인 로컬 테스트와 단위 테스트에 대해 알아보겠습니다.

로컬 테스트

로컬에서 Lambda 함수를 테스트하기 위해서는 AWS Lambda runtime을 로컬에서 실행할 수 있도록 설정해야 합니다. 이를 위해 localstack과 같은 도구를 사용할 수 있습니다. Localstack은 AWS 클라우드 환경을 로컬에서 에뮬레이트하는 도구로서, Lambda 함수를 로컬에서 실행하고 테스트할 수 있도록 도와줍니다.

로컬 테스트를 위해 다음 단계를 따릅니다.

Localstack 설치: pip install localstack
Docker 실행: Localstack은 Docker를 사용해야 하므로 Docker를 실행합니다.
Lambda 함수 실행: localstack start 명령어를 실행하여 Localstack이 시작되면, Lambda 함수를 실행합니다.
API Gateway 설정: 필요한 경우 API Gateway를 설정하여 Lambda 함수를 트리거할 수 있도록 설정합니다.
요청 전송: 설치된 AWS CLI 또는 cURL 등을 사용하여 API Gateway로 요청을 전송하고 응답을 확인합니다.

단위 테스트

Lambda 함수의 단위 테스트를 위해서는 unittest와 mock 모듈을 사용할 수 있습니다. 예를 들어, Lambda 함수의 입력 값을 테스트하고 예상한 결과 값을 검증하는 단위 테스트 코드를 작성할 수 있습니다.

import unittest
from unittest import mock
from lambda_function import handler

class TestLambdaFunction(unittest.TestCase):
    def test_handler(self):
        event = {
            "key": "value"
        }
        
        expected_response = {
            "statusCode": 200,
            "body": "Hello, value!"
        }
        
        with mock.patch("lambda_function.function_dependency") as mock_function:
            mock_function.return_value = "Hello"
            
            response = handler(event, {})
            
            self.assertEqual(response, expected_response)
            
if __name__ == "__main__":
    unittest.main()

위의 예제에서 lambda_function은 테스트할 Lambda 함수의 코드를 포함하고 있습니다. function_dependency는 Lambda 함수에서 호출하는 외부 의존성을 나타내며, 모의 객체(Mock Object)로서 대체됩니다.

디버깅

Lambda 함수를 디버깅하기 위해서는 print 문을 사용하거나, 로그 관리 도구를 활용할 수 있습니다.

Lambda 함수를 디버깅하기 위한 몇 가지 팁은 다음과 같습니다.

print 문을 사용하여 변수의 값을 출력하고 확인할 수 있습니다. 이를 통해 함수의 흐름과 값을 추적할 수 있습니다.
AWS CloudWatch 로그를 사용하면 Lambda 함수의 실행 로그를 확인할 수 있습니다.
try-except 문을 사용하여 예외 정보를 로깅하고 처리하는 코드를 추가할 수 있습니다.

마무리

이 글에서는 파이썬 애플리케이션을 AWS Lambda에서 테스트하고 디버깅하는 방법에 대해 소개했습니다. 로컬 테스트와 단위 테스트를 사용하여 Lambda 함수를 확인하고, print 문을 통해 값을 출력하고 CloudWatch 로그를 사용하여 디버깅하는 방법을 알아보았습니다. 효과적인 테스트와 디버깅은 Lambda 애플리케이션의 개발과 운영에 있어서 중요한 요소이므로, 개발자들은 이러한 기술을 숙지하고 활용하는 것이 좋습니다.