Add local context to chat operation failures

[onechat] / daemon_message_helper_test.go

diff --git a/daemon_message_helper_test.go b/daemon_message_helper_test.go
index 4ac80d310816f366bc388f4cffe15ed6fbfc21de..6b599ae17dfb7bb1a4e5a430594e48ed5e77050b 100644 (file)
--- a/daemon_message_helper_test.go
+++ b/daemon_message_helper_test.go
@@ -58,6 +58,25 @@ func newDaemonMessageTest(t *testing.T, dir string, server *ModelFake) *DaemonMe
        return test
 }
 
+func (r *DaemonMessageTest) Close(t *testing.T) {
+       t.Helper()
+
+       if r.cancel != nil {
+               r.cancel()
+
+               for _, done := range r.done {
+                       require.ErrorIs(t, <-done, context.Canceled)
+               }
+
+               r.cancel = nil
+               r.done = nil
+       }
+
+       require.NoError(t, r.agents.Close())
+
+       r.agents = nil
+}
+
 func (r *DaemonMessageTest) Start(t *testing.T, indices ...int) {
        t.Helper()
 
@@ -84,22 +103,3 @@ func (r *DaemonMessageTest) Queue(t *testing.T) DaemonMessageFile {
 
        return file
 }
-
-func (r *DaemonMessageTest) Close(t *testing.T) {
-       t.Helper()
-
-       if r.cancel != nil {
-               r.cancel()
-
-               for _, done := range r.done {
-                       require.ErrorIs(t, <-done, context.Canceled)
-               }
-
-               r.cancel = nil
-               r.done = nil
-       }
-
-       require.NoError(t, r.agents.Close())
-
-       r.agents = nil
-}