container/stub: override goexit methods

FailNow, Fatal, Fatalf, SkipNow, Skip and Skipf must be called from the goroutine created by the test.

Signed-off-by: Ophestra <cat@gensokyo.uk>

system/dispatcher_test.go

@@ -43,8 +43,8 @@ func checkOpBehaviour(t *testing.T, testCases []opBehaviourTestCase) {
 					ec = (*Criteria)(&tc.ec)
 				}
 
-				defer stub.HandleExit()
 				sys, s := InternalNew(t, stub.Expect{Calls: slices.Concat(tc.apply, []stub.Call{{Name: stub.CallSeparator}}, tc.revert)}, tc.uid)
+				defer s.HandleExit()
 				errApply := tc.op.apply(sys)
 				s.Expects(stub.CallSeparator)
 				if !reflect.DeepEqual(errApply, tc.wantErrApply) {
@@ -90,8 +90,8 @@ func checkOpsBuilder(t *testing.T, fname string, testCases []opsBuilderTestCase)
 			t.Run(tc.name, func(t *testing.T) {
 				t.Helper()
 
-				defer stub.HandleExit()
 				sys, s := InternalNew(t, tc.exp, tc.uid)
+				defer s.HandleExit()
 				tc.f(sys)
 				s.VisitIncomplete(func(s *stub.Stub[syscallDispatcher]) {
 					t.Helper()