使用 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 建立使用者。建立新的使用者 (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