优化结构

internal/worker/task_scheduler_test.go

@@ -1,6 +1,9 @@
 package worker
 
-import "testing"
+import (
+	"context"
+	"testing"
+)
 
 func TestCloseSubscribersClosesEveryAlertStream(t *testing.T) {
 	scheduler := &TaskScheduler{subscribers: map[uint]map[chan []byte]struct{}{}}
@@ -48,3 +51,17 @@ func TestSubscribeCapsConnectionsPerUser(t *testing.T) {
 	}
 	scheduler.closeSubscribers()
 }
+
+func TestSubscribeAfterStopReturnsClosedStream(t *testing.T) {
+	scheduler := newTestTaskScheduler(&workerTaskRepo{})
+	if err := scheduler.Stop(context.Background()); err != nil {
+		t.Fatalf("Stop() error = %v", err)
+	}
+	ch := scheduler.Subscribe(9)
+	if _, open := <-ch; open {
+		t.Fatal("subscription remains open after scheduler shutdown")
+	}
+	if len(scheduler.subscribers) != 0 {
+		t.Fatalf("subscription was added after shutdown: %#v", scheduler.subscribers)
+	}
+}