package store

import (
	"errors"
	"strings"
	"testing"
)

// A runner that dies between claiming a build and reporting it leaves the row
// claimed. The next claim resolves it rather than leaving the build running and
// the commit pending forever.
func TestReapStaleBuilds(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, err := s.CreateUser("cmc", true)
	if err != nil {
		t.Fatal(err)
	}
	if _, err := s.CreateRepo("user", uid, "orgo", "public"); err != nil {
		t.Fatal(err)
	}

	stuck, err := s.CreateBuild(1, "test", "abc123", "main", `["true"]`, "", "", true)
	if err != nil {
		t.Fatal(err)
	}
	fresh, err := s.CreateBuild(1, "pages", "abc123", "main", `["true"]`, "", "", true)
	if err != nil {
		t.Fatal(err)
	}

	// Claim both, then age only the first past the deadline.
	for range 2 {
		if _, ok, err := s.ClaimBuild(nil, false); err != nil || !ok {
			t.Fatalf("claim: %v ok=%v", err, ok)
		}
	}
	if _, err := s.DB.Exec(
		`UPDATE builds SET started_at = '2020-01-01T00:00:00Z' WHERE number = ?`, stuck); err != nil {
		t.Fatal(err)
	}

	reaped, err := s.ReapStaleBuilds()
	if err != nil {
		t.Fatal(err)
	}
	if len(reaped) != 1 || reaped[0].Number != stuck {
		t.Fatalf("reaped %+v, want only build %d", reaped, stuck)
	}

	b, err := s.BuildByNumber(1, stuck)
	if err != nil {
		t.Fatal(err)
	}
	if b.Status != "failure" || b.FinishedAt == "" {
		t.Fatalf("stale build is %s finished %q, want failure with a timestamp", b.Status, b.FinishedAt)
	}
	log, err := s.BuildLog(b.ID)
	if err != nil {
		t.Fatal(err)
	}
	if !strings.Contains(string(log), "abandoned") {
		t.Fatalf("log does not say why it failed: %q", log)
	}

	// A build still inside the deadline is left alone.
	if b, err := s.BuildByNumber(1, fresh); err != nil || b.Status != "running" {
		t.Fatalf("fresh build is %v (%v), want running", b.Status, err)
	}
}

// Checks on a merge request report how long their build ran, which means
// pairing ci/<job> statuses with builds on the same commit.
func TestBuildsForCommitTiming(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, err := s.CreateUser("cmc", true)
	if err != nil {
		t.Fatal(err)
	}
	repoID, err := s.CreateRepo("user", uid, "lib", "public")
	if err != nil {
		t.Fatal(err)
	}
	// Two runs of the same job on one commit: the retry is what counts.
	for range 2 {
		if _, err := s.CreateBuild(repoID, "test", "abc123", "main", `["true"]`, "", "", true); err != nil {
			t.Fatal(err)
		}
	}
	if _, err := s.CreateBuild(repoID, "lint", "def456", "main", `["true"]`, "", "", true); err != nil {
		t.Fatal(err)
	}
	if _, err := s.DB.Exec(`UPDATE builds SET started_at = '2026-08-28T04:42:54Z',
		finished_at = '2026-08-28T04:44:06Z', status = 'success' WHERE number = 2`); err != nil {
		t.Fatal(err)
	}

	byJob, err := s.BuildsForCommit(repoID, "abc123")
	if err != nil {
		t.Fatal(err)
	}
	if len(byJob) != 1 {
		t.Fatalf("builds for commit: %+v", byJob)
	}
	b := byJob["test"]
	if b.Number != 2 {
		t.Fatalf("older run won: %d", b.Number)
	}
	if got := b.Elapsed().String(); got != "1m12s" {
		t.Fatalf("elapsed: %s", got)
	}
	// A build that never finished has no duration to report.
	if d := byJob["lint"].Elapsed(); d != 0 {
		t.Fatalf("unfinished build reported %s", d)
	}
}

