package outcome

import (
	"context"
	"errors"
	"testing"

	domain "proxy-pool/internal/domain/outcome"
)

func TestQueueBatchesWithoutBlockingAndCountsOverflow(t *testing.T) {
	queue, err := NewQueue(QueueOptions{Capacity: 2, MaxBatch: 2})
	if err != nil {
		t.Fatalf("NewQueue() = %v", err)
	}
	queue.Record(domain.Event{ProxyID: "proxy-a"})
	queue.Record(domain.Event{ProxyID: "proxy-b"})
	queue.Record(domain.Event{ProxyID: "proxy-c"})
	if queue.Dropped() != 1 {
		t.Fatalf("Dropped() = %d, want 1", queue.Dropped())
	}
	batch, err := queue.Next(context.Background())
	if err != nil || len(batch) != 2 || batch[0].ProxyID != "proxy-a" || batch[1].ProxyID != "proxy-b" {
		t.Fatalf("Next() = %+v, %v", batch, err)
	}
	if batch, ok := queue.TryNext(); ok || batch != nil {
		t.Fatalf("TryNext() = %+v, %t; want nil, false", batch, ok)
	}
}

func TestQueueHonorsCanceledContext(t *testing.T) {
	queue, err := NewQueue(QueueOptions{Capacity: 1, MaxBatch: 1})
	if err != nil {
		t.Fatalf("NewQueue() = %v", err)
	}
	ctx, cancel := context.WithCancel(context.Background())
	cancel()
	if _, err := queue.Next(ctx); !errors.Is(err, context.Canceled) {
		t.Fatalf("Next(canceled) = %v, want context canceled", err)
	}
}