115 lines
2.3 KiB
Go
115 lines
2.3 KiB
Go
package datastore
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/jmoiron/modl"
|
|
"github.com/thermokarst/bactdb/models"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
func insertUser(t *testing.T, tx *modl.Transaction) *models.User {
|
|
// Test on a clean database
|
|
tx.Exec(`DELETE FROM users;`)
|
|
|
|
user := newUser()
|
|
if err := tx.Insert(user); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return user
|
|
}
|
|
|
|
func newUser() *models.User {
|
|
hashedPassword, _ := bcrypt.GenerateFromPassword([]byte("password"), 10)
|
|
return &models.User{
|
|
Username: "Test User",
|
|
Password: string(hashedPassword),
|
|
}
|
|
}
|
|
|
|
func TestUsersStore_Get_db(t *testing.T) {
|
|
tx, _ := DB.Begin()
|
|
defer tx.Rollback()
|
|
|
|
want := insertUser(t, tx)
|
|
|
|
d := NewDatastore(tx)
|
|
|
|
user, err := d.Users.Get(want.Id)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
normalizeTime(&want.CreatedAt, &want.UpdatedAt, &want.DeletedAt)
|
|
normalizeTime(&user.CreatedAt, &user.UpdatedAt, &user.DeletedAt)
|
|
if !reflect.DeepEqual(user, want) {
|
|
t.Errorf("got user %+v, want %+v", user, want)
|
|
}
|
|
}
|
|
|
|
func TestUsersStore_Create_db(t *testing.T) {
|
|
tx, _ := DB.Begin()
|
|
defer tx.Rollback()
|
|
|
|
user := newUser()
|
|
|
|
d := NewDatastore(tx)
|
|
|
|
created, err := d.Users.Create(user)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !created {
|
|
t.Error("!created")
|
|
}
|
|
if user.Id == 0 {
|
|
t.Error("want nonzero user.Id after submitting")
|
|
}
|
|
}
|
|
|
|
func TestUsersStore_List_db(t *testing.T) {
|
|
tx, _ := DB.Begin()
|
|
defer tx.Rollback()
|
|
|
|
user := insertUser(t, tx)
|
|
want := []*models.User{user}
|
|
|
|
d := NewDatastore(tx)
|
|
|
|
users, err := d.Users.List(&models.UserListOptions{ListOptions: models.ListOptions{Page: 1, PerPage: 10}})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
for i := range want {
|
|
normalizeTime(&want[i].CreatedAt, &want[i].UpdatedAt, &want[i].DeletedAt)
|
|
normalizeTime(&users[i].CreatedAt, &users[i].UpdatedAt, &users[i].DeletedAt)
|
|
}
|
|
if !reflect.DeepEqual(users, want) {
|
|
t.Errorf("got users %+v, want %+v", users, want)
|
|
}
|
|
}
|
|
|
|
func TestUsersStore_Authenticate_db(t *testing.T) {
|
|
tx, _ := DB.Begin()
|
|
defer tx.Rollback()
|
|
|
|
user := insertUser(t, tx)
|
|
|
|
want := &models.UserSession{
|
|
AccessLevel: "read",
|
|
Genus: "hymenobacter",
|
|
}
|
|
|
|
d := NewDatastore(tx)
|
|
|
|
user_session, err := d.Users.Authenticate(user.Username, "password")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(user_session, want) {
|
|
t.Errorf("got session %+v, want %+v", user_session, want)
|
|
}
|
|
}
|