adding agent collaborator test

Author: alexa-perlov

examples/agent-collaborator/.header.md

@@ -1 +1 @@
-This example shows how to deploy a basic Bedrock agent collaborator.
+This example shows how to deploy a basic Bedrock agent collaborator with a supervisor agent and a collaborator agent with agent alias.

examples/agent-collaborator/README.md

@@ -1,5 +1,5 @@
 <!-- BEGIN_TF_DOCS -->
-This example shows how to deploy a basic Bedrock agent collaborator.
+This example shows how to deploy a basic Bedrock agent collaborator with a supervisor agent and a collaborator agent with agent alias.
 
 ## Requirements
 

iam.tf

@@ -5,7 +5,7 @@ locals {
 
 
 resource "aws_iam_role" "agent_role" {
-  count              = var.create_agent || var.create_collaborator ? 1 : 0
+  count              = var.create_agent ? 1 : 0
   assume_role_policy = data.aws_iam_policy_document.agent_trust[0].json
   name_prefix        = var.name_prefix
 }

tests/05_agent_collaborator.tftest.hcl

@@ -0,0 +1,13 @@
+run "agent_collaborator_plan" {
+  command = plan
+  module {
+    source = "./examples/agent-collaborator"
+  }
+}
+
+run "agent_collaborator_apply" {
+  command = apply
+  module {
+    source = "./examples/agent-collaborator"
+  }
+}