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 CLI가 AWS 계정과 상호 작용하고 애플리케이션이 테스트하려는 Lambda 함수에 액세스할 수 있도록 AWS 자격 증명을 구성합니다.

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