import files, add README & LICENSE

2026-06-29 17:06:34 -07:00
commit aecd90eb84
17 changed files with 2552 additions and 0 deletions
@@ -0,0 +1,293 @@
+package cursor
+
+import (
+	"encoding/json"
+	"testing"
+	"time"
+
+	"github.com/go-jet/jet/v2/mysql"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestColumnKey(t *testing.T) {
+	assert := assert.New(t)
+
+	empty := ColumnKey{}
+	assert.True(empty.IsEmpty())
+	assert.Equal("user.id", NewColumnKey(User.ID).String())
+
+	columnKey := NewColumnKey(User.ID)
+	assert.Equal(User.TableName(), columnKey.Table)
+	assert.Equal(User.ID.Name(), columnKey.Column)
+	assert.False(columnKey.IsEmpty())
+
+	bytes, err := json.Marshal(columnKey)
+	assert.NoError(err)
+	assert.Equal(`{"table":"user","column":"id"}`, string(bytes))
+
+	var res ColumnKey
+	err = json.Unmarshal(bytes, &res)
+	assert.NoError(err)
+	assert.Equal(User.TableName(), res.Table)
+	assert.Equal(User.ID.Name(), res.Column)
+}
+
+func TestColumnRegistry(t *testing.T) {
+	assert := assert.New(t)
+
+	RegisterColumn(User.ID)
+
+	col, err := GetColumn(User.TableName(), User.ID.Name())
+	assert.NoError(err)
+	assert.Equal(User.ID, col)
+
+	_, err = GetColumn(User.TableName(), "nonexistent")
+	assert.Error(err)
+
+	col, err = GetColumnByKey(ColumnKey{
+		Table:  User.TableName(),
+		Column: User.ID.Name(),
+	})
+	assert.NoError(err)
+	assert.Equal(User.ID, col)
+
+	_, err = GetColumnByKey(ColumnKey{
+		Table:  User.TableName(),
+		Column: "nonexistent",
+	})
+	assert.Error(err)
+
+	_, err = GetColumn(User.TableName(), User.CreatedAt.Name())
+	assert.Error(err)
+}
+
+func TestRegisterColumnList(t *testing.T) {
+	assert := assert.New(t)
+
+	RegisterColumnList(Address.AllColumns)
+
+	col, err := GetColumn(Address.TableName(), Address.ID.Name())
+	assert.NoError(err)
+	assert.Equal(Address.ID, col)
+}
+
+func TestOrderDirection(t *testing.T) {
+	assert := assert.New(t)
+
+	bytes, err := json.Marshal(OrderAscending)
+	assert.NoError(err)
+	assert.Equal(`"ASC"`, string(bytes))
+
+	var res OrderDirection
+	err = json.Unmarshal(bytes, &res)
+	assert.NoError(err)
+	assert.Equal(OrderAscending, res)
+
+	bytes, err = json.Marshal(OrderDescending)
+	assert.NoError(err)
+	assert.Equal(`"DESC"`, string(bytes))
+
+	err = json.Unmarshal(bytes, &res)
+	assert.NoError(err)
+	assert.Equal(OrderDescending, res)
+
+	var invalid OrderDirection
+	err = json.Unmarshal([]byte(`"INVALID"`), &invalid)
+	assert.Error(err)
+
+	_, err = OrderDirection(99).MarshalJSON()
+	assert.Error(err)
+}
+
+func TestInt64Value(t *testing.T) {
+	assert := assert.New(t)
+
+	v := NewInt64Value(1, Scratch.ID)
+	assert.Equal(int64(1), v.Val)
+	assert.Equal(Scratch.ID.Name(), v.Key.Column)
+	assert.Equal(Scratch.TableName(), v.Key.Table)
+	assert.Equal(mysql.Int64(1), v.Expr())
+	assert.Panics(func() { v.Col() })
+	RegisterColumn(Scratch.ID)
+	assert.Equal(Scratch.ID, v.Col())
+	assert.False(v.IsEmpty())
+
+	bytes, err := json.Marshal(v)
+	assert.NoError(err)
+	assert.Equal(`{"key":{"table":"scratch","column":"id"},"val":1}`, string(bytes))
+
+	var res Int64Value
+	assert.True(res.IsEmpty())
+	err = json.Unmarshal(bytes, &res)
+	assert.NoError(err)
+	assert.Equal(int64(1), res.Val)
+	assert.Equal(Scratch.ID.Name(), res.Key.Column)
+	assert.Equal(Scratch.TableName(), res.Key.Table)
+	assert.Equal(mysql.Int64(1), res.Expr())
+	assert.Equal(Scratch.ID, v.Col())
+}
+
+func TestUint64Value(t *testing.T) {
+	assert := assert.New(t)
+
+	v := NewUint64Value(42, Spare.ID)
+	assert.Equal(uint64(42), v.Val)
+	assert.Equal(mysql.Uint64(42), v.Expr())
+	assert.Panics(func() { v.Col() })
+	RegisterColumn(Spare.ID)
+	assert.Equal(Spare.ID, v.Col())
+	assert.False(v.IsEmpty())
+
+	bytes, err := json.Marshal(v)
+	assert.NoError(err)
+
+	var res Uint64Value
+	assert.True(res.IsEmpty())
+	err = json.Unmarshal(bytes, &res)
+	assert.NoError(err)
+	assert.Equal(uint64(42), res.Val)
+}
+
+func TestStringValueIsEmpty(t *testing.T) {
+	assert := assert.New(t)
+	registerMeetingCursorColumns(t)
+
+	empty := NewStringValue("", Meeting.ID)
+	assert.True(empty.IsEmpty())
+
+	nonEmpty := NewStringValue("meeting-uuid-1", Meeting.ID)
+	assert.False(nonEmpty.IsEmpty())
+
+	defaultMeetingCursor := NewCursor(
+		NewStringValue("", Meeting.ID),
+		Meeting.StartTime,
+		OrderDescending,
+	)
+	assert.True(defaultMeetingCursor.IsEmpty())
+}
+
+func TestNewCursor(t *testing.T) {
+	assert := assert.New(t)
+
+	empty := Cursor[mysql.IntegerExpression, mysql.ColumnInteger]{}
+	assert.True(empty.IsEmpty())
+
+	v := NewInt64Value(1, User.ID)
+	cursor := NewCursor(v, User.ID, OrderAscending)
+
+	assert.Equal(v, cursor.Index)
+	assert.Equal(NewColumnKey(User.ID), cursor.OrderColumnKey)
+	assert.Equal(OrderAscending, cursor.OrderDir)
+
+	bytes, err := json.Marshal(cursor)
+	assert.NoError(err)
+	assert.Equal(`{"index":{"key":{"table":"user","column":"id"},"val":1},"order_col":{"table":"user","column":"id"},"order_dir":"ASC"}`, string(bytes))
+}
+
+func TestCopyWithVal(t *testing.T) {
+	assert := assert.New(t)
+	RegisterColumn(User.ID)
+
+	original := NewCursor(NewInt64Value(1, User.ID), User.ID, OrderDescending)
+	copied := original.CopyWithVal(NewInt64Value(99, User.ID))
+
+	assert.Equal(int64(99), copied.Index.(*Int64Value).Val)
+	assert.Equal(OrderDescending, copied.OrderDir)
+	assert.Equal(NewColumnKey(User.ID), copied.OrderColumnKey)
+}
+
+func TestNewCursorFromJSON(t *testing.T) {
+	assert := assert.New(t)
+
+	v := NewInt64Value(1, User.ID)
+	cursor := NewCursor(v, User.ID, OrderDescending)
+	bytes, err := json.Marshal(cursor)
+	assert.NoError(err)
+
+	zv := NewInt64Value(0, User.ID)
+	res, err := NewCursorFromJSON(zv, bytes)
+	assert.NoError(err)
+	assert.NotNil(res)
+	assert.Equal(int64(1), res.Index.(*Int64Value).Val)
+	assert.Equal(NewColumnKey(User.ID), res.OrderColumnKey)
+	assert.Equal(OrderDescending, res.OrderDir)
+}
+
+func TestNewCursorFromJSONComposite(t *testing.T) {
+	assert := assert.New(t)
+	registerMeetingCursorColumns(t)
+
+	when := time.Date(2026, 6, 1, 9, 0, 0, 0, time.UTC)
+	original := NewCursor(
+		NewStringValue("", Meeting.ID),
+		Meeting.StartTime,
+		OrderDescending,
+	).CopyWithVals(
+		NewStringValue("meeting-uuid-1", Meeting.ID),
+		NewTimestampValue(when, Meeting.StartTime),
+	)
+
+	bytes, err := json.Marshal(original)
+	require.NoError(t, err)
+
+	zv := NewStringValue("", Meeting.ID)
+	res, err := NewCursorFromJSON(zv, bytes)
+	require.NoError(t, err)
+	assert.True(res.IsComposite())
+	assert.Equal("meeting-uuid-1", res.Index.(*StringValue).Val)
+	assert.Equal(when, res.GenericOrderValue().(*TimestampValue).Val)
+}
+
+func TestDecodeAndOrder(t *testing.T) {
+	assert := assert.New(t)
+	RegisterColumn(User.ID)
+
+	encoded, err := NewCursor(
+		NewInt64Value(5, User.ID),
+		User.ID,
+		OrderDescending,
+	).Encode()
+	require.NoError(t, err)
+
+	c := NewCursor(NewInt64Value(0, User.ID), User.ID, OrderAscending)
+	require.NoError(t, c.DecodeAndOrder(encoded, OrderAscending))
+	assert.Equal(OrderAscending, c.OrderDir)
+	assert.Equal(int64(5), c.Index.(*Int64Value).Val)
+}
+
+func TestCursorDecodeNilReceiver(t *testing.T) {
+	var c *Cursor[mysql.IntegerExpression, mysql.ColumnInteger]
+	assert.Error(t, c.Decode("{}"))
+}
+
+func TestNewCursorFromAfterPtr(t *testing.T) {
+	assert := assert.New(t)
+	require := require.New(t)
+	RegisterColumn(User.ID)
+
+	newZero := func() *Cursor[mysql.IntegerExpression, mysql.ColumnInteger] {
+		return NewCursor(NewInt64Value(0, User.ID), User.ID, OrderDescending)
+	}
+
+	c, err := NewCursorFromAfterPtr(newZero, nil)
+	require.NoError(err)
+	assert.Nil(c)
+
+	empty := ""
+	c, err = NewCursorFromAfterPtr(newZero, &empty)
+	require.NoError(err)
+	assert.Nil(c)
+
+	encoded, err := NewCursor(
+		NewInt64Value(42, User.ID),
+		User.ID,
+		OrderDescending,
+	).Encode()
+	require.NoError(err)
+
+	c, err = NewCursorFromAfterPtr(newZero, &encoded)
+	require.NoError(err)
+	require.NotNil(c)
+	assert.Equal(int64(42), c.Index.(*Int64Value).Val)
+}