Function TestInsertIntoTTLTask

pkg/ttl/cache/task_test.go:94–122

(t *testing.T)

}

func TestInsertIntoTTLTask(t *testing.T) {
	store := testkit.CreateMockStore(t)
	tk := testkit.NewTestKit(t, store)
	tk.Session().GetSessionVars().TimeZone = time.Local

	ctx := kv.WithInternalSourceType(context.Background(), kv.InternalTxnTTL)
	tg := newTaskGetter(ctx, t, tk)

	rangeStart := []types.Datum{types.NewDatum(1)}
	rangeEnd := []types.Datum{types.NewDatum(2)}

	now := time.Now()
	now = now.Round(time.Second)

	sql, args, err := cache.InsertIntoTTLTask(tk.Session(), "test-job", 1, 1,
		rangeStart, rangeEnd, now, now)
	require.NoError(t, err)
	// tk.MustExec cannot handle the NULL parameter, use the `tk.Session().ExecuteInternal` instead here.
	_, err = tk.Session().ExecuteInternal(ctx, sql, args...)
	require.NoError(t, err)
	task := tg.mustGetTestTask()
	require.Equal(t, "test-job", task.JobID)
	require.Equal(t, int64(1), task.TableID)
	require.Equal(t, int64(1), task.ScanID)
	require.Equal(t, []types.Datum{types.NewDatum(1)}, task.ScanRangeStart)
	require.Equal(t, []types.Datum{types.NewDatum(2)}, task.ScanRangeEnd)
	require.Equal(t, now, task.ExpireTime)
	require.Equal(t, now, task.CreatedTime)
}