Outsider's Dev Story

Stay Hungry. Stay Foolish. Don't Be Satisfied.
RetroTech 팟캐스트 44BITS 팟캐스트

LambCI의 Docker 이미지로 AWS Lambda 함수 로컬에서 테스트하기

AWS Lambda를 사용할 때 불편한 점은 Lambda에 배포해야 테스트를 해볼 수 있다는 부분이다. 그동안 Lambda를 쓰면서 느낀 경험은 최대한 로직은 유닛테스트로 기능을 확인하고 Lambda의 핸들러에서는 간단히 이 로직은 호출만 하는 정도로 작성하면 로컬에서 테스트를 해보면서 코드를 작성할 수 있다. 코드를 수정할 때마다 Lambda에 배포하는 것은 매우 불편한 일이므로 로컬에서 테스트할 수 있으면 시간을 많이 줄일 수 있다.

Node.js이든 Python이든 간단한 코드면 유닛테스트로 충분하지만 좀 더 복잡하게 사용할 일이 있다면 Lambda에 배포를 해야 하므로 매우 귀찮을 일이 된다. 특히 얼마 전에 사용해 본 Headless Chrome을 Lambda에서 사용하는 작업은 Lambda 환경의 제약을 확인해 봐야 하므로 잦은 배포를 해야 해서 아주 귀찮을 일이었다. 특히 Lambda 함수의 용량이 크다면 배포할 때마다 시간도 걸리기 때문에 더욱 작업이 번거로운 일이 된다.

lambci/docker-lambda

LambCI는 Lambda에서 CI를 할 수 있게 하는 서버리스 솔루션인데 여기서 Lambda 환경을 거의 같게 만든 Docker를 제공하고 있다. 내가 작업한 것은 아니지만 LambCI는 nodejs-ko에서 배포용으로 쓰고 있어서 알고는 있었지만 docker-lambda는 모르고 있다가 얼마 전에 AWS의 Lambda를 로컬에서 테스트해볼 수 있게 하는 SAM Local을 보다가 여기서 로컬 테스트 용도로 LambCI의 docker-lambda 이미지를 사용한다는 것을 알게 되어 Docker만 사용해보았다.(sam-local의 경우 CloudFormation기반이라서 테스트는 해보지 않았다.)

테스트 목적으로 docker-lambda를 사용해봤는데 100% Lambda와 같지는 않아도 거의 비슷하게 환경이 구성되어 있으므로 로컬에서 쉽게 테스트해볼 수 있다. 사용해본 느낌으로는 몇 가지 차이점만 실제 AWS Lambda에 배포해서 테스트해보고 나머지는 docker-lambda에서 동작 여부를 확인해 보는 정도로 충분해서 아주 편했다. AWS Lambda에서 안 되는 경우 docker-lambda에서도 안 되는지 확인하고 docker-lambda에서 동작하도록 코드를 수정한 뒤에 AWS Lambda에 배포하는 식으로 작업하니까 시간을 꽤 줄일 수 있었다. 물론 docker-lambda에서는 동작하지만 다른 환경 문제고 AWS Lambda에서는 안 되는 경우가 없는 것은 아니다.

docker-lambda로 Lambda 함수 실행하기

Lambda 함수를 LambCI의 Docker 이미지로 테스트해볼 수 있는데 Docker Hub에 환경별로 올려져 있다. nodejs4.3, 'nodejs', 'nodejs6.10', 'python2.7', 'python3.6', 'java8' 등 Lambda에서 지원하는 환경이 모두 Docker 이미지로 존재한다.

Node.js를 사용한다고 할 때 다음과 같은 index.js가 있다고 해보자.

// index.js
console.log('starting function');
exports.handler = (event, context, callback) => {
  console.log('event:', event);
  context.succeed('hello world');
};

여기서는 코드의 내용이 중요하진 않으므로 간단한 Lambda 코드만 사용했다. 터미널에서 Docker 이미지로 다음과 같이 Lambda 함수를 실행할 수 있다.

