connectivity: Add GetTestOrDie()

Add a convenient function GetTestOrDie(). There isn't much you can do if GetTest() fails anyways. Signed-off-by: Michi Mutsuzaki <[email protected]>
cilium · Nov 8, 2023 · 302b672 · 302b672
1 parent 3490b39
commit 302b672
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/connectivity/check/context.go b/connectivity/check/context.go
@@ -293,6 +293,16 @@ func (ct *ConnectivityTest) GetTest(name string) (*Test, error) {
 	panic("missing test descriptor for a registered name")
 }
 
+// GetTestOrDie returns the test scope for test named "name" if found,
+// or panics otherwise.
+func (ct *ConnectivityTest) GetTestOrDie(name string) *Test {
+	test, err := ct.GetTest(name)
+	if err != nil {
+		panic(err)
+	}
+	return test
+}
+
 // SetupAndValidate sets up and validates the connectivity test infrastructure
 // such as the client pods and validates the deployment of them along with
 // Cilium. This must be run before Run() is called.