package notify

import (
	"testing"
	"time"
)

func TestPendingWaitPoolDeliverWithScope(t *testing.T) {
	pool := newPendingWaitPool()
	wait := pool.createAndStoreWithScope(TransferMsg{ID: 11, Key: "k", Type: MSG_SYNC_ASK}, "scope-a")
	if delivered := pool.deliverWithScopes(11, []string{"scope-b"}, Message{}); delivered {
		t.Fatal("deliverWithScopes should reject mismatched scope")
	}
	msg := Message{TransferMsg: TransferMsg{ID: 11, Key: "k", Type: MSG_SYNC_REPLY}}
	if delivered := pool.deliverWithScopes(11, []string{"scope-a"}, msg); !delivered {
		t.Fatal("deliverWithScopes should accept matching scope")
	}
	select {
	case got := <-wait.Reply:
		if got.ID != msg.ID || got.Type != msg.Type {
			t.Fatalf("delivered message mismatch: got=%+v want=%+v", got.TransferMsg, msg.TransferMsg)
		}
	default:
		t.Fatal("wait reply should receive delivered message")
	}
}

func TestPendingWaitPoolCloseScopeClosesReplies(t *testing.T) {
	pool := newPendingWaitPool()
	wait := pool.createAndStoreWithScope(TransferMsg{ID: 12, Key: "k", Type: MSG_SYNC_ASK}, "scope-close")
	pool.closeScope("scope-close")
	select {
	case _, ok := <-wait.Reply:
		if ok {
			t.Fatal("wait reply channel should be closed")
		}
	case <-time.After(time.Second):
		t.Fatal("timed out waiting for closed wait reply")
	}
}

func TestSignalAckPoolDeliverAny(t *testing.T) {
	pool := newSignalAckPool()
	wait := pool.prepare("scope-a", 21)
	if delivered := pool.deliverAny([]string{"scope-b"}, 21); delivered {
		t.Fatal("deliverAny should reject mismatched scope")
	}
	go func() {
		time.Sleep(10 * time.Millisecond)
		_ = pool.deliverAny([]string{"scope-a"}, 21)
	}()
	if err := pool.waitPrepared(wait, time.Second); err != nil {
		t.Fatalf("waitPrepared failed: %v", err)
	}
}

func TestSignalAckPoolCloseScopeCancelsWait(t *testing.T) {
	pool := newSignalAckPool()
	wait := pool.prepare("scope-close", 22)
	go func() {
		time.Sleep(10 * time.Millisecond)
		pool.closeScope("scope-close")
	}()
	if err := pool.waitPrepared(wait, time.Second); err != errSignalAckCanceled {
		t.Fatalf("waitPrepared error = %v, want %v", err, errSignalAckCanceled)
	}
}