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) } }