// A runner that names repositories claims only their builds, so a runner on a
// machine that should not execute every repository's steps does not pick one
// up by being first to ask.
func TestClaimBuildScopedToRepos(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, err := s.CreateUser("cmc", true)
	if err != nil {
		t.Fatal(err)
	}
	mine, err := s.CreateRepo("user", uid, "site", "public")
	if err != nil {
		t.Fatal(err)
	}
	theirs, err := s.CreateRepo("user", uid, "stranger", "public")
	if err != nil {
		t.Fatal(err)
	}
	// Queued first, so an unscoped claim would take it.
	if _, err := s.CreateBuild(theirs, "evil", "abc123", "main", `["true"]`, "", "", true); err != nil {
		t.Fatal(err)
	}
	wanted, err := s.CreateBuild(mine, "deploy", "def456", "main", `["true"]`, "", "", true)
	if err != nil {
		t.Fatal(err)
	}

	b, ok, err := s.ClaimBuild([]int64{mine}, false)
	if err != nil || !ok {
		t.Fatalf("claim: %v ok=%v", err, ok)
	}
	if b.RepoID != mine || b.Number != wanted {
		t.Fatalf("claimed repo %d build %d, want repo %d build %d",
			b.RepoID, b.Number, mine, wanted)
	}

	// Nothing left for that scope, even though another repo's build is pending.
	if _, ok, err := s.ClaimBuild([]int64{mine}, false); err != nil || ok {
		t.Fatalf("second scoped claim: err=%v ok=%v, want no build", err, ok)
	}
	// An unscoped runner still takes it.
	if b, ok, err := s.ClaimBuild(nil, false); err != nil || !ok || b.RepoID != theirs {
		t.Fatalf("unscoped claim: err=%v ok=%v repo=%d", err, ok, b.RepoID)
	}
}

// A log that stops at the cap reads exactly like a build that died mid-step,
// which is what sent people hunting for a test failure that was never there.
// It says so instead, once.
func TestBuildLogSaysWhenItTruncates(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, err := s.CreateUser("cmc", true)
	if err != nil {
		t.Fatal(err)
	}
	if _, err := s.CreateRepo("user", uid, "orgo", "public"); err != nil {
		t.Fatal(err)
	}
	id, err := s.CreateBuild(1, "test", "abc123", "main", `["true"]`, "", "", true)
	if err != nil {
		t.Fatal(err)
	}

	chunk := make([]byte, 256<<10)
	for i := range chunk {
		chunk[i] = 'x'
	}
	// Well past the cap, so plenty of appends land after it.
	for written := 0; written < MaxBuildLog+(4*len(chunk)); written += len(chunk) {
		if err := s.AppendBuildLog(id, chunk); err != nil {
			t.Fatal(err)
		}
	}

	log, err := s.BuildLog(id)
	if err != nil {
		t.Fatal(err)
	}
	if n := strings.Count(string(log), "log truncated"); n != 1 {
		t.Errorf("truncation notice appears %d times, want exactly 1", n)
	}
	if !strings.HasSuffix(string(log), string(truncNotice)) {
		t.Error("notice is not at the end of the log")
	}
	if len(log) > MaxBuildLog+len(truncNotice)+len(chunk) {
		t.Errorf("log grew to %d, past the cap plus one chunk", len(log))
	}
}

// A success is found by tree across commits; an empty tree never matches,
// so builds queued without one (scheduled, tag) are never reused (#177).
func TestSuccessBuildForTree(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, err := s.CreateUser("cmc", true)
	if err != nil {
		t.Fatal(err)
	}
	repoID, err := s.CreateRepo("user", uid, "app", "public")
	if err != nil {
		t.Fatal(err)
	}
	if _, err := s.CreateBuild(repoID, "unit", "aaa", "main", `["true"]`, "", "tree1", true); err != nil {
		t.Fatal(err)
	}
	b, _ := s.BuildsForCommit(repoID, "aaa")
	if _, ok, err := s.ClaimBuild([]int64{repoID}, false); err != nil || !ok {
		t.Fatalf("claim: ok=%v err=%v", ok, err)
	}
	if err := s.FinishBuild(b["unit"].ID, "success"); err != nil {
		t.Fatal(err)
	}
	if prev, ok, _ := s.SuccessBuildForTree(repoID, "tree1", "unit", ""); !ok || prev.SHA != "aaa" {
		t.Fatalf("success not found by tree: ok=%v prev=%+v", ok, prev)
	}
	if _, ok, _ := s.SuccessBuildForTree(repoID, "tree1", "other", ""); ok {
		t.Error("matched a different job")
	}
	if _, ok, _ := s.SuccessBuildForTree(repoID, "", "unit", ""); ok {
		t.Error("an empty tree matched")
	}
}

