diff options
Diffstat (limited to 'system')
| -rw-r--r-- | system/dispatcher_test.go | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/system/dispatcher_test.go b/system/dispatcher_test.go index 7fbb75c2..53d2b664 100644 --- a/system/dispatcher_test.go +++ b/system/dispatcher_test.go @@ -44,7 +44,7 @@ func checkOpBehaviour(t *testing.T, testCases []opBehaviourTestCase) { } sys, s := InternalNew(t, stub.Expect{Calls: slices.Concat(tc.apply, []stub.Call{{Name: stub.CallSeparator}}, tc.revert)}, tc.uid) - defer s.HandleExit() + defer stub.HandleExit(t) errApply := tc.op.apply(sys) s.Expects(stub.CallSeparator) if !reflect.DeepEqual(errApply, tc.wantErrApply) { @@ -91,7 +91,7 @@ func checkOpsBuilder(t *testing.T, fname string, testCases []opsBuilderTestCase) t.Helper() sys, s := InternalNew(t, tc.exp, tc.uid) - defer s.HandleExit() + defer stub.HandleExit(t) tc.f(sys) s.VisitIncomplete(func(s *stub.Stub[syscallDispatcher]) { t.Helper() |