vendor/github.com/jinzhu/gorm/pointer_test.go

   1 package gorm_test
   2
   3 import "testing"
   4
   5 type PointerStruct struct {
   6         ID   int64
   7         Name *string
   8         Num  *int
   9 }
  10
  11 type NormalStruct struct {
  12         ID   int64
  13         Name string
  14         Num  int
  15 }
  16
  17 func TestPointerFields(t *testing.T) {
  18         DB.DropTable(&PointerStruct{})
  19         DB.AutoMigrate(&PointerStruct{})
  20         var name = "pointer struct 1"
  21         var num = 100
  22         pointerStruct := PointerStruct{Name: &name, Num: &num}
  23         if DB.Create(&pointerStruct).Error != nil {
  24                 t.Errorf("Failed to save pointer struct")
  25         }
  26
  27         var pointerStructResult PointerStruct
  28         if err := DB.First(&pointerStructResult, "id = ?", pointerStruct.ID).Error; err != nil || *pointerStructResult.Name != name || *pointerStructResult.Num != num {
  29                 t.Errorf("Failed to query saved pointer struct")
  30         }
  31
  32         var tableName = DB.NewScope(&PointerStruct{}).TableName()
  33
  34         var normalStruct NormalStruct
  35         DB.Table(tableName).First(&normalStruct)
  36         if normalStruct.Name != name || normalStruct.Num != num {
  37                 t.Errorf("Failed to query saved Normal struct")
  38         }
  39
  40         var nilPointerStruct = PointerStruct{}
  41         if err := DB.Create(&nilPointerStruct).Error; err != nil {
  42                 t.Error("Failed to save nil pointer struct", err)
  43         }
  44
  45         var pointerStruct2 PointerStruct
  46         if err := DB.First(&pointerStruct2, "id = ?", nilPointerStruct.ID).Error; err != nil {
  47                 t.Error("Failed to query saved nil pointer struct", err)
  48         }
  49
  50         var normalStruct2 NormalStruct
  51         if err := DB.Table(tableName).First(&normalStruct2, "id = ?", nilPointerStruct.ID).Error; err != nil {
  52                 t.Error("Failed to query saved nil pointer struct", err)
  53         }
  54
  55         var partialNilPointerStruct1 = PointerStruct{Num: &num}
  56         if err := DB.Create(&partialNilPointerStruct1).Error; err != nil {
  57                 t.Error("Failed to save partial nil pointer struct", err)
  58         }
  59
  60         var pointerStruct3 PointerStruct
  61         if err := DB.First(&pointerStruct3, "id = ?", partialNilPointerStruct1.ID).Error; err != nil || *pointerStruct3.Num != num {
  62                 t.Error("Failed to query saved partial nil pointer struct", err)
  63         }
  64
  65         var normalStruct3 NormalStruct
  66         if err := DB.Table(tableName).First(&normalStruct3, "id = ?", partialNilPointerStruct1.ID).Error; err != nil || normalStruct3.Num != num {
  67                 t.Error("Failed to query saved partial pointer struct", err)
  68         }
  69
  70         var partialNilPointerStruct2 = PointerStruct{Name: &name}
  71         if err := DB.Create(&partialNilPointerStruct2).Error; err != nil {
  72                 t.Error("Failed to save partial nil pointer struct", err)
  73         }
  74
  75         var pointerStruct4 PointerStruct
  76         if err := DB.First(&pointerStruct4, "id = ?", partialNilPointerStruct2.ID).Error; err != nil || *pointerStruct4.Name != name {
  77                 t.Error("Failed to query saved partial nil pointer struct", err)
  78         }
  79
  80         var normalStruct4 NormalStruct
  81         if err := DB.Table(tableName).First(&normalStruct4, "id = ?", partialNilPointerStruct2.ID).Error; err != nil || normalStruct4.Name != name {
  82                 t.Error("Failed to query saved partial pointer struct", err)
  83         }
  84 }