// A result stands for another commit only when it came from a trusted
// build on the same image: a fork's green build, or one on an image the
// job has since left, proves nothing about the repository's own (#258).
func TestSuccessReuseNeedsTrustAndImage(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, _ := s.CreateUser("cmc", true)
	repoID, _ := s.CreateRepo("user", uid, "app", "public")
	for _, b := range []struct {
		sha, image string
		trusted    bool
	}{
		{"aaa", "", false},
		{"bbb", "localhost/old:1", true},
	} {
		if _, err := s.CreateBuild(repoID, "unit", b.sha, "main", `["true"]`, b.image, "tree1", b.trusted); err != nil {
			t.Fatal(err)
		}
		claimed, ok, err := s.ClaimBuild([]int64{repoID}, true)
		if err != nil || !ok {
			t.Fatalf("claim: ok=%v err=%v", ok, err)
		}
		if err := s.FinishBuild(claimed.ID, "success"); err != nil {
			t.Fatal(err)
		}
	}
	if prev, ok, _ := s.SuccessBuildForTree(repoID, "tree1", "unit", ""); ok {
		t.Fatalf("reused build %d: untrusted, or on another image", prev.Number)
	}
	if prev, ok, _ := s.SuccessBuildForTree(repoID, "tree1", "unit", "localhost/old:1"); !ok || prev.SHA != "bbb" {
		t.Fatalf("trusted build on the same image not found: ok=%v prev=%+v", ok, prev)
	}
	if _, ok, _ := s.SuccessBuildFor(repoID, "aaa", "unit"); ok {
		t.Error("an untrusted success stood for its commit")
	}
}

// A running build whose log stream ended is reaped after StaleLogGrace,
// well before the deadline; one whose stream is still open is not (#179).
func TestReapStaleBuildsAfterLogClosed(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, _ := s.CreateUser("cmc", true)
	repoID, _ := s.CreateRepo("user", uid, "app", "public")
	for _, job := range []string{"gone", "alive"} {
		if _, err := s.CreateBuild(repoID, job, "abc", "main", `["true"]`, "", "", true); err != nil {
			t.Fatal(err)
		}
		if _, ok, err := s.ClaimBuild([]int64{repoID}, false); err != nil || !ok {
			t.Fatalf("claim %s: %v", job, err)
		}
	}
	builds, _ := s.BuildsForCommit(repoID, "abc")
	gone := builds["gone"].ID
	if err := s.MarkBuildLogClosed(gone); err != nil {
		t.Fatal(err)
	}
	// Just closed: within the grace period, nothing is reaped.
	if stale, _ := s.ReapStaleBuilds(); len(stale) != 0 {
		t.Fatalf("reaped inside the grace period: %+v", stale)
	}
	// Backdate the close past the grace period.
	if _, err := s.DB.Exec("UPDATE builds SET log_closed_at = '2020-01-01T00:00:00Z' WHERE id = ?", gone); err != nil {
		t.Fatal(err)
	}
	stale, err := s.ReapStaleBuilds()
	if err != nil {
		t.Fatal(err)
	}
	if len(stale) != 1 || stale[0].ID != gone {
		t.Fatalf("reaped %+v, want only the build whose log closed", stale)
	}
	if b, _ := s.BuildByID(gone); b.Status != "failure" {
		t.Errorf("reaped build is %s, want failure", b.Status)
	}
	if b, _ := s.BuildByID(builds["alive"].ID); b.Status != "running" {
		t.Errorf("build with an open stream is %s, want running", b.Status)
	}
	// Marking is a no-op on a build that is no longer running.
	if err := s.MarkBuildLogClosed(gone); err != nil {
		t.Fatal(err)
	}
}

// AVG is a float in SQLite; the stats scan it as whole seconds.
func TestQueueStatsFractionalAverage(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, err := s.CreateUser("cmc", true)
	if err != nil {
		t.Fatal(err)
	}
	if _, err := s.CreateRepo("user", uid, "orgo", "public"); err != nil {
		t.Fatal(err)
	}
	for range 2 {
		if _, err := s.CreateBuild(1, "test", "abc123", "main", `["true"]`, "", "", true); err != nil {
			t.Fatal(err)
		}
		if _, ok, err := s.ClaimBuild(nil, false); err != nil || !ok {
			t.Fatalf("claim: %v ok=%v", err, ok)
		}
	}
	// Waits of 1 s and 2 s: an average of 1.5.
	if _, err := s.DB.Exec(`UPDATE builds SET created_at = strftime('%Y-%m-%dT%H:%M:%SZ', started_at, '-' || number || ' seconds')`); err != nil {
		t.Fatal(err)
	}
	q, err := s.QueueStats()
	if err != nil {
		t.Fatal(err)
	}
	if q.Claimed24h != 2 || q.ClaimWaitAvgS != 1 || q.ClaimWaitMaxS != 2 || q.Pending != 0 || q.Reaped24h != 0 {
		t.Fatalf("stats: %+v", q)
	}
}

