From b5cff5efecea27a515980321131ddae7cddd8338 Mon Sep 17 00:00:00 2001
From: jiangzhi <jiangzhi@qiniu.com>
Date: Sat, 19 Sep 2020 17:10:56 +0800
Subject: [PATCH] filed for upsert

---
 field/field_test.go | 76 ++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 75 insertions(+), 1 deletion(-)

diff --git a/field/field_test.go b/field/field_test.go
index 12b1b6b..b9bba55 100644
--- a/field/field_test.go
+++ b/field/field_test.go
@@ -112,6 +112,20 @@ func TestBeforeUpdate(t *testing.T) {
 
 }
 
+type UserField struct {
+	DefaultField `bson:",inline"`
+
+	Name         string             `bson:"name"`
+	Age          int                `bson:"age"`
+	CreateTimeAt int64              `bson:"createTimeAt"`
+	UpdateTimeAt time.Time          `bson:"updateTimeAt"`
+	MyId         primitive.ObjectID `bson:"myId"`
+}
+
+func (u *UserField) CustomFields() CustomFieldsBuilder {
+	return NewCustom().SetCreateAt("CreateTimeAt").SetUpdateAt("UpdateTimeAt").SetId("MyId")
+}
+
 func TestBeforeUpsert(t *testing.T) {
 	ast := require.New(t)
 
@@ -125,7 +139,7 @@ func TestBeforeUpsert(t *testing.T) {
 	ast.NotEqual(primitive.NilObjectID, u.Id)
 	// custom fields
 	ast.NotEqual(time.Time{}, u.CreateTimeAt)
-	ast.NotEqual(time.Time{}, u.UpdateTimeAt)
+	ast.NotEqual(0, u.UpdateTimeAt)
 	ast.NotEqual(primitive.NilObjectID, u.MyId)
 
 	u1, u2 := &User{Name: "Lucas", Age: 7}, &User{Name: "Alice", Age: 8}
@@ -136,6 +150,8 @@ func TestBeforeUpsert(t *testing.T) {
 	for _, v := range us {
 		ast.NotEqual(time.Time{}, v.CreateAt)
 		ast.NotEqual(time.Time{}, v.UpdateAt)
+		ast.NotEqual(time.Time{}, u.CreateTimeAt)
+		ast.NotEqual(0, u.UpdateTimeAt)
 		ast.NotEqual(primitive.NilObjectID, v.Id)
 	}
 
@@ -167,6 +183,64 @@ func TestBeforeUpsert(t *testing.T) {
 
 }
 
+// same as TestBeforeUpsert, just switch type of CreateTimeAt and UpdateTimeAt
+func TestBeforeUpsertUserFiled(t *testing.T) {
+	ast := require.New(t)
+
+	// with empty fileds
+	u := &UserField{Name: "Lucas", Age: 7}
+	err := Do(u, BeforeUpsert)
+	ast.NoError(err)
+	// default fields
+	ast.NotEqual(time.Time{}, u.CreateAt)
+	ast.NotEqual(time.Time{}, u.UpdateAt)
+	ast.NotEqual(primitive.NilObjectID, u.Id)
+	// custom fields
+	ast.NotEqual(0, u.CreateTimeAt)
+	ast.NotEqual(time.Time{}, u.UpdateTimeAt)
+	ast.NotEqual(primitive.NilObjectID, u.MyId)
+
+	u1, u2 := &UserField{Name: "Lucas", Age: 7}, &UserField{Name: "Alice", Age: 8}
+	us := []*UserField{u1, u2}
+	err = Do(us, BeforeUpsert)
+	ast.NoError(err)
+
+	for _, v := range us {
+		ast.NotEqual(time.Time{}, v.CreateAt)
+		ast.NotEqual(time.Time{}, v.UpdateAt)
+		ast.NotEqual(0, u.CreateTimeAt)
+		ast.NotEqual(time.Time{}, u.UpdateTimeAt)
+		ast.NotEqual(primitive.NilObjectID, v.Id)
+	}
+
+	u3 := User{Name: "Lucas", Age: 7}
+	err = Do(u3, BeforeUpsert)
+	ast.NoError(err)
+
+	// upsert with valid value
+	tBefore3s := time.Now().Add(-3 * time.Second)
+	id := primitive.NewObjectID()
+	u = &UserField{Name: "Lucas", Age: 7}
+	u.CreateAt = tBefore3s
+	u.UpdateAt = tBefore3s
+	u.Id = id
+	u.MyId = id
+	u.CreateTimeAt = tBefore3s.Unix()
+	u.UpdateTimeAt = tBefore3s
+
+	err = Do(u, BeforeUpsert)
+	ast.NoError(err)
+
+	ast.Equal(tBefore3s, u.CreateAt)
+	ast.Equal(id, u.Id)
+	ast.NotEqual(tBefore3s, u.UpdateAt)
+
+	ast.NotEqual(tBefore3s, u.UpdateTimeAt)
+	ast.Equal(id, u.MyId)
+	ast.Equal(tBefore3s.Unix(), u.CreateTimeAt)
+
+}
+
 func TestNilError(t *testing.T) {
 	ast := require.New(t)