forked from mirror/gorm
116 lines
3.1 KiB
Go
116 lines
3.1 KiB
Go
package gorm_test
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestCreate(t *testing.T) {
|
|
float := 35.03554004971999
|
|
user := User{Name: "CreateUser", Age: 18, Birthday: time.Now(), UserNum: Num(111), PasswordHash: []byte{'f', 'a', 'k', '4'}, Latitude: float}
|
|
|
|
if !DB.NewRecord(user) || !DB.NewRecord(&user) {
|
|
t.Error("User should be new record before create")
|
|
}
|
|
|
|
if count := DB.Save(&user).RowsAffected; count != 1 {
|
|
t.Error("There should be one record be affected when create record")
|
|
}
|
|
|
|
if DB.NewRecord(user) || DB.NewRecord(&user) {
|
|
t.Error("User should not new record after save")
|
|
}
|
|
|
|
var newUser User
|
|
DB.First(&newUser, user.Id)
|
|
|
|
if !reflect.DeepEqual(newUser.PasswordHash, []byte{'f', 'a', 'k', '4'}) {
|
|
t.Errorf("User's PasswordHash should be saved ([]byte)")
|
|
}
|
|
|
|
if newUser.Age != 18 {
|
|
t.Errorf("User's Age should be saved (int)")
|
|
}
|
|
|
|
if newUser.UserNum != Num(111) {
|
|
t.Errorf("User's UserNum should be saved (custom type)")
|
|
}
|
|
|
|
if newUser.Latitude != float {
|
|
t.Errorf("Float64 should not be changed after save")
|
|
}
|
|
|
|
if user.CreatedAt.IsZero() {
|
|
t.Errorf("Should have created_at after create")
|
|
}
|
|
|
|
if newUser.CreatedAt.IsZero() {
|
|
t.Errorf("Should have created_at after create")
|
|
}
|
|
|
|
DB.Model(user).Update("name", "create_user_new_name")
|
|
DB.First(&user, user.Id)
|
|
if user.CreatedAt != newUser.CreatedAt {
|
|
t.Errorf("CreatedAt should not be changed after update")
|
|
}
|
|
}
|
|
|
|
func TestCreateWithNoStdPrimaryKeyAndDefaultValues(t *testing.T) {
|
|
animal := Animal{Name: "Ferdinand"}
|
|
if DB.Save(&animal).Error != nil {
|
|
t.Errorf("No error should happen when create an record without std primary key")
|
|
}
|
|
|
|
if animal.Counter == 0 {
|
|
t.Errorf("No std primary key should be filled value after create")
|
|
}
|
|
|
|
if animal.Name != "Ferdinand" {
|
|
t.Errorf("Default value should be overrided")
|
|
}
|
|
|
|
// Test create with default value not overrided
|
|
an := Animal{From: "nerdz"}
|
|
|
|
if DB.Save(&an).Error != nil {
|
|
t.Errorf("No error should happen when create an record without std primary key")
|
|
}
|
|
|
|
// We must fetch the value again, to have the default fields updated
|
|
// (We can't do this in the update statements, since sql default can be expressions
|
|
// And be different from the fields' type (eg. a time.Time fiels has a default value of "now()"
|
|
DB.Model(Animal{}).Where(&Animal{Counter: an.Counter}).First(&an)
|
|
|
|
if an.Name != "galeone" {
|
|
t.Errorf("Default value should fill the field. But got %v", an.Name)
|
|
}
|
|
}
|
|
|
|
func TestAnonymousScanner(t *testing.T) {
|
|
user := User{Name: "anonymous_scanner", Role: Role{Name: "admin"}}
|
|
DB.Save(&user)
|
|
|
|
var user2 User
|
|
DB.First(&user2, "name = ?", "anonymous_scanner")
|
|
if user2.Role.Name != "admin" {
|
|
t.Errorf("Should be able to get anonymous scanner")
|
|
}
|
|
|
|
if !user2.IsAdmin() {
|
|
t.Errorf("Should be able to get anonymous scanner")
|
|
}
|
|
}
|
|
|
|
func TestAnonymousField(t *testing.T) {
|
|
user := User{Name: "anonymous_field", Company: Company{Name: "company"}}
|
|
DB.Save(&user)
|
|
|
|
var user2 User
|
|
DB.First(&user2, "name = ?", "anonymous_field")
|
|
DB.Model(&user2).Related(&user2.Company)
|
|
if user2.Company.Name != "company" {
|
|
t.Errorf("Should be able to get anonymous field")
|
|
}
|
|
}
|