// ListBuilds narrows on ref, status and job independently, and combines
// when more than one is given (#224).
func TestListBuildsFilters(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, err := s.CreateUser("cmc", true)
	if err != nil {
		t.Fatal(err)
	}
	repoID, err := s.CreateRepo("user", uid, "app", "public")
	if err != nil {
		t.Fatal(err)
	}
	if _, err := s.CreateBuild(repoID, "unit", "aaa", "main", `["true"]`, "", "", true); err != nil {
		t.Fatal(err)
	}
	if _, err := s.CreateBuild(repoID, "lint", "bbb", "feature", `["true"]`, "", "", true); err != nil {
		t.Fatal(err)
	}
	if err := s.FinishBuild(mustClaim(t, s, repoID).ID, "failure"); err != nil {
		t.Fatal(err)
	}

	all, err := s.ListBuilds(repoID, BuildFilter{}, 10)
	if err != nil || len(all) != 2 {
		t.Fatalf("unfiltered: %+v %v", all, err)
	}
	if byRef, err := s.ListBuilds(repoID, BuildFilter{Ref: "main"}, 10); err != nil || len(byRef) != 1 || byRef[0].Ref != "main" {
		t.Fatalf("by ref: %+v %v", byRef, err)
	}
	if byJob, err := s.ListBuilds(repoID, BuildFilter{Job: "lint"}, 10); err != nil || len(byJob) != 1 || byJob[0].Job != "lint" {
		t.Fatalf("by job: %+v %v", byJob, err)
	}
	if byStatus, err := s.ListBuilds(repoID, BuildFilter{Status: "failure"}, 10); err != nil || len(byStatus) != 1 || byStatus[0].Status != "failure" {
		t.Fatalf("by status: %+v %v", byStatus, err)
	}
	if combined, err := s.ListBuilds(repoID, BuildFilter{Ref: "main", Status: "failure"}, 10); err != nil || len(combined) != 1 {
		t.Fatalf("combined filter: %+v %v", combined, err)
	}
	if none, err := s.ListBuilds(repoID, BuildFilter{Ref: "main", Status: "pending"}, 10); err != nil || len(none) != 0 {
		t.Fatalf("non-matching combination: %+v %v", none, err)
	}
}

// mustClaim claims the oldest pending build for repoID, failing the test
// if none is available.
func mustClaim(t *testing.T, s *Store, repoID int64) Build {
	t.Helper()
	b, ok, err := s.ClaimBuild([]int64{repoID}, false)
	if err != nil || !ok {
		t.Fatalf("claim: %v ok=%v", err, ok)
	}
	return b
}

// A merge request head from a fork is untrusted. A claim skips it unless
// the runner asked for untrusted builds, so a runner on someone's laptop
// never executes a stranger's branch by default.
func TestClaimBuildSkipsUntrustedUnlessAsked(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, err := s.CreateUser("cmc", true)
	if err != nil {
		t.Fatal(err)
	}
	repo, err := s.CreateRepo("user", uid, "app", "public")
	if err != nil {
		t.Fatal(err)
	}
	// Queued first, so an unfiltered claim would take it.
	forkBuild, err := s.CreateBuild(repo, "unit", "abc123", "refs/merge-requests/1/head", `["true"]`, "", "", false)
	if err != nil {
		t.Fatal(err)
	}
	own, err := s.CreateBuild(repo, "unit", "def456", "main", `["true"]`, "", "", true)
	if err != nil {
		t.Fatal(err)
	}
	b, ok, err := s.ClaimBuild(nil, false)
	if err != nil || !ok || b.Number != own {
		t.Fatalf("trusted-only claim: err=%v ok=%v number=%d, want %d", err, ok, b.Number, own)
	}
	if _, ok, _ := s.ClaimBuild(nil, false); ok {
		t.Fatal("trusted-only claim took the fork build")
	}
	b, ok, err = s.ClaimBuild(nil, true)
	if err != nil || !ok || b.Number != forkBuild {
		t.Fatalf("untrusted claim: err=%v ok=%v number=%d, want %d", err, ok, b.Number, forkBuild)
	}
}

