NullTime in normalizeTime & tests
This commit is contained in:
parent
58344b82a5
commit
f48ee87ccd
8 changed files with 49 additions and 17 deletions
|
@ -1,9 +1,23 @@
|
|||
package datastore
|
||||
|
||||
import "time"
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
func normalizeTime(t ...*time.Time) {
|
||||
"github.com/lib/pq"
|
||||
)
|
||||
|
||||
func normalizeTime(t ...interface{}) {
|
||||
for _, v := range t {
|
||||
*v = v.In(time.UTC)
|
||||
switch u := v.(type) {
|
||||
default:
|
||||
fmt.Printf("unexpected type %T", u)
|
||||
case *time.Time:
|
||||
x, _ := v.(*time.Time)
|
||||
*x = x.In(time.UTC)
|
||||
case *pq.NullTime:
|
||||
x, _ := v.(*pq.NullTime)
|
||||
*x = pq.NullTime{Time: x.Time.In(time.UTC), Valid: x.Valid}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -36,6 +36,7 @@ func TestGeneraStore_Get_db(t *testing.T) {
|
|||
}
|
||||
|
||||
normalizeTime(&want.CreatedAt, &want.UpdatedAt, &want.DeletedAt)
|
||||
normalizeTime(&genus.CreatedAt, &genus.UpdatedAt, &genus.DeletedAt)
|
||||
if !reflect.DeepEqual(genus, want) {
|
||||
t.Errorf("got genus %+v, want %+v", genus, want)
|
||||
}
|
||||
|
@ -74,8 +75,9 @@ func TestGeneraStore_List_db(t *testing.T) {
|
|||
t.Fatal(err)
|
||||
}
|
||||
|
||||
for _, g := range want {
|
||||
normalizeTime(&g.CreatedAt, &g.UpdatedAt, &g.DeletedAt)
|
||||
for i := range want {
|
||||
normalizeTime(&want[i].CreatedAt, &want[i].UpdatedAt, &want[i].DeletedAt)
|
||||
normalizeTime(&genera[i].CreatedAt, &genera[i].UpdatedAt, &genera[i].DeletedAt)
|
||||
}
|
||||
if !reflect.DeepEqual(genera, want) {
|
||||
t.Errorf("got genera %+v, want %+v", genera, want)
|
||||
|
|
|
@ -38,6 +38,7 @@ func TestSpeciesStore_Get_db(t *testing.T) {
|
|||
}
|
||||
|
||||
normalizeTime(&want.CreatedAt, &want.UpdatedAt, &want.DeletedAt)
|
||||
normalizeTime(&species.CreatedAt, &species.UpdatedAt, &species.DeletedAt)
|
||||
if !reflect.DeepEqual(species, want) {
|
||||
t.Errorf("got species %+v, want %+v", species, want)
|
||||
}
|
||||
|
@ -77,8 +78,9 @@ func TestSpeciesStore_List_db(t *testing.T) {
|
|||
t.Fatal(err)
|
||||
}
|
||||
|
||||
for _, g := range want {
|
||||
normalizeTime(&g.CreatedAt, &g.UpdatedAt, &g.DeletedAt)
|
||||
for i := range want {
|
||||
normalizeTime(&want[i].CreatedAt, &want[i].UpdatedAt, &want[i].DeletedAt)
|
||||
normalizeTime(&species[i].CreatedAt, &species[i].UpdatedAt, &species[i].DeletedAt)
|
||||
}
|
||||
if !reflect.DeepEqual(species, want) {
|
||||
t.Errorf("got species %+v, want %+v", species, want)
|
||||
|
|
|
@ -40,6 +40,7 @@ func TestStrainsStore_Get_db(t *testing.T) {
|
|||
}
|
||||
|
||||
normalizeTime(&want.CreatedAt, &want.UpdatedAt, &want.DeletedAt)
|
||||
normalizeTime(&strain.CreatedAt, &strain.UpdatedAt, &strain.DeletedAt)
|
||||
|
||||
if !reflect.DeepEqual(strain, want) {
|
||||
t.Errorf("got strain %+v, want %+v", strain, want)
|
||||
|
@ -80,8 +81,9 @@ func TestStrainsStore_List_db(t *testing.T) {
|
|||
t.Fatal(err)
|
||||
}
|
||||
|
||||
for _, g := range want {
|
||||
normalizeTime(&g.CreatedAt, &g.UpdatedAt, &g.DeletedAt)
|
||||
for i := range want {
|
||||
normalizeTime(&want[i].CreatedAt, &want[i].UpdatedAt, &want[i].DeletedAt)
|
||||
normalizeTime(&strains[i].CreatedAt, &strains[i].UpdatedAt, &strains[i].DeletedAt)
|
||||
}
|
||||
if !reflect.DeepEqual(strains, want) {
|
||||
t.Errorf("got strains %+v, want %+v", strains, want)
|
||||
|
|
Reference in a new issue