github.com/RevenueMonster/sqlike@v1.0.6/sqlike/paginate_test.go

package sqlike

import (
	"context"
	"testing"

	"github.com/RevenueMonster/sqlike/sql/expr"
	"github.com/RevenueMonster/sqlike/sqlike/actions"
	"github.com/RevenueMonster/sqlike/sqlike/primitive"
	"github.com/stretchr/testify/require"
)

func TestPaginate(t *testing.T) {
	var (
		err error
		pg  *Paginator
		ctx = context.Background()
	)

	tb := Table{pk: "ID"}

	t.Run("Ascending", func(ti *testing.T) {
		pg, err = tb.Paginate(
			ctx,
			actions.Paginate().
				Where(
					expr.Equal("A", "testing"),
					expr.Between("B", 1, 100),
					expr.IsNull("C"),
				).
				OrderBy(
					expr.Asc("A"),
					expr.Asc("B"),
				),
		)

		require.NoError(ti, err)
		require.ElementsMatch(ti, []primitive.Sort{
			expr.Asc("A"),
			expr.Asc("B"),
			expr.Asc("ID"),
		}, pg.action.Sorts)
	})

	t.Run("Descending", func(ti *testing.T) {
		pg, err = tb.Paginate(
			ctx,
			actions.Paginate().
				OrderBy(
					expr.Desc("A"),
					expr.Desc("B"),
				),
		)

		require.NoError(ti, err)
		require.ElementsMatch(ti, []primitive.Sort{
			expr.Desc("A"),
			expr.Desc("B"),
			expr.Desc("ID"),
		}, pg.action.Sorts)
	})

	t.Run("Ascending & Descending", func(ti *testing.T) {
		pg, err = tb.Paginate(
			ctx,
			actions.Paginate().
				OrderBy(
					expr.Asc("A"),
					expr.Desc("B"),
					expr.Asc("C"),
				),
		)

		require.NoError(ti, err)
		require.ElementsMatch(ti, []primitive.Sort{
			expr.Asc("A"),
			expr.Desc("B"),
			expr.Asc("C"),
			expr.Asc("ID"),
		}, pg.action.Sorts)
	})

}