Carlos Florêncio
32 lines
868 B
Go
32 lines
868 B
Go
package db
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
)
|
|
|
|
func TestConfigurePostgresTimeTypesUsesUTCScans(t *testing.T) {
|
|
typeMap := pgtype.NewMap()
|
|
if err := configurePostgresTimeTypes(typeMap); err != nil {
|
|
t.Fatalf("configurePostgresTimeTypes: %v", err)
|
|
}
|
|
|
|
timestamp, ok := typeMap.TypeForName("timestamp")
|
|
if !ok {
|
|
t.Fatal("timestamp type is not registered")
|
|
}
|
|
if codec, ok := timestamp.Codec.(*pgtype.TimestampCodec); !ok || codec.ScanLocation != time.UTC {
|
|
t.Fatalf("timestamp scan location = %#v, want UTC", timestamp.Codec)
|
|
}
|
|
|
|
timestamptz, ok := typeMap.TypeForName("timestamptz")
|
|
if !ok {
|
|
t.Fatal("timestamptz type is not registered")
|
|
}
|
|
if codec, ok := timestamptz.Codec.(*pgtype.TimestamptzCodec); !ok || codec.ScanLocation != time.UTC {
|
|
t.Fatalf("timestamptz scan location = %#v, want UTC", timestamptz.Codec)
|
|
}
|
|
}
|