package tryjobs

import (
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"go.skia.org/infra/go/testutils"
	"go.skia.org/infra/task_scheduler/go/db"
	"go.skia.org/infra/task_scheduler/go/window"
)

func makeJob(created time.Time) *db.Job {
	return &db.Job{
		Created:             created,
		BuildbucketLeaseKey: 12345,
	}
}

func TestJobCache(t *testing.T) {
	testutils.SmallTest(t)
	d := db.NewInMemoryJobDB()

	// Pre-load a job into the DB.
	now := time.Now()
	j1 := makeJob(now.Add(-10 * time.Minute))
	assert.NoError(t, d.PutJob(j1))

	// Create the cache. Ensure that the existing job is present.
	w, err := window.New(time.Hour, 0, nil)
	assert.NoError(t, err)
	c, err := newJobCache(d, w)
	assert.NoError(t, err)
	jobs, err := c.GetActiveTryJobs()
	assert.NoError(t, err)
	testutils.AssertDeepEqual(t, []*db.Job{j1}, jobs)

	// Create another job. Ensure that it gets picked up.
	j2 := makeJob(now.Add(-5 * time.Minute))
	assert.NoError(t, d.PutJob(j2))
	assert.NoError(t, c.Update())
	jobs, err = c.GetActiveTryJobs()
	assert.NoError(t, err)
	testutils.AssertDeepEqual(t, []*db.Job{j1, j2}, jobs)

	// j1 is not active.
	j1.BuildbucketLeaseKey = 0
	assert.NoError(t, d.PutJob(j1))
	assert.NoError(t, c.Update())
	jobs, err = c.GetActiveTryJobs()
	assert.NoError(t, err)
	testutils.AssertDeepEqual(t, []*db.Job{j2}, jobs)

	// Expire j2.
	assert.NoError(t, w.UpdateWithTime(now.Add(time.Hour)))
	assert.NoError(t, c.Update())
	jobs, err = c.GetActiveTryJobs()
	assert.NoError(t, err)
	testutils.AssertDeepEqual(t, []*db.Job{}, jobs)
}