Merge pull request #201 from Bytom/v0.1

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

diff --git a/vendor/github.com/jinzhu/gorm/embedded_struct_test.go b/vendor/github.com/jinzhu/gorm/embedded_struct_test.go
new file mode 100755 (executable)
index 0000000..5f8ece5
--- /dev/null
+++ b/vendor/github.com/jinzhu/gorm/embedded_struct_test.go
@@ -0,0 +1,91 @@
+package gorm_test
+
+import "testing"
+
+type BasePost struct {
+       Id    int64
+       Title string
+       URL   string
+}
+
+type Author struct {
+       ID    string
+       Name  string
+       Email string
+}
+
+type HNPost struct {
+       BasePost
+       Author  `gorm:"embedded_prefix:user_"` // Embedded struct
+       Upvotes int32
+}
+
+type EngadgetPost struct {
+       BasePost BasePost `gorm:"embedded"`
+       Author   Author   `gorm:"embedded;embedded_prefix:author_"` // Embedded struct
+       ImageUrl string
+}
+
+func TestPrefixColumnNameForEmbeddedStruct(t *testing.T) {
+       dialect := DB.NewScope(&EngadgetPost{}).Dialect()
+       engadgetPostScope := DB.NewScope(&EngadgetPost{})
+       if !dialect.HasColumn(engadgetPostScope.TableName(), "author_id") || !dialect.HasColumn(engadgetPostScope.TableName(), "author_name") || !dialect.HasColumn(engadgetPostScope.TableName(), "author_email") {
+               t.Errorf("should has prefix for embedded columns")
+       }
+
+       if len(engadgetPostScope.PrimaryFields()) != 1 {
+               t.Errorf("should have only one primary field with embedded struct, but got %v", len(engadgetPostScope.PrimaryFields()))
+       }
+
+       hnScope := DB.NewScope(&HNPost{})
+       if !dialect.HasColumn(hnScope.TableName(), "user_id") || !dialect.HasColumn(hnScope.TableName(), "user_name") || !dialect.HasColumn(hnScope.TableName(), "user_email") {
+               t.Errorf("should has prefix for embedded columns")
+       }
+}
+
+func TestSaveAndQueryEmbeddedStruct(t *testing.T) {
+       DB.Save(&HNPost{BasePost: BasePost{Title: "news"}})
+       DB.Save(&HNPost{BasePost: BasePost{Title: "hn_news"}})
+       var news HNPost
+       if err := DB.First(&news, "title = ?", "hn_news").Error; err != nil {
+               t.Errorf("no error should happen when query with embedded struct, but got %v", err)
+       } else if news.Title != "hn_news" {
+               t.Errorf("embedded struct's value should be scanned correctly")
+       }
+
+       DB.Save(&EngadgetPost{BasePost: BasePost{Title: "engadget_news"}})
+       var egNews EngadgetPost
+       if err := DB.First(&egNews, "title = ?", "engadget_news").Error; err != nil {
+               t.Errorf("no error should happen when query with embedded struct, but got %v", err)
+       } else if egNews.BasePost.Title != "engadget_news" {
+               t.Errorf("embedded struct's value should be scanned correctly")
+       }
+
+       if DB.NewScope(&HNPost{}).PrimaryField() == nil {
+               t.Errorf("primary key with embedded struct should works")
+       }
+
+       for _, field := range DB.NewScope(&HNPost{}).Fields() {
+               if field.Name == "BasePost" {
+                       t.Errorf("scope Fields should not contain embedded struct")
+               }
+       }
+}
+
+func TestEmbeddedPointerTypeStruct(t *testing.T) {
+       type HNPost struct {
+               *BasePost
+               Upvotes int32
+       }
+
+       DB.Create(&HNPost{BasePost: &BasePost{Title: "embedded_pointer_type"}})
+
+       var hnPost HNPost
+       if err := DB.First(&hnPost, "title = ?", "embedded_pointer_type").Error; err != nil {
+               t.Errorf("No error should happen when find embedded pointer type, but got %v", err)
+       }
+
+       if hnPost.Title != "embedded_pointer_type" {
+               t.Errorf("Should find correct value for embedded pointer type")
+       }
+}