phase 10

internal/broker/broker_test.go

@@ -10,14 +10,14 @@ import (
 	"github.com/uopi/uopi/internal/datasource/stub"
 )
 
-func newBroker(t *testing.T) (*broker.Broker, context.CancelFunc) {
-	t.Helper()
+func newBroker(tb testing.TB) (*broker.Broker, context.CancelFunc) {
+	tb.Helper()
 	ctx, cancel := context.WithCancel(context.Background())
 	log := slog.Default()
 	b := broker.New(ctx, log)
 	ds := stub.New()
 	if err := ds.Connect(ctx); err != nil {
-		t.Fatal(err)
+		tb.Fatal(err)
 	}
 	b.Register(ds)
 	return b, cancel