X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/c781f79f3c8cb9a7577ee1574d196988da2719b3..0cb73c3dde36d0078f3c1aa4200f2edcb5ceed38:/daemon_message_helper_test.go

diff --git a/daemon_message_helper_test.go b/daemon_message_helper_test.go
index 4ac80d3..6b599ae 100644
--- 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
-}