mirror of https://github.com/go-gorm/gorm.git
60 lines
2.3 KiB
Go
60 lines
2.3 KiB
Go
package tests_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func TestDefaultValue(t *testing.T) {
|
|
type Harumph struct {
|
|
gorm.Model
|
|
Email string `gorm:"not null;index:,unique"`
|
|
Name string `gorm:"notNull;default:foo"`
|
|
Name2 string `gorm:"size:233;not null;default:'foo'"`
|
|
Name3 string `gorm:"size:233;notNull;default:''"`
|
|
Age int `gorm:"default:18"`
|
|
Created time.Time `gorm:"default:2000-01-02"`
|
|
Enabled bool `gorm:"default:true"`
|
|
}
|
|
|
|
DB.Migrator().DropTable(&Harumph{})
|
|
|
|
if err := DB.AutoMigrate(&Harumph{}); err != nil {
|
|
t.Fatalf("Failed to migrate with default value, got error: %v", err)
|
|
}
|
|
|
|
harumph := Harumph{Email: "hello@gorm.io"}
|
|
if err := DB.Create(&harumph).Error; err != nil {
|
|
t.Fatalf("Failed to create data with default value, got error: %v", err)
|
|
} else if harumph.Name != "foo" || harumph.Name2 != "foo" || harumph.Name3 != "" || harumph.Age != 18 || !harumph.Enabled || harumph.Created.Format("20060102") != "20000102" {
|
|
t.Fatalf("Failed to create data with default value, got: %+v", harumph)
|
|
}
|
|
|
|
var result Harumph
|
|
if err := DB.First(&result, "email = ?", "hello@gorm.io").Error; err != nil {
|
|
t.Fatalf("Failed to find created data, got error: %v", err)
|
|
} else if result.Name != "foo" || result.Name2 != "foo" || result.Name3 != "" || result.Age != 18 || !result.Enabled || result.Created.Format("20060102") != "20000102" {
|
|
t.Fatalf("Failed to find created data with default data, got %+v", result)
|
|
}
|
|
|
|
type Harumph2 struct {
|
|
ID int `gorm:"default:0"`
|
|
Email string `gorm:"not null;index:,unique"`
|
|
Name string `gorm:"notNull;default:foo"`
|
|
Name2 string `gorm:"size:233;not null;default:'foo'"`
|
|
Name3 string `gorm:"size:233;notNull;default:''"`
|
|
Age int `gorm:"default:18"`
|
|
Created time.Time `gorm:"default:2000-01-02"`
|
|
Enabled bool `gorm:"default:true"`
|
|
}
|
|
|
|
harumph2 := Harumph2{ID: 2, Email: "hello2@gorm.io"}
|
|
if err := DB.Table("harumphs").Create(&harumph2).Error; err != nil {
|
|
t.Fatalf("Failed to create data with default value, got error: %v", err)
|
|
} else if harumph2.ID != 2 || harumph2.Name != "foo" || harumph2.Name2 != "foo" || harumph2.Name3 != "" || harumph2.Age != 18 || !harumph2.Enabled || harumph2.Created.Format("20060102") != "20000102" {
|
|
t.Fatalf("Failed to create data with default value, got: %+v", harumph2)
|
|
}
|
|
}
|