Merge pull request #201 from Bytom/v0.1

[bytom/vapor.git] / vendor / github.com / jinzhu / gorm / pointer_test.go

diff --git a/vendor/github.com/jinzhu/gorm/pointer_test.go b/vendor/github.com/jinzhu/gorm/pointer_test.go
new file mode 100755 (executable)
index 0000000..2a68a5a
--- /dev/null
+++ b/vendor/github.com/jinzhu/gorm/pointer_test.go
@@ -0,0 +1,84 @@
+package gorm_test
+
+import "testing"
+
+type PointerStruct struct {
+       ID   int64
+       Name *string
+       Num  *int
+}
+
+type NormalStruct struct {
+       ID   int64
+       Name string
+       Num  int
+}
+
+func TestPointerFields(t *testing.T) {
+       DB.DropTable(&PointerStruct{})
+       DB.AutoMigrate(&PointerStruct{})
+       var name = "pointer struct 1"
+       var num = 100
+       pointerStruct := PointerStruct{Name: &name, Num: &num}
+       if DB.Create(&pointerStruct).Error != nil {
+               t.Errorf("Failed to save pointer struct")
+       }
+
+       var pointerStructResult PointerStruct
+       if err := DB.First(&pointerStructResult, "id = ?", pointerStruct.ID).Error; err != nil || *pointerStructResult.Name != name || *pointerStructResult.Num != num {
+               t.Errorf("Failed to query saved pointer struct")
+       }
+
+       var tableName = DB.NewScope(&PointerStruct{}).TableName()
+
+       var normalStruct NormalStruct
+       DB.Table(tableName).First(&normalStruct)
+       if normalStruct.Name != name || normalStruct.Num != num {
+               t.Errorf("Failed to query saved Normal struct")
+       }
+
+       var nilPointerStruct = PointerStruct{}
+       if err := DB.Create(&nilPointerStruct).Error; err != nil {
+               t.Error("Failed to save nil pointer struct", err)
+       }
+
+       var pointerStruct2 PointerStruct
+       if err := DB.First(&pointerStruct2, "id = ?", nilPointerStruct.ID).Error; err != nil {
+               t.Error("Failed to query saved nil pointer struct", err)
+       }
+
+       var normalStruct2 NormalStruct
+       if err := DB.Table(tableName).First(&normalStruct2, "id = ?", nilPointerStruct.ID).Error; err != nil {
+               t.Error("Failed to query saved nil pointer struct", err)
+       }
+
+       var partialNilPointerStruct1 = PointerStruct{Num: &num}
+       if err := DB.Create(&partialNilPointerStruct1).Error; err != nil {
+               t.Error("Failed to save partial nil pointer struct", err)
+       }
+
+       var pointerStruct3 PointerStruct
+       if err := DB.First(&pointerStruct3, "id = ?", partialNilPointerStruct1.ID).Error; err != nil || *pointerStruct3.Num != num {
+               t.Error("Failed to query saved partial nil pointer struct", err)
+       }
+
+       var normalStruct3 NormalStruct
+       if err := DB.Table(tableName).First(&normalStruct3, "id = ?", partialNilPointerStruct1.ID).Error; err != nil || normalStruct3.Num != num {
+               t.Error("Failed to query saved partial pointer struct", err)
+       }
+
+       var partialNilPointerStruct2 = PointerStruct{Name: &name}
+       if err := DB.Create(&partialNilPointerStruct2).Error; err != nil {
+               t.Error("Failed to save partial nil pointer struct", err)
+       }
+
+       var pointerStruct4 PointerStruct
+       if err := DB.First(&pointerStruct4, "id = ?", partialNilPointerStruct2.ID).Error; err != nil || *pointerStruct4.Name != name {
+               t.Error("Failed to query saved partial nil pointer struct", err)
+       }
+
+       var normalStruct4 NormalStruct
+       if err := DB.Table(tableName).First(&normalStruct4, "id = ?", partialNilPointerStruct2.ID).Error; err != nil || normalStruct4.Name != name {
+               t.Error("Failed to query saved partial pointer struct", err)
+       }
+}