$ docker run -v "$PWD":/var/task lambci/lambda:nodejs6.10
START RequestId: 543c5021-013e-17b4-ffea-e179e78c5b7b Version: $LATEST
2017-08-27T16:35:19.970Z  543c5021-013e-17b4-ffea-e179e78c5b7b  starting function
2017-08-27T16:35:19.972Z  543c5021-013e-17b4-ffea-e179e78c5b7b  event: {}
END RequestId: 543c5021-013e-17b4-ffea-e179e78c5b7b
REPORT RequestId: 543c5021-013e-17b4-ffea-e179e78c5b7b  Duration: 7.65 ms Billed Duration: 100 ms Memory Size: 1536 MB  Max Memory Used: 28 MB

"hello world"

위에서 보듯이 현재 폴더에 있는 Lambda 함수를 nodejs6.10 환경에서 실행해 볼 수 있고 실제 AWS Lambda에서 볼 수 있는 로그처럼 실행 결과까지 모두 볼 수 있다. Node.js 6.10 환경이 아니라면 다른 이미지를 사용하면 되고 -v "$PWD":/var/task에서 볼 수 있듯이 현재 폴더를 Docker 내에서 실행하게 된다.

// dist.js
console.log('starting function');
exports.entry = (event, context, callback) => {
  console.log('event:', event);
  context.succeed('hello world');
};

위처럼 기본값인 index.jshandler 함수를 사용하지 않고 dist.js 파일에 entry라는 함수를 진입점으로 쓰고 싶다면 Docker 명령어 마지막에 dist.entry처럼 핸들러를 지정해 주면 된다.(AWS Lambda의 설정에서도 handler를 지정할 수 있다.)

docker run -v "$PWD":/var/task lambci/lambda:nodejs6.10 dist.entry
START RequestId: d5f50084-0ea8-1530-8183-dc2fce92b7b4 Version: $LATEST
2017-08-27T16:45:18.067Z  d5f50084-0ea8-1530-8183-dc2fce92b7b4  starting function
2017-08-27T16:45:18.069Z  d5f50084-0ea8-1530-8183-dc2fce92b7b4  event: {}
END RequestId: d5f50084-0ea8-1530-8183-dc2fce92b7b4
REPORT RequestId: d5f50084-0ea8-1530-8183-dc2fce92b7b4  Duration: 8.79 ms Billed Duration: 100 ms Memory Size: 1536 MB  Max Memory Used: 28 MB

"hello world"

지금까지는 입력 이벤트가 없이 실행했는데 입력 이벤트가 있는 경우에는 docker 명령어 마지막에 JSON을 전달하면 된다.

$ docker run -v "$PWD":/var/task lambci/lambda:nodejs6.10 dist.entry '{ "name": "outsider" }'
START RequestId: b86d5ed1-293d-1be5-9fc6-fb87968e741c Version: $LATEST
2017-08-27T16:48:21.068Z  b86d5ed1-293d-1be5-9fc6-fb87968e741c  starting function
2017-08-27T16:48:21.070Z  b86d5ed1-293d-1be5-9fc6-fb87968e741c  event: { name: 'outsider' }
END RequestId: b86d5ed1-293d-1be5-9fc6-fb87968e741c
REPORT RequestId: b86d5ed1-293d-1be5-9fc6-fb87968e741c  Duration: 11.63 ms  Billed Duration: 100 ms Memory Size: 1536 MB  Max Memory Used: 28 MB

"hello world"

위 로그에서 보듯이 전달한 JSON이 출력된 것을 볼 수 있다.

Headless Chrome을 사용하면서 꽤 많은 테스트를 해보았을 때 Lambda 환경이 거의 그대로 구현되어 있어서 테스트를 상당히 편리하게 도와준다. 그리고 sam-local에서도 채택한 걸 보면 LambCI에서 제공하는 Docker 이미지가 신뢰할 수 있는 퀄리티를 제공하고 있다고 생각된다. AWS에서 직접 공식 이미지를 제공해 주면 아주 좋겠지만....

2017/08/28 01:54 2017/08/28 01:54