Memoh/internal/channel/adapters/local/hub_test.go

package local

import (
	"context"
	"testing"
	"time"

	"github.com/memohai/memoh/internal/channel"
)

func TestRouteHubPublishEvent(t *testing.T) {
	t.Parallel()

	hub := NewRouteHub()
	_, stream, cancel := hub.Subscribe("bot-1")
	defer cancel()

	hub.PublishEvent("bot-1", channel.StreamEvent{
		Type:  channel.StreamEventDelta,
		Delta: "hello",
	})

	select {
	case item := <-stream:
		if item.Target != "bot-1" {
			t.Fatalf("unexpected target: %s", item.Target)
		}
		if item.Event.Type != channel.StreamEventDelta {
			t.Fatalf("unexpected event type: %s", item.Event.Type)
		}
	case <-time.After(time.Second):
		t.Fatal("expected event but timed out")
	}
}

func TestLocalOutboundStreamClose(t *testing.T) {
	t.Parallel()

	hub := NewRouteHub()
	stream := newLocalOutboundStream(hub, "bot-2")
	if err := stream.Close(context.Background()); err != nil {
		t.Fatalf("unexpected close error: %v", err)
	}
	if err := stream.Push(context.Background(), channel.PreparedStreamEvent{
		Type:  channel.StreamEventDelta,
		Delta: "should fail",
	}); err == nil {
		t.Fatal("expected push on closed stream to fail")
	}
}