diff --git a/test/e2e/instrumentation/logging/stackdrvier/basic.go b/test/e2e/instrumentation/logging/stackdrvier/basic.go index e7ace396a82..865c9bf1a18 100644 --- a/test/e2e/instrumentation/logging/stackdrvier/basic.go +++ b/test/e2e/instrumentation/logging/stackdrvier/basic.go @@ -162,8 +162,11 @@ var _ = instrumentation.SIGDescribe("Cluster level logging implemented by Stackd }() ginkgo.By("Waiting for events to ingest") - zone := framework.TestContext.CloudConfig.Zone - c := utils.NewLogChecker(p, utils.UntilFirstEntryFromZone(zone), utils.JustTimeout, "") + location := framework.TestContext.CloudConfig.Zone + if framework.TestContext.CloudConfig.MultiZone { + location = framework.TestContext.CloudConfig.Region + } + c := utils.NewLogChecker(p, utils.UntilFirstEntryFromLocation(location), utils.JustTimeout, "") err := utils.WaitForLogs(c, ingestionInterval, ingestionTimeout) framework.ExpectNoError(err) }) diff --git a/test/e2e/instrumentation/logging/utils/wait.go b/test/e2e/instrumentation/logging/utils/wait.go index 5992f8ac1b2..ae4b1608bb8 100644 --- a/test/e2e/instrumentation/logging/utils/wait.go +++ b/test/e2e/instrumentation/logging/utils/wait.go @@ -55,12 +55,12 @@ func UntilFirstEntryFromLog(log string) IngestionPred { } } -// UntilFirstEntryFromZone is a IngestionPred that checks that at least one +// UntilFirstEntryFromLocation is a IngestionPred that checks that at least one // entry from the log with a given name was ingested. -func UntilFirstEntryFromZone(zone string) IngestionPred { +func UntilFirstEntryFromLocation(location string) IngestionPred { return func(_ string, entries []LogEntry) (bool, error) { for _, e := range entries { - if e.Location == zone { + if e.Location == location { return true, nil } }