jldb/mdb/db-testlist_test.go

package mdb

import (
	"fmt"
	"reflect"
	"testing"
)

func TestDBList(t *testing.T) {
	db := NewTestDBPrimary(t, t.TempDir())

	var (
		user1 = User{
			ID:    NewID(),
			Name:  "User1",
			Email: "user1@gmail.com",
		}

		user2 = User{
			ID:    NewID(),
			Name:  "User2",
			Email: "user2@gmail.com",
		}

		user3 = User{
			ID:    NewID(),
			Name:  "User3",
			Email: "user3@gmail.com",
		}
		user1Data = make([]UserDataItem, 10)
		user2Data = make([]UserDataItem, 4)
		user3Data = make([]UserDataItem, 8)
	)

	err := db.Update(func(tx *Snapshot) error {
		if err := db.Users.Insert(tx, &user1); err != nil {
			return err
		}

		if err := db.Users.Insert(tx, &user2); err != nil {
			return err
		}

		for i := range user1Data {
			user1Data[i] = UserDataItem{
				ID:     NewID(),
				UserID: user1.ID,
				Name:   fmt.Sprintf("Name1: %d", i),
				Data:   fmt.Sprintf("Data: %d", i),
			}

			if err := db.UserData.Insert(tx, &user1Data[i]); err != nil {
				return err
			}
		}

		for i := range user2Data {
			user2Data[i] = UserDataItem{
				ID:     NewID(),
				UserID: user2.ID,
				Name:   fmt.Sprintf("Name2: %d", i),
				Data:   fmt.Sprintf("Data: %d", i),
			}

			if err := db.UserData.Insert(tx, &user2Data[i]); err != nil {
				return err
			}
		}

		for i := range user3Data {
			user3Data[i] = UserDataItem{
				ID:     NewID(),
				UserID: user3.ID,
				Name:   fmt.Sprintf("Name3: %d", i),
				Data:   fmt.Sprintf("Data: %d", i),
			}

			if err := db.UserData.Insert(tx, &user3Data[i]); err != nil {
				return err
			}
		}

		return nil
	})

	if err != nil {
		t.Fatal(err)
	}

	type TestCase struct {
		Name     string
		Args     ListArgs[UserDataItem]
		Expected []UserDataItem
	}

	cases := []TestCase{
		{
			Name: "User1 all",
			Args: ListArgs[UserDataItem]{
				After: &UserDataItem{
					UserID: user1.ID,
				},
				While: func(item *UserDataItem) bool {
					return item.UserID == user1.ID
				},
			},
			Expected: user1Data,
		}, {
			Name: "User1 limited",
			Args: ListArgs[UserDataItem]{
				After: &UserDataItem{
					UserID: user1.ID,
				},
				While: func(item *UserDataItem) bool {
					return item.UserID == user1.ID
				},
				Limit: 4,
			},
			Expected: user1Data[:4],
		},
	}

	for _, tc := range cases {
		t.Run(tc.Name, func(t *testing.T) {
			tx := db.Snapshot()
			l := db.UserData.ByName.List(tx, tc.Args, nil)
			if len(l) != len(tc.Expected) {
				t.Fatal(tc.Name, l)
			}

			for i := range l {
				if !reflect.DeepEqual(*l[i], tc.Expected[i]) {
					t.Fatal(tc.Name, l)
				}
			}
		})
	}
}