From bcfe3726e281a694484450a58df7d45e4a4ebf09 Mon Sep 17 00:00:00 2001 From: Lucas Bajolet Date: Mon, 6 May 2024 19:38:34 +0000 Subject: [PATCH] backport of commit 4d6d3cc81e27c9253dc60d2c30ffd0a681afcbd6 --- test/gadgets_test.go | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/test/gadgets_test.go b/test/gadgets_test.go index 9ad4819d8..03ee4c35e 100644 --- a/test/gadgets_test.go +++ b/test/gadgets_test.go @@ -132,3 +132,20 @@ func (_ PanicCheck) Check(stdout, stderr string, _ error) error { } return nil } + +// CustomCheck is meant to be a one-off checker with a user-provided function. +// +// Use this if none of the existing checkers match your use case, and it is not +// reusable/generic enough for use in other tests. +type CustomCheck struct { + name string + checkFunc func(stdout, stderr string, err error) error +} + +func (c CustomCheck) Check(stdout, stderr string, err error) error { + return c.checkFunc(stdout, stderr, err) +} + +func (c CustomCheck) Name() string { + return fmt.Sprintf("custom check - %s", c.name) +}