package spindle

import (
	"fmt"
	"testing"
	"time"

	"tangled.org/core/spindle/db"
)

func TestAPullEventBacklogDrainsInBatches(t *testing.T) {
	e := newPullEnv(t)
	d := e.n.s.db
	if err := d.AddPullTicket(testRepoDid, "bulk", db.PullTicket{Author: testBob, State: "open", Version: "v0"}); err != nil {
		t.Fatal(err)
	}
	const events = 3 * pullEventBatch
	for i := 1; i <= events; i++ {
		ev := db.PullEvent{RepoDid: testRepoDid, Ticket: "bulk", Event: "closed", Author: testBob, Sender: testBob, Pull: []byte("{}")}
		if err := d.AdvancePullTicket(testRepoDid, "bulk", fmt.Sprintf("v%d", i-1), fmt.Sprintf("v%d", i), "closed", &ev); err != nil {
			t.Fatal(err)
		}
	}

	e.n.s.retries.jobs.kick(testRepoDid)
	eventually(t, 10*time.Second, "the backlog to be worked through, a batch at a time", func() bool {
		due, err := d.DuePullEvents(testRepoDid, time.Now().Add(time.Hour), events+1)
		return err == nil && len(due) == 0
	})
}

func TestPullEventsStayDueWhileThePoolIsFull(t *testing.T) {
	e := newPullEnv(t)
	d := e.n.s.db
	e.n.s.cfg.Server.NotifySlots = 1
	repo, err := e.n.s.servedRepo(testRepoDid)
	if err != nil || repo == nil {
		t.Fatalf("setup: the repo isn't served: %v", err)
	}
	if !e.n.s.notifySlots.acquire("other-knot.test", e.n.s.notifyLimits()) {
		t.Fatal("setup: the pool was already full")
	}
	if err := d.AddPullTicket(testRepoDid, "held", db.PullTicket{Author: testBob, State: "open", Version: "v0"}); err != nil {
		t.Fatal(err)
	}
	ev := db.PullEvent{RepoDid: testRepoDid, Ticket: "held", Event: "closed", Author: testBob, Sender: testBob, Pull: []byte("{}")}
	if err := d.AdvancePullTicket(testRepoDid, "held", "v0", "v1", "closed", &ev); err != nil {
		t.Fatal(err)
	}

	e.n.s.runDuePullEvents(e.n.s.rootCtx, testRepoDid)
	due, err := d.DuePullEvents(testRepoDid, time.Now().Add(time.Hour), 10)
	if err != nil || len(due) != 1 {
		t.Fatalf("%d events due, %v, want the one that found no slot", len(due), err)
	}
	e.n.s.notifySlots.release("other-knot.test")
	e.n.s.runDuePullEvents(e.n.s.rootCtx, testRepoDid)
	if due, _ := d.DuePullEvents(testRepoDid, time.Now().Add(time.Hour), 10); len(due) != 0 {
		t.Fatalf("%d events left after the slot freed", len(due))
	}
}