package sink

import (
	"context"
	"testing"
	"time"

	"github.com/topfans/backend/services/statisticService/model"
)

func TestChannelEventSink_Submit(t *testing.T) {
	ch := make(chan *model.Event, 10)
	s := NewChannelEventSink(ch)
	e := &model.Event{EventID: "test-1"}
	if err := s.Submit(context.Background(), e); err != nil {
		t.Fatalf("Submit failed: %v", err)
	}
	select {
	case got := <-ch:
		if got.EventID != "test-1" {
			t.Fatal("event mismatch")
		}
	case <-time.After(100 * time.Millisecond):
		t.Fatal("no event received")
	}
}

func TestChannelEventSink_SubmitBatch(t *testing.T) {
	ch := make(chan *model.Event, 10)
	s := NewChannelEventSink(ch)
	events := []*model.Event{{EventID: "a"}, {EventID: "b"}}
	if err := s.SubmitBatch(context.Background(), events); err != nil {
		t.Fatalf("SubmitBatch failed: %v", err)
	}
	if len(ch) != 2 {
		t.Fatalf("expected 2 events, got %d", len(ch))
	}
}

func TestChannelEventSink_ChannelFull(t *testing.T) {
	ch := make(chan *model.Event, 1)
	s := NewChannelEventSink(ch)
	// 第一个事件占用 channel
	if err := s.Submit(context.Background(), &model.Event{EventID: "first"}); err != nil {
		t.Fatalf("first submit failed: %v", err)
	}
	// 第二个应失败（channel 满）
	if err := s.Submit(context.Background(), &model.Event{EventID: "second"}); err != ErrChannelFull {
		t.Fatalf("expected ErrChannelFull, got %v", err)
	}
}

func TestChannelEventSink_NilEvent(t *testing.T) {
	ch := make(chan *model.Event, 10)
	s := NewChannelEventSink(ch)
	if err := s.Submit(context.Background(), nil); err == nil {
		t.Fatal("expected error for nil event")
	}
}