Amazon Bedrock AgentCore を使用して Lambda テストイベントの生成を自動化する

リポジトリのクローン作成

GitHub リポジトリをローカルシステムにクローンし、プロジェクトディレクトリに移動します。

git clone https://github.com/aws-samples/sample-lambda-test-event-generator.git

cd sample-lambda-test-event-generator

このリポジトリには、Python アプリケーション、CloudFormation テンプレート、および設定ファイルが含まれています。

AWS 認証情報を設定します。

AWS 認証情報を設定して、AWS CLI が AWS アカウントとやり取りし、アプリケーションがテストする Lambda 関数にアクセスできるようにします。

これを行うには、AWS CLI 設定コマンドを使用します。

aws configure

プロンプトが表示されたら、次の情報を入力します。

CloudFormation を使用してインフラストラクチャをデプロイします。

作成されるもの:

設定変数をエクスポートします。

CloudFormation スタック出力からすべての値を環境変数としてエクスポートします。

export AGENTCORE_ROLE_ARN=$(aws cloudformation describe-stacks \
      --stack-name lambda-test-generator-infra \
      --query 'Stacks[0].Outputs[?OutputKey==`AgentCoreExecutionRoleArn`].OutputValue' \
      --output text)

export DISCOVERY_URL=$(aws cloudformation describe-stacks \
      --stack-name lambda-test-generator-infra \
      --query 'Stacks[0].Outputs[?OutputKey==`DiscoveryUrl`].OutputValue' \
      --output text)

export CLIENT_ID=$(aws cloudformation describe-stacks \
      --stack-name lambda-test-generator-infra \
      --query 'Stacks[0].Outputs[?OutputKey==`ClientId`].OutputValue' \
      --output text)

export DYNAMODB_TABLE=$(aws cloudformation describe-stacks \
      --stack-name lambda-test-generator-infra \
      --query 'Stacks[0].Outputs[?OutputKey==`DynamoDBTableName`].OutputValue' \
      --output text)

export COGNITO_POOL_ID=$(aws cloudformation describe-stacks \
      --stack-name lambda-test-generator-infra \
      --query 'Stacks[0].Outputs[?OutputKey==`UserPoolId`].OutputValue' \
      --output text)

export BEDROCK_GUARDRAIL_ID=$(aws cloudformation describe-stacks \
  --stack-name lambda-test-generator-infra \
  --query 'Stacks[0].Outputs[?OutputKey==`BedrockGuardrailId`].OutputValue' \
  --output text)

export BEDROCK_GUARDRAIL_VERSION=$(aws cloudformation describe-stacks \
  --stack-name lambda-test-generator-infra \
  --query 'Stacks[0].Outputs[?OutputKey==`BedrockGuardrailVersion`].OutputValue' \
  --output text)

これらの変数は、AgentCore 設定と .env ファイル作成の後続のステップで使用されます。

仮想環境を作成します。

AgentCore を設定してデプロイします。

ローカル開発用の .env ファイルを作成します。

すべての設定値を使用して、プロジェクトルートディレクトリに .env ファイルを作成します。

cat > .env << EOF
AWS_REGION=us-east-1
DYNAMODB_TABLE_NAME=$DYNAMODB_TABLE
COGNITO_POOL_ID=$COGNITO_POOL_ID
COGNITO_CLIENT_ID=$CLIENT_ID
COGNITO_REGION=us-east-1
AGENTCORE_ENDPOINT=$AGENTCORE_ENDPOINT
BEDROCK_GUARDRAIL_ID=$BEDROCK_GUARDRAIL_ID
BEDROCK_GUARDRAIL_VERSION=$BEDROCK_GUARDRAIL_VERSION
EOF

AWS サービスへの接続をテストします。

aws dynamodb describe-table --table-name $DYNAMODB_TABLE
aws bedrock list-foundation-models --region us-east-1

どちらのコマンドも成功したレスポンスを返すはずです。アクセス許可エラーが発生した場合は、IAM ポリシーが正しく設定されていることを確認します。

新しい Cognito ユーザーを作成します。

Cognito ユーザープールは、セキュリティのために管理者専用のユーザー作成で設定されているため、ユーザーは自己登録できません。CloudFormation スタック出力からエクスポートされた Cognito プール ID とクライアント ID を使用して、AWS CLI 経由でユーザーを作成します。新しいユーザーを作成します ( を E メールuser@example.comに置き換えます）。

aws cognito-idp admin-create-user \
  --user-pool-id $COGNITO_POOL_ID \
  --username user@example.com \
  --user-attributes Name=email,Value=user@example.com Name=email_verified,Value=true \
  --temporary-password '[PASSWORD]!' \
  --region us-east-1

永続的なパスワードを設定します (8 文字以上で、大文字、小文字、数字を含める必要があります）。

aws cognito-idp admin-set-user-password \
  --user-pool-id $COGNITO_POOL_ID \
  --username user@example.com \
  --password '[PASSWORD]!' \
  --permanent \
  --region us-east-1

これらの認証情報を使用して、次のステップで Streamlit UI を通じてサインインします。

Streamlit UI を起動します。

アプリケーションを起動します。

streamlit run app.py

UI がデフォルトのブラウザで で開きます。 http://localhost:8501

前のステップで作成した認証情報を使用してサインインする

生成オプションを設定する

UI で、次のオプションを設定します。

テストケースを生成します。

「テストケースの生成」ボタンをクリックします。システムは以下を行います。

生成されたテストを確認します。

生成された各テストケースを確認します。これには、以下が含まれます。

フィードバックを提供します。

テストケースごとに、フィードバックを提供します。

テストケースを受け入れるには:

テストケースを拒否するには:

フィードバックはメモリに保存します。

すべてのテストケースを確認した後:

繰り返して改善します。

同じ関数のテストを 2～3 回生成して品質を向上させます。

AgentCore API を直接呼び出します。

自動化と統合では、Cognito 認証で API コールを使用して AgentCore バックエンドを直接呼び出します。

Cognito トークンを取得します。

TOKEN=$(agentcore identity get-cognito-inbound-token)

テスト生成を呼び出す:

agentcore invoke --bearer-token "$TOKEN" '{"action": "generate_test_cases", "function_name": "your-lambda-function", "num_test_cases": 10,"custom_instructions": "Focus on authentication scenarios", "target_filter": "validate_user","ignore_patterns": ["tests/", "*.test.js"] }'

または、curl を使用します。

curl -X POST "$AGENTCORE_ENDPOINT" \ -H "Authorization: Bearer $TOKEN" \ -H "Content-Type: application/json" \ -d '{ "action": "generate_test_cases", "function_name": "your-lambda-function", "num_test_cases": 10}'

DynamoDB パターンをモニタリングし、コストを追跡します。

デプロイされたリソースを削除します。

デプロイされたリソースをすべて削除するには:

AgentCore エージェントを削除します。

agentcore destroy

CloudFormation スタックを削除します (DynamoDB テーブル、Cognito ユーザープール、IAM ロールを削除します）。

aws cloudformation delete-stack 
\ --stack-name lambda-test-generator-infra 
\ --region us-east-1