task_scheduler/go/db/memory/memory_test.go

package memory

import (
	"os"
	"testing"

	"go.skia.org/infra/go/deepequal"
	"go.skia.org/infra/go/testutils"
	"go.skia.org/infra/task_scheduler/go/db"
)

func TestMain(m *testing.M) {
	db.AssertDeepEqual = deepequal.AssertDeepEqual
	os.Exit(m.Run())
}

func TestInMemoryTaskDB(t *testing.T) {
	testutils.SmallTest(t)
	db.TestTaskDB(t, NewInMemoryTaskDB(nil))
}

func TestInMemoryTaskDBConcurrentUpdate(t *testing.T) {
	testutils.SmallTest(t)
	db.TestTaskDBConcurrentUpdate(t, NewInMemoryTaskDB(nil))
}

func TestInMemoryTaskDBUpdateTasksWithRetries(t *testing.T) {
	testutils.SmallTest(t)
	db.TestUpdateTasksWithRetries(t, NewInMemoryTaskDB(nil))
}

func TestInMemoryTaskDBGetTasksFromDateRangeByRepo(t *testing.T) {
	testutils.SmallTest(t)
	db.TestTaskDBGetTasksFromDateRangeByRepo(t, NewInMemoryTaskDB(nil))
}

func TestInMemoryTaskDBGetTasksFromWindow(t *testing.T) {
	testutils.LargeTest(t)
	db.TestTaskDBGetTasksFromWindow(t, NewInMemoryTaskDB(nil))
}

func TestInMemoryUpdateDBFromSwarmingTask(t *testing.T) {
	testutils.SmallTest(t)
	db.TestUpdateDBFromSwarmingTask(t, NewInMemoryTaskDB(nil))
}

func TestInMemoryUpdateDBFromSwarmingTaskTryjob(t *testing.T) {
	testutils.SmallTest(t)
	db.TestUpdateDBFromSwarmingTaskTryJob(t, NewInMemoryTaskDB(nil))
}

func TestInMemoryJobDB(t *testing.T) {
	testutils.SmallTest(t)
	db.TestJobDB(t, NewInMemoryJobDB(nil))
}

func TestInMemoryJobDBConcurrentUpdate(t *testing.T) {
	testutils.SmallTest(t)
	db.TestJobDBConcurrentUpdate(t, NewInMemoryJobDB(nil))
}