test: fix tests

Signed-off-by: Darren Murray <darren.murray@lacework.net>
lacework · Jul 31, 2023 · a69bf52 · a69bf52
1 parent fe35e9e
commit a69bf52
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/integration/aws_controltower_generation_test.go b/integration/aws_controltower_generation_test.go
@@ -96,7 +96,7 @@ func TestGenerationControlTowerPromptOrgAccountMappings(t *testing.T) {
 				MsgRsp{cmd.QuestionAwsControlTowerCoreAuditProfile, "AWSAdministratorAccess"},
 				MsgRsp{cmd.QuestionAwsControlTowerCoreAuditRegion, "us-east-1"},
 				MsgRsp{cmd.QuestionAwsControlTowerConfigureAdvanced, "y"},
-				MsgMenu{cmd.ControlTowerAdvancedOptMappings, 3},
+				MsgMenu{cmd.ControlTowerAdvancedOptMappings, 5},
 				MsgRsp{cmd.QuestionControlTowerOrgAccountMappingsLWDefaultAccount, "main"},
 				MsgRsp{cmd.QuestionControlTowerOrgAccountMappingsLWAccount, "sub-account-1"},
 				MsgMulitlineRsp{cmd.QuestionControlTowerOrgAccountMappingsAwsAccounts, []string{"123456789011"}},
@@ -212,7 +212,8 @@ func TestGenerationControlTowerPromptOptionalAttributes(t *testing.T) {
 				MsgRsp{cmd.QuestionAwsControlTowerCoreIamRoleExternalID, "01234567"},
 				MsgRsp{cmd.QuestionControlTowerAnotherAdvancedOpt, "y"},
 				MsgMenu{cmd.ControlTowerIntegrationPrefixOpt, 3},
-				MsgRsp{cmd.QuestionControlTowerAnotherAdvancedOpt, "prefix-"},
+				MsgRsp{cmd.QuestionControlTowerPrefix, "prefix-"},
+				MsgRsp{cmd.QuestionControlTowerAnotherAdvancedOpt, "y"},
 				MsgMenu{cmd.ControlTowerIntegrationSqsOpt, 4},
 				MsgRsp{cmd.QuestionControlTowerSqsQueueName, "lw-queue-name"},
 				MsgRsp{cmd.QuestionControlTowerAnotherAdvancedOpt, "n"},
@@ -248,7 +249,6 @@ func TestGenerationControlTowerPromptOptionalAttributes(t *testing.T) {
 			aws_controltower.NewAwsSubAccount("AWSAdministratorAccess", "us-east-1", "audit")),
 		aws_controltower.WithOrgAccountMappings(orgAccountMappings),
 		aws_controltower.WithPrefix("prefix-"),
-		aws_controltower.WithEnableLogFileValidation(),
 		aws_controltower.WithSqsQueueName("lw-queue-name"),
 		aws_controltower.WithExisitingIamRole("arn:aws:iam::12345678:role/test",
 			"lw-role-name", "01234567")).Generate()