84 lines
1.7 KiB
Go
84 lines
1.7 KiB
Go
package mdb
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
)
|
|
|
|
func TestCollection(t *testing.T) {
|
|
testWithDB(t, "insert on secondary", func(t *testing.T, db *DB) {
|
|
c := db.Users.c
|
|
c.primary = false
|
|
if _, err := c.Insert(User{}); !errors.Is(err, ErrReadOnly) {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
|
|
testWithDB(t, "insert validation error", func(t *testing.T, db *DB) {
|
|
c := db.Users.c
|
|
user := User{
|
|
ID: c.NextID(),
|
|
Email: "a@b.com",
|
|
}
|
|
if _, err := c.Insert(user); !errors.Is(err, ErrInvalidName) {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
|
|
testWithDB(t, "insert duplicate", func(t *testing.T, db *DB) {
|
|
c := db.Users.c
|
|
user := User{
|
|
ID: c.NextID(),
|
|
Name: "adsf",
|
|
Email: "a@b.com",
|
|
}
|
|
if _, err := c.Insert(user); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := c.Insert(user); !errors.Is(err, ErrDuplicate) {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
|
|
testWithDB(t, "update on secondary", func(t *testing.T, db *DB) {
|
|
c := db.Users.c
|
|
user := User{ID: c.NextID(), Name: "adsf", Email: "a@b.com"}
|
|
if _, err := c.Insert(user); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
c.primary = false
|
|
|
|
err := c.Update(user.ID, func(u User) (User, error) {
|
|
u.Name = "xxx"
|
|
return u, nil
|
|
})
|
|
if !errors.Is(err, ErrReadOnly) {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
|
|
testWithDB(t, "update not found", func(t *testing.T, db *DB) {
|
|
c := db.Users.c
|
|
user := User{ID: c.NextID(), Name: "adsf", Email: "a@b.com"}
|
|
if _, err := c.Insert(user); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err := c.Update(user.ID+1, func(u User) (User, error) {
|
|
u.Name = "xxx"
|
|
return u, nil
|
|
})
|
|
if !errors.Is(err, ErrNotFound) {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
|
|
// Update w/ failed valiation
|
|
|
|
// Delete on secondary
|
|
// Delete not found
|
|
// Get found
|
|
// Get not found
|
|
}
|