Stop traceback correctly when rapid is vendored

flyingmutant · Sep 2, 2020 · 93c490e · 93c490e
1 parent f03305c
commit 93c490e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/engine.go b/engine.go
@@ -225,7 +225,7 @@ func panicToError(p interface{}, skip int) *testError {
 
 	b := &strings.Builder{}
 	f, more, skipSpecial := runtime.Frame{}, true, true
-	for more && f.Function != tracebackStop {
+	for more && !strings.HasSuffix(f.Function, tracebackStop) {
 		f, more = frames.Next()
 
 		if skipSpecial && (tracebackBlacklist[f.Function] || strings.HasPrefix(f.Function, runtimePrefix)) {