705 lines
16 KiB
Go
705 lines
16 KiB
Go
package storm
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/coreos/bbolt"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFind(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
for i := 0; i < 100; i++ {
|
|
w := User{Name: "John", ID: i + 1, Slug: fmt.Sprintf("John%d", i+1)}
|
|
|
|
if i%2 == 0 {
|
|
w.Group = "staff"
|
|
} else {
|
|
w.Group = "normal"
|
|
}
|
|
|
|
err := db.Save(&w)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
err := db.Find("Name", "John", &User{})
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrSlicePtrNeeded, err)
|
|
|
|
err = db.Find("Name", "John", &[]struct {
|
|
Name string
|
|
ID int
|
|
}{})
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrNoName, err)
|
|
|
|
notTheRightUsers := []UniqueNameUser{}
|
|
|
|
err = db.Find("Name", "John", ¬TheRightUsers)
|
|
require.Error(t, err)
|
|
require.EqualError(t, err, "not found")
|
|
|
|
users := []User{}
|
|
|
|
err = db.Find("unexportedField", "John", &users)
|
|
require.Error(t, err)
|
|
require.EqualError(t, err, "field unexportedField not found")
|
|
|
|
err = db.Find("DateOfBirth", "John", &users)
|
|
require.Error(t, err)
|
|
require.True(t, ErrNotFound == err)
|
|
|
|
err = db.Find("Group", "staff", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 50)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 99, users[49].ID)
|
|
|
|
err = db.Find("Group", "staff", &users, Reverse())
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 50)
|
|
require.Equal(t, 99, users[0].ID)
|
|
require.Equal(t, 1, users[49].ID)
|
|
|
|
err = db.Find("Group", "admin", &users)
|
|
require.Error(t, err)
|
|
require.True(t, ErrNotFound == err)
|
|
|
|
err = db.Find("Name", "John", users)
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrSlicePtrNeeded, err)
|
|
|
|
err = db.Find("Name", "John", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 100)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 100, users[99].ID)
|
|
|
|
err = db.Find("Name", "John", &users, Reverse())
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 100)
|
|
require.Equal(t, 100, users[0].ID)
|
|
require.Equal(t, 1, users[99].ID)
|
|
|
|
users = []User{}
|
|
err = db.Find("Slug", "John10", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 1)
|
|
require.Equal(t, 10, users[0].ID)
|
|
|
|
users = []User{}
|
|
err = db.Find("Name", nil, &users)
|
|
require.Error(t, err)
|
|
require.True(t, ErrNotFound == err)
|
|
|
|
err = db.Find("Name", "John", &users, Limit(10), Skip(20))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 21, users[0].ID)
|
|
require.Equal(t, 30, users[9].ID)
|
|
|
|
err = db.Find("Age", 10, &users)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestFindNil(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
type User struct {
|
|
ID int `storm:"increment"`
|
|
CreatedAt *time.Time `storm:"index"`
|
|
DeletedAt *time.Time `storm:"unique"`
|
|
}
|
|
|
|
t1 := time.Now()
|
|
for i := 0; i < 10; i++ {
|
|
now := time.Now()
|
|
var u User
|
|
|
|
if i%2 == 0 {
|
|
u.CreatedAt = &t1
|
|
u.DeletedAt = &now
|
|
}
|
|
|
|
err := db.Save(&u)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
var users []User
|
|
err := db.Find("CreatedAt", nil, &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 5)
|
|
|
|
users = nil
|
|
err = db.Find("CreatedAt", t1, &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 5)
|
|
|
|
users = nil
|
|
err = db.Find("DeletedAt", nil, &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 5)
|
|
}
|
|
|
|
func TestFindIntIndex(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
type Score struct {
|
|
ID int `storm:"increment"`
|
|
Score uint64 `storm:"index"`
|
|
}
|
|
|
|
for i := 0; i < 10; i++ {
|
|
w := Score{Score: uint64(i % 3)}
|
|
err := db.Save(&w)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
var scores []Score
|
|
err := db.Find("Score", 2, &scores)
|
|
require.NoError(t, err)
|
|
require.Len(t, scores, 3)
|
|
require.Equal(t, []Score{
|
|
{ID: 3, Score: 2},
|
|
{ID: 6, Score: 2},
|
|
{ID: 9, Score: 2},
|
|
}, scores)
|
|
}
|
|
|
|
func TestAllByIndex(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
for i := 0; i < 100; i++ {
|
|
w := User{Name: "John", ID: i + 1, Slug: fmt.Sprintf("John%d", i+1), DateOfBirth: time.Now().Add(-time.Duration(i*10) * time.Minute)}
|
|
err := db.Save(&w)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
err := db.AllByIndex("", nil)
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrSlicePtrNeeded, err)
|
|
|
|
var users []User
|
|
|
|
err = db.AllByIndex("Unknown field", &users)
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrNotFound, err)
|
|
|
|
err = db.AllByIndex("DateOfBirth", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 100)
|
|
require.Equal(t, 100, users[0].ID)
|
|
require.Equal(t, 1, users[99].ID)
|
|
|
|
err = db.AllByIndex("Name", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 100)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 100, users[99].ID)
|
|
|
|
y := UniqueNameUser{Name: "Jake", ID: 200}
|
|
err = db.Save(&y)
|
|
require.NoError(t, err)
|
|
|
|
var y2 []UniqueNameUser
|
|
err = db.AllByIndex("ID", &y2)
|
|
require.NoError(t, err)
|
|
require.Len(t, y2, 1)
|
|
|
|
n := NestedID{}
|
|
n.ID = "100"
|
|
n.Name = "John"
|
|
|
|
err = db.Save(&n)
|
|
require.NoError(t, err)
|
|
|
|
var n2 []NestedID
|
|
err = db.AllByIndex("ID", &n2)
|
|
require.NoError(t, err)
|
|
require.Len(t, n2, 1)
|
|
|
|
err = db.AllByIndex("Name", &users, Limit(10))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 10, users[9].ID)
|
|
|
|
err = db.AllByIndex("Name", &users, Limit(200))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 100)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 100, users[99].ID)
|
|
|
|
err = db.AllByIndex("Name", &users, Limit(-10))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 100)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 100, users[99].ID)
|
|
|
|
err = db.AllByIndex("Name", &users, Skip(200))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 0)
|
|
|
|
err = db.AllByIndex("Name", &users, Skip(-10))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 100)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 100, users[99].ID)
|
|
|
|
err = db.AllByIndex("ID", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 100)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 100, users[99].ID)
|
|
|
|
err = db.AllByIndex("ID", &users, Limit(10))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 10, users[9].ID)
|
|
|
|
err = db.AllByIndex("ID", &users, Skip(10))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 90)
|
|
require.Equal(t, 11, users[0].ID)
|
|
require.Equal(t, 100, users[89].ID)
|
|
|
|
err = db.AllByIndex("Name", &users, Limit(10), Skip(10))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 11, users[0].ID)
|
|
require.Equal(t, 20, users[9].ID)
|
|
|
|
err = db.AllByIndex("Name", &users, Limit(10), Skip(10), Reverse())
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 90, users[0].ID)
|
|
require.Equal(t, 81, users[9].ID)
|
|
|
|
err = db.AllByIndex("Age", &users, Limit(10))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
}
|
|
|
|
func TestAll(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
for i := 0; i < 100; i++ {
|
|
w := User{Name: "John", ID: i + 1, Slug: fmt.Sprintf("John%d", i+1), DateOfBirth: time.Now().Add(-time.Duration(i*10) * time.Minute)}
|
|
err := db.Save(&w)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
var users []User
|
|
|
|
err := db.All(&users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 100)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 100, users[99].ID)
|
|
|
|
err = db.All(&users, Reverse())
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 100)
|
|
require.Equal(t, 100, users[0].ID)
|
|
require.Equal(t, 1, users[99].ID)
|
|
|
|
var users2 []*User
|
|
|
|
err = db.All(&users2)
|
|
require.NoError(t, err)
|
|
require.Len(t, users2, 100)
|
|
require.Equal(t, 1, users2[0].ID)
|
|
require.Equal(t, 100, users2[99].ID)
|
|
|
|
err = db.Save(&NestedID{
|
|
ToEmbed: ToEmbed{ID: "id1"},
|
|
Name: "John",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
err = db.Save(&NestedID{
|
|
ToEmbed: ToEmbed{ID: "id2"},
|
|
Name: "Mike",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
db.Save(&NestedID{
|
|
ToEmbed: ToEmbed{ID: "id3"},
|
|
Name: "Steve",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
var nested []NestedID
|
|
err = db.All(&nested)
|
|
require.NoError(t, err)
|
|
require.Len(t, nested, 3)
|
|
|
|
err = db.All(&users, Limit(10), Skip(10))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 11, users[0].ID)
|
|
require.Equal(t, 20, users[9].ID)
|
|
|
|
err = db.All(&users, Limit(0), Skip(0))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 0)
|
|
}
|
|
|
|
func TestCount(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
for i := 0; i < 100; i++ {
|
|
w := User{Name: "John", ID: i + 1, Slug: fmt.Sprintf("John%d", i+1), DateOfBirth: time.Now().Add(-time.Duration(i*10) * time.Minute)}
|
|
err := db.Save(&w)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
count, err := db.Count(&User{})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 100, count)
|
|
|
|
w := User{Name: "John", ID: 101, Slug: fmt.Sprintf("John%d", 101), DateOfBirth: time.Now().Add(-time.Duration(101*10) * time.Minute)}
|
|
err = db.Save(&w)
|
|
require.NoError(t, err)
|
|
|
|
count, err = db.Count(&User{})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 101, count)
|
|
|
|
tx, err := db.Begin(true)
|
|
require.NoError(t, err)
|
|
|
|
_, err = tx.Count(User{})
|
|
require.Equal(t, ErrStructPtrNeeded, err)
|
|
|
|
count, err = tx.Count(&User{})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 101, count)
|
|
|
|
w = User{Name: "John", ID: 102, Slug: fmt.Sprintf("John%d", 102), DateOfBirth: time.Now().Add(-time.Duration(101*10) * time.Minute)}
|
|
err = tx.Save(&w)
|
|
require.NoError(t, err)
|
|
|
|
count, err = tx.Count(&User{})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 102, count)
|
|
|
|
tx.Commit()
|
|
}
|
|
|
|
func TestCountEmpty(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
user := &User{}
|
|
err := db.Init(user)
|
|
require.NoError(t, err)
|
|
|
|
count, err := db.Count(user)
|
|
require.Zero(t, count)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestOne(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
u := UniqueNameUser{Name: "John", ID: 10}
|
|
err := db.Save(&u)
|
|
require.NoError(t, err)
|
|
|
|
v := UniqueNameUser{}
|
|
err = db.One("Name", "John", &v)
|
|
require.NoError(t, err)
|
|
require.Equal(t, u, v)
|
|
|
|
for i := 0; i < 10; i++ {
|
|
w := IndexedNameUser{Name: "John", ID: i + 1, Group: "staff"}
|
|
err = db.Save(&w)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
var x IndexedNameUser
|
|
err = db.One("Name", "John", &x)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "John", x.Name)
|
|
require.Equal(t, 1, x.ID)
|
|
require.Zero(t, x.age)
|
|
require.True(t, x.DateOfBirth.IsZero())
|
|
|
|
err = db.One("Name", "Mike", &x)
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrNotFound, err)
|
|
|
|
err = db.One("", nil, &x)
|
|
require.Error(t, err)
|
|
require.True(t, ErrNotFound == err)
|
|
|
|
err = db.One("", "Mike", nil)
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrStructPtrNeeded, err)
|
|
|
|
err = db.One("", nil, nil)
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrStructPtrNeeded, err)
|
|
|
|
err = db.One("Group", "staff", &x)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, x.ID)
|
|
|
|
err = db.One("Score", 5, &x)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 5, x.ID)
|
|
|
|
err = db.One("Group", "admin", &x)
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrNotFound, err)
|
|
|
|
y := UniqueNameUser{Name: "Jake", ID: 200}
|
|
err = db.Save(&y)
|
|
require.NoError(t, err)
|
|
|
|
var y2 UniqueNameUser
|
|
err = db.One("ID", 200, &y2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, y, y2)
|
|
|
|
n := NestedID{}
|
|
n.ID = "100"
|
|
n.Name = "John"
|
|
|
|
err = db.Save(&n)
|
|
require.NoError(t, err)
|
|
|
|
var n2 NestedID
|
|
err = db.One("ID", "100", &n2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, n, n2)
|
|
}
|
|
|
|
func TestOneNotWritable(t *testing.T) {
|
|
dir, _ := ioutil.TempDir(os.TempDir(), "storm")
|
|
defer os.RemoveAll(dir)
|
|
db, _ := Open(filepath.Join(dir, "storm.db"))
|
|
|
|
err := db.Save(&User{ID: 10, Name: "John"})
|
|
require.NoError(t, err)
|
|
|
|
db.Close()
|
|
|
|
db, _ = Open(filepath.Join(dir, "storm.db"), BoltOptions(0660, &bolt.Options{
|
|
ReadOnly: true,
|
|
}))
|
|
defer db.Close()
|
|
|
|
err = db.Save(&User{ID: 20, Name: "John"})
|
|
require.Error(t, err)
|
|
|
|
var u User
|
|
err = db.One("ID", 10, &u)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 10, u.ID)
|
|
require.Equal(t, "John", u.Name)
|
|
|
|
err = db.One("Name", "John", &u)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 10, u.ID)
|
|
require.Equal(t, "John", u.Name)
|
|
}
|
|
|
|
func TestRange(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
for i := 0; i < 100; i++ {
|
|
w := User{
|
|
Name: "John",
|
|
ID: i + 1,
|
|
Slug: fmt.Sprintf("John%03d", i+1),
|
|
DateOfBirth: time.Now().Add(-time.Duration(i) * time.Hour),
|
|
Group: fmt.Sprintf("Group%03d", i%5),
|
|
}
|
|
err := db.Save(&w)
|
|
require.NoError(t, err)
|
|
z := User{Name: fmt.Sprintf("Zach%03d", i+1), ID: i + 101, Slug: fmt.Sprintf("Zach%03d", i+1)}
|
|
err = db.Save(&z)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
min := "John010"
|
|
max := "John020"
|
|
var users []User
|
|
|
|
err := db.Range("Slug", min, max, users)
|
|
require.Equal(t, ErrSlicePtrNeeded, err)
|
|
|
|
err = db.Range("Slug", min, max, &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 11)
|
|
require.Equal(t, "John010", users[0].Slug)
|
|
require.Equal(t, "John020", users[10].Slug)
|
|
|
|
err = db.Range("Slug", min, max, &users, Reverse())
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 11)
|
|
require.Equal(t, "John020", users[0].Slug)
|
|
require.Equal(t, "John010", users[10].Slug)
|
|
|
|
min = "Zach010"
|
|
max = "Zach020"
|
|
users = nil
|
|
err = db.Range("Name", min, max, &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 11)
|
|
require.Equal(t, "Zach010", users[0].Name)
|
|
require.Equal(t, "Zach020", users[10].Name)
|
|
|
|
err = db.Range("Name", min, max, &users, Reverse())
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 11)
|
|
require.Equal(t, "Zach020", users[0].Name)
|
|
require.Equal(t, "Zach010", users[10].Name)
|
|
|
|
err = db.Range("Name", min, max, &User{})
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrSlicePtrNeeded, err)
|
|
|
|
notTheRightUsers := []UniqueNameUser{}
|
|
|
|
err = db.Range("Name", min, max, ¬TheRightUsers)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 0, len(notTheRightUsers))
|
|
|
|
users = nil
|
|
|
|
err = db.Range("Age", min, max, &users)
|
|
require.Error(t, err)
|
|
require.EqualError(t, err, "not found")
|
|
|
|
err = db.Range("Age", 2, 5, &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 4)
|
|
|
|
dateMin := time.Now().Add(-time.Duration(50) * time.Hour)
|
|
dateMax := dateMin.Add(time.Duration(3) * time.Hour)
|
|
err = db.Range("DateOfBirth", dateMin, dateMax, &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 3)
|
|
require.Equal(t, "John050", users[0].Slug)
|
|
require.Equal(t, "John048", users[2].Slug)
|
|
|
|
err = db.Range("Slug", "John010", "John040", &users, Limit(10), Skip(20))
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 30, users[0].ID)
|
|
require.Equal(t, 39, users[9].ID)
|
|
|
|
err = db.Range("Slug", "John010", "John040", &users, Limit(10), Skip(20), Reverse())
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 20, users[0].ID)
|
|
require.Equal(t, 11, users[9].ID)
|
|
|
|
err = db.Range("Group", "Group002", "Group004", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 60)
|
|
}
|
|
|
|
func TestPrefix(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
for i := 0; i < 50; i++ {
|
|
w := User{
|
|
ID: i + 1,
|
|
}
|
|
|
|
if i%5 == 0 {
|
|
w.Name = "John"
|
|
w.Group = "group100"
|
|
} else {
|
|
w.Name = "Jack"
|
|
w.Group = "group200"
|
|
}
|
|
|
|
err := db.Save(&w)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
var users []User
|
|
err := db.Prefix("Name", "Jo", users)
|
|
require.Equal(t, ErrSlicePtrNeeded, err)
|
|
|
|
// Using indexes
|
|
err = db.Prefix("Name", "Jo", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 46, users[9].ID)
|
|
|
|
err = db.Prefix("Name", "Ja", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 40)
|
|
require.Equal(t, 2, users[0].ID)
|
|
require.Equal(t, 50, users[39].ID)
|
|
|
|
err = db.Prefix("Name", "Ja", &users, Limit(10), Skip(20), Reverse())
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 25, users[0].ID)
|
|
require.Equal(t, 14, users[9].ID)
|
|
|
|
// Using Select
|
|
err = db.Prefix("Group", "group1", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 1, users[0].ID)
|
|
require.Equal(t, 46, users[9].ID)
|
|
|
|
err = db.Prefix("Group", "group2", &users)
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 40)
|
|
require.Equal(t, 2, users[0].ID)
|
|
require.Equal(t, 50, users[39].ID)
|
|
|
|
err = db.Prefix("Group", "group2", &users, Limit(10), Skip(20), Reverse())
|
|
require.NoError(t, err)
|
|
require.Len(t, users, 10)
|
|
require.Equal(t, 25, users[0].ID)
|
|
require.Equal(t, 14, users[9].ID)
|
|
|
|
// Bad value
|
|
err = db.Prefix("Group", "group3", &users)
|
|
require.Equal(t, ErrNotFound, err)
|
|
}
|
|
|
|
func TestPrefixWithID(t *testing.T) {
|
|
db, cleanup := createDB(t)
|
|
defer cleanup()
|
|
|
|
type User struct {
|
|
ID string
|
|
}
|
|
|
|
require.NoError(t, db.Save(&User{ID: "1"}))
|
|
require.NoError(t, db.Save(&User{ID: "10"}))
|
|
|
|
var users []User
|
|
|
|
require.NoError(t, db.Prefix("ID", "1", &users))
|
|
require.Len(t, users, 2)
|
|
}
|