// A follower's channel closes on each kind of change to its build, and
// only its build.
func TestBuildLogWaitWakes(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, err := s.CreateUser("cmc", true)
	if err != nil {
		t.Fatal(err)
	}
	if _, err := s.CreateRepo("user", uid, "orgo", "public"); err != nil {
		t.Fatal(err)
	}
	newBuild := func() int64 {
		t.Helper()
		id, err := s.CreateBuild(1, "test", "abc123", "main", `["true"]`, "", "", true)
		if err != nil {
			t.Fatal(err)
		}
		return id
	}
	closed := func(ch <-chan struct{}) bool {
		select {
		case <-ch:
			return true
		default:
			return false
		}
	}

	a, b := newBuild(), newBuild()
	wa, wb := s.BuildLogWait(a), s.BuildLogWait(b)
	if err := s.AppendBuildLog(a, []byte("x")); err != nil {
		t.Fatal(err)
	}
	if !closed(wa) {
		t.Error("append did not wake its build")
	}
	if closed(wb) {
		t.Error("append woke another build")
	}

	mustClaim(t, s, 1) // claims a, the oldest
	wa = s.BuildLogWait(a)
	if err := s.FinishBuild(a, "success"); err != nil {
		t.Fatal(err)
	}
	if !closed(wa) {
		t.Error("finish did not wake")
	}

	if err := s.CancelBuild(b); err != nil {
		t.Fatal(err)
	}
	if !closed(wb) {
		t.Error("cancel did not wake")
	}
}

// Offsets are bytes, not characters: || stores the log as text, and a
// multibyte character must not shift where the next read starts.
func TestBuildLogFrom(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, err := s.CreateUser("cmc", true)
	if err != nil {
		t.Fatal(err)
	}
	if _, err := s.CreateRepo("user", uid, "orgo", "public"); err != nil {
		t.Fatal(err)
	}
	id, err := s.CreateBuild(1, "test", "abc123", "main", `["true"]`, "", "", true)
	if err != nil {
		t.Fatal(err)
	}
	first := "héllo — ok\n"
	for _, c := range []string{first, "wörld\n"} {
		if err := s.AppendBuildLog(id, []byte(c)); err != nil {
			t.Fatal(err)
		}
	}
	status, all, err := s.BuildLogFrom(id, 0)
	if err != nil || status != "pending" || string(all) != first+"wörld\n" {
		t.Fatalf("from 0: %q %q %v", status, all, err)
	}
	_, rest, err := s.BuildLogFrom(id, int64(len(first)))
	if err != nil || string(rest) != "wörld\n" {
		t.Fatalf("from %d: %q %v", len(first), rest, err)
	}
	_, none, err := s.BuildLogFrom(id, int64(len(all)))
	if err != nil || len(none) != 0 {
		t.Fatalf("from the end: %q %v", none, err)
	}
	if _, _, err := s.BuildLogFrom(9999, 0); err != ErrNotFound {
		t.Fatalf("missing build: %v", err)
	}
}

// Where a failed build stopped is recorded while it runs, before the
// outcome, and a finished build is not rewritten (#266).
func TestSetBuildFailure(t *testing.T) {
	s := open(t)
	if err := s.MigrateUp(); err != nil {
		t.Fatal(err)
	}
	uid, _ := s.CreateUser("cmc", true)
	repoID, _ := s.CreateRepo("user", uid, "app", "public")
	if _, err := s.CreateBuild(repoID, "unit", "abc", "main", `["true","false"]`, "", "", true); err != nil {
		t.Fatal(err)
	}
	b, ok, err := s.ClaimBuild(nil, false)
	if err != nil || !ok {
		t.Fatalf("claim: ok=%v err=%v", ok, err)
	}
	if err := s.SetBuildFailure(b.ID, 2, "exit 1"); err != nil {
		t.Fatal(err)
	}
	if err := s.FinishBuild(b.ID, "failure"); err != nil {
		t.Fatal(err)
	}
	got, err := s.BuildByID(b.ID)
	if err != nil {
		t.Fatal(err)
	}
	if got.FailedStep != 2 || got.FailedReason != "exit 1" {
		t.Fatalf("failed step %d reason %q", got.FailedStep, got.FailedReason)
	}
	if err := s.SetBuildFailure(b.ID, 1, "late"); !errors.Is(err, ErrNotFound) {
		t.Fatalf("rewrote a finished build: %v", err)
	}
}