This commit is contained in:
Lunny Xiao 2021-07-17 17:29:58 +08:00
parent 466516dafc
commit c882353b9a
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
1 changed files with 45 additions and 0 deletions

View File

@ -53,9 +53,18 @@ func TestTimeUserTimeDiffLoc(t *testing.T) {
assert.NoError(t, PrepareEngine()) assert.NoError(t, PrepareEngine())
loc, err := time.LoadLocation("Asia/Shanghai") loc, err := time.LoadLocation("Asia/Shanghai")
assert.NoError(t, err) assert.NoError(t, err)
oldTZLoc := testEngine.GetTZLocation()
defer func() {
testEngine.SetTZLocation(oldTZLoc)
}()
testEngine.SetTZLocation(loc) testEngine.SetTZLocation(loc)
dbLoc, err := time.LoadLocation("America/New_York") dbLoc, err := time.LoadLocation("America/New_York")
assert.NoError(t, err) assert.NoError(t, err)
oldDBLoc := testEngine.GetTZDatabase()
defer func() {
testEngine.SetTZDatabase(oldDBLoc)
}()
testEngine.SetTZDatabase(dbLoc) testEngine.SetTZDatabase(dbLoc)
type TimeUser2 struct { type TimeUser2 struct {
@ -118,9 +127,18 @@ func TestTimeUserCreatedDiffLoc(t *testing.T) {
assert.NoError(t, PrepareEngine()) assert.NoError(t, PrepareEngine())
loc, err := time.LoadLocation("Asia/Shanghai") loc, err := time.LoadLocation("Asia/Shanghai")
assert.NoError(t, err) assert.NoError(t, err)
oldTZLoc := testEngine.GetTZLocation()
defer func() {
testEngine.SetTZLocation(oldTZLoc)
}()
testEngine.SetTZLocation(loc) testEngine.SetTZLocation(loc)
dbLoc, err := time.LoadLocation("America/New_York") dbLoc, err := time.LoadLocation("America/New_York")
assert.NoError(t, err) assert.NoError(t, err)
oldDBLoc := testEngine.GetTZDatabase()
defer func() {
testEngine.SetTZDatabase(oldDBLoc)
}()
testEngine.SetTZDatabase(dbLoc) testEngine.SetTZDatabase(dbLoc)
type UserCreated2 struct { type UserCreated2 struct {
@ -204,9 +222,18 @@ func TestTimeUserUpdatedDiffLoc(t *testing.T) {
assert.NoError(t, PrepareEngine()) assert.NoError(t, PrepareEngine())
loc, err := time.LoadLocation("Asia/Shanghai") loc, err := time.LoadLocation("Asia/Shanghai")
assert.NoError(t, err) assert.NoError(t, err)
oldTZLoc := testEngine.GetTZLocation()
defer func() {
testEngine.SetTZLocation(oldTZLoc)
}()
testEngine.SetTZLocation(loc) testEngine.SetTZLocation(loc)
dbLoc, err := time.LoadLocation("America/New_York") dbLoc, err := time.LoadLocation("America/New_York")
assert.NoError(t, err) assert.NoError(t, err)
oldDBLoc := testEngine.GetTZDatabase()
defer func() {
testEngine.SetTZDatabase(oldDBLoc)
}()
testEngine.SetTZDatabase(dbLoc) testEngine.SetTZDatabase(dbLoc)
type UserUpdated2 struct { type UserUpdated2 struct {
@ -311,9 +338,18 @@ func TestTimeUserDeletedDiffLoc(t *testing.T) {
assert.NoError(t, PrepareEngine()) assert.NoError(t, PrepareEngine())
loc, err := time.LoadLocation("Asia/Shanghai") loc, err := time.LoadLocation("Asia/Shanghai")
assert.NoError(t, err) assert.NoError(t, err)
oldTZLoc := testEngine.GetTZLocation()
defer func() {
testEngine.SetTZLocation(oldTZLoc)
}()
testEngine.SetTZLocation(loc) testEngine.SetTZLocation(loc)
dbLoc, err := time.LoadLocation("America/New_York") dbLoc, err := time.LoadLocation("America/New_York")
assert.NoError(t, err) assert.NoError(t, err)
oldDBLoc := testEngine.GetTZDatabase()
defer func() {
testEngine.SetTZDatabase(oldDBLoc)
}()
testEngine.SetTZDatabase(dbLoc) testEngine.SetTZDatabase(dbLoc)
type UserDeleted2 struct { type UserDeleted2 struct {
@ -435,9 +471,18 @@ func TestCustomTimeUserDeletedDiffLoc(t *testing.T) {
assert.NoError(t, PrepareEngine()) assert.NoError(t, PrepareEngine())
loc, err := time.LoadLocation("Asia/Shanghai") loc, err := time.LoadLocation("Asia/Shanghai")
assert.NoError(t, err) assert.NoError(t, err)
oldTZLoc := testEngine.GetTZLocation()
defer func() {
testEngine.SetTZLocation(oldTZLoc)
}()
testEngine.SetTZLocation(loc) testEngine.SetTZLocation(loc)
dbLoc, err := time.LoadLocation("America/New_York") dbLoc, err := time.LoadLocation("America/New_York")
assert.NoError(t, err) assert.NoError(t, err)
oldDBLoc := testEngine.GetTZDatabase()
defer func() {
testEngine.SetTZDatabase(oldDBLoc)
}()
testEngine.SetTZDatabase(dbLoc) testEngine.SetTZDatabase(dbLoc)
type UserDeleted4 struct { type UserDeleted4 struct {