chore: refactor RoutingBehaviour

internal/coord/behaviour_test.go

@@ -2,11 +2,12 @@ package coord
 
 import (
 	"context"
+	"testing"
 )
 
 type RecordingSM[E any, S any] struct {
 	State    S
-	Received E
+	Received []E
 }
 
 func NewRecordingSM[E any, S any](response S) *RecordingSM[E, S] {
@@ -16,6 +17,27 @@ func NewRecordingSM[E any, S any](response S) *RecordingSM[E, S] {
 }
 
 func (r *RecordingSM[E, S]) Advance(ctx context.Context, e E) S {
-	r.Received = e
+	r.Received = append(r.Received, e)
 	return r.State
 }
+
+func (r *RecordingSM[E, S]) first() E {
+	if len(r.Received) == 0 {
+		var zero E
+		return zero
+	}
+	return r.Received[0]
+}
+
+func DrainBehaviour[I BehaviourEvent, O BehaviourEvent](t *testing.T, ctx context.Context, b Behaviour[I, O]) {
+	for {
+		select {
+		case <-b.Ready():
+			b.Perform(ctx)
+		case <-ctx.Done():
+			t.Fatal("context cancelled while draining behaviour")
+		default:
+			return
+		}
+	}
+}