AWS API Gateway + Lambda (Python) でモックを作ってみよう

#前回記事での失敗
前回記事のAWS API Gatewayでモックを作ってみようで、API Gatewayの統合タイプ：MockでAPIのモックを作ってみました。
が、統合タイプ：MockではPOSTのリクエストデータが取得できないようでした。
（参考：teratail　AWS API GatewayでHTTPリクエストのjsonを取得したい）
そのため今回は統合タイプ：Mockを使わず、統合タイプ：LAMBDAで作っていくことにします。

#PythonでLambda関数を作る

1. Amazon Lambdaコンソールで関数の作成ボタンから関数の作成画面に遷移する
2. 基本的な情報の関数名に任意の関数名を入力する
3. 基本的な情報のランタイムで使用する言語を選択する（この記事ではPython 3.7を選んでいます）
4. 基本的な情報のアクセス権限でロールを選択する（ロールの説明は省略 説明できるほど詳しくないので）
5. 画面右下の関数の作成ボタンをクリックする
6. 関数を作成する画面に遷移するので、関数コード欄にPythonコードを打ち込んでいく

##Lambdaプロキシ統合を使用しない場合

• API Gatewayでの設定
• 統合タイプ：Lambda関数
• Lambdaプロキシ統合の使用：チェックOff
• Lambda関数コード

import json
def lambda_handler(event, context):
    return {
        'test_data': 123456,
        'test_message': 'Hello, World!'
    }

上記のように記述すると、ステータス 200としてreturnの内容がレスポンス本文として返されます。
ステータス 400等で返す場合は

import json

def lambda_handler(event, context):
    raise Exception('400')

というような形で例外を発生させます。
例外を発生させた場合のレスポンス本文は、API Gatewayの統合レスポンスで設定します。
API Gatewayでメソッドレスポンスに該当ステータス（この場合は400）を追加し、統合レスポンスの該当ステータスのマッピングテンプレートにレスポンス本文を入力します。
##Lambdaプロキシ統合を使用する場合

• API Gatewayでの設定
• 統合タイプ：Lambda関数
• Lambdaプロキシ統合の使用：チェックOn
• Lambda関数コード

import json
def lambda_handler(event, context):
    return {
        'statusCode': 200,
        'body': json.dumps(
            {
                'test': 123456,
                'test_message': 'Hello, World!'
            }
        )
    }

ステータス 400等で返す場合は、statusCodeとbodyを書き換えます。
Lambdaプロキシ統合を使用している場合はAPI Gatewayの統合レスポンスでレスポンス本文を設定する必要はありません。（というかLambdaプロキシ統合を使用している場合は統合レスポンスは触れない）
Lambdaプロキシ統合を使用する場合の出力の基本は以下になります。（公式ドキュメントより）

{
  "isBase64Encoded" : "boolean",
  "statusCode": "number",
  "headers": { ... },
  "body": "JSON string"
}

#複数のリソースからのリクエストを振り分ける
Lambdaプロキシ統合を使用するとAPI GatewayのリソースパスをLambda関数で受け取ることができるので、リソース毎にLambda関数を作るのではなくLambda関数を一つにまとめてみましょう。
API Gatewayのリソースパスはevent['path']で取得できます。

import json

def lambda_handler(event, context):
    path = event['path']

    if path == '/proxy200':
        return {
            'statusCode': 200,
            'body': json.dumps('Good Request!') 
        }
    elif path == '/proxy400':
        return {
            'statusCode': 400,
            'body': json.dumps('Bad Request..')
        }

（API Gatewayで/proxy200と/proxy400というというリソースを作成した場合）
このようにevent['path']の値で処理内容を分ければ、一つのLambda関数で複数のリソースに対応できます。
#次回
前回と今回の内容で、API Gatewayでのモック作りがある程度出来るようになりました。
次回はSlackからAPI Gatewayモック作りを行えるようにしてみます。

#プチ連載
1回目　AWS API Gatewayでモックを作ってみよう
2回目　AWS API Gateway + Lambda (Python) でモックを作ってみよう（本記事）
3回目　Slack + Lambda (Python) でAWS API Gatewayモックを作ってみよう

#We're hiring!
AIチャットボットを開発しています。
ご興味ある方は Wantedlyページ からお気軽にご連絡ください！

#参考記事
Qiita 【API Gateway】AWS Lambda統合のPythonでHello, world
AWSドキュメント API Gateway で Lambda エラーを処理する