From 83d65c02a0e73c9b452834a4202be55f97b4f7b4 Mon Sep 17 00:00:00 2001 From: Nicolas De Loof Date: Wed, 12 Aug 2020 14:45:10 +0200 Subject: [PATCH] Fix flacky test Signed-off-by: Nicolas De Loof --- ecs/pkg/amazon/backend/cloudformation_test.go | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/ecs/pkg/amazon/backend/cloudformation_test.go b/ecs/pkg/amazon/backend/cloudformation_test.go index f2a735c51..307b60364 100644 --- a/ecs/pkg/amazon/backend/cloudformation_test.go +++ b/ecs/pkg/amazon/backend/cloudformation_test.go @@ -64,7 +64,6 @@ services: } } assert.Check(t, found, "environment variable FOO not set") - } func TestEnvFileAndEnv(t *testing.T) { @@ -79,8 +78,14 @@ services: `) def := template.Resources["FooTaskDefinition"].(*ecs.TaskDefinition) env := def.ContainerDefinitions[0].Environment - assert.Equal(t, env[0].Name, "FOO") - assert.Equal(t, env[0].Value, "ZOT") + var found bool + for _, pair := range env { + if pair.Name == "FOO" { + assert.Equal(t, pair.Value, "ZOT") + found = true + } + } + assert.Check(t, found, "environment variable FOO not set") } func TestRollingUpdateLimits(t *testing.T) {