From 81abfca4078fee860e7bb744989df01f955905e6 Mon Sep 17 00:00:00 2001 From: Itamar Holder Date: Wed, 19 Jul 2023 13:45:58 +0300 Subject: [PATCH] Add a haveKeys() helper function to match multiple keys Signed-off-by: Itamar Holder --- test/e2e_node/resource_metrics_test.go | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/test/e2e_node/resource_metrics_test.go b/test/e2e_node/resource_metrics_test.go index edb0808f646..20812a02461 100644 --- a/test/e2e_node/resource_metrics_test.go +++ b/test/e2e_node/resource_metrics_test.go @@ -165,3 +165,18 @@ func boundedSample(lower, upper interface{}) types.GomegaMatcher { "Histogram": gstruct.Ignore(), })) } + +func haveKeys(keys ...string) types.GomegaMatcher { + gomega.ExpectWithOffset(1, keys).ToNot(gomega.BeEmpty()) + matcher := gomega.HaveKey(keys[0]) + + if len(keys) == 1 { + return matcher + } + + for _, key := range keys[1:] { + matcher = gomega.And(matcher, gomega.HaveKey(key)) + } + + return matcher +}