mirror of https://github.com/mattn/go-sqlite3.git
Refactor timestamp tests
This commit is contained in:
Micah Stetson
parent
f6d10a2a58
commit
f65db67ea3
211
sqlite3_test.go
211
sqlite3_test.go
|
|
@ -245,35 +245,34 @@ func TestTimestamp(t *testing.T) {
|
||||||
defer db.Close()
|
defer db.Close()
|
||||||
|
|
||||||
_, err = db.Exec("DROP TABLE foo")
|
_, err = db.Exec("DROP TABLE foo")
|
||||||
_, err = db.Exec("CREATE TABLE foo(id INTEGER, ts timeSTAMP)")
|
_, err = db.Exec("CREATE TABLE foo(id INTEGER, ts timeSTAMP, dt DATETIME)")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal("Failed to create table:", err)
|
t.Fatal("Failed to create table:", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
timestamp1 := time.Date(2012, time.April, 6, 22, 50, 0, 0, time.UTC)
|
timestamp1 := time.Date(2012, time.April, 6, 22, 50, 0, 0, time.UTC)
|
||||||
_, err = db.Exec("INSERT INTO foo(id, ts) VALUES(1, ?)", timestamp1)
|
timestamp2 := time.Date(2006, time.January, 2, 15, 4, 5, 3000000, time.UTC)
|
||||||
if err != nil {
|
tests := []struct {
|
||||||
t.Fatal("Failed to insert timestamp:", err)
|
value interface{}
|
||||||
|
expected time.Time
|
||||||
|
}{
|
||||||
|
{timestamp1, timestamp1},
|
||||||
|
{timestamp1.Unix(), timestamp1},
|
||||||
|
{"nonsense", time.Time{}},
|
||||||
|
{"0000-00-00 00:00:00", time.Time{}},
|
||||||
|
{timestamp1.In(time.FixedZone("TEST", -7*3600)), timestamp1},
|
||||||
|
{"2012-11-04", time.Date(2012, time.November, 4, 0, 0, 0, 0, time.UTC)},
|
||||||
|
{timestamp2, timestamp2},
|
||||||
|
{"2006-01-02 15:04:05.003", timestamp2},
|
||||||
|
}
|
||||||
|
for i := range tests {
|
||||||
|
_, err = db.Exec("INSERT INTO foo(id, ts, dt) VALUES(?, ?, ?)", i, tests[i].value, tests[i].value)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal("Failed to insert timestamp:", err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
timestamp2 := time.Date(2012, time.April, 6, 23, 22, 0, 0, time.UTC)
|
rows, err := db.Query("SELECT id, ts, dt FROM foo ORDER BY id ASC")
|
||||||
_, err = db.Exec("INSERT INTO foo(id, ts) VALUES(2, ?)", timestamp2.Unix())
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Failed to insert timestamp:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = db.Exec("INSERT INTO foo(id, ts) VALUES(3, ?)", "nonsense")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Failed to insert nonsense:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
timestamp4 := time.Date(2012, time.April, 6, 23, 22, 0, 0, time.FixedZone("TEST", -7*3600))
|
|
||||||
_, err = db.Exec("INSERT INTO foo(id, ts) VALUES(4, ?)", timestamp4)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Failed to insert timestamp:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
rows, err := db.Query("SELECT id, ts FROM foo ORDER BY id ASC")
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal("Unable to query foo table:", err)
|
t.Fatal("Unable to query foo table:", err)
|
||||||
}
|
}
|
||||||
|
|
@ -282,44 +281,27 @@ func TestTimestamp(t *testing.T) {
|
||||||
seen := 0
|
seen := 0
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
var id int
|
var id int
|
||||||
var ts time.Time
|
var ts, dt time.Time
|
||||||
|
|
||||||
if err := rows.Scan(&id, &ts); err != nil {
|
if err := rows.Scan(&id, &ts, &dt); err != nil {
|
||||||
t.Error("Unable to scan results:", err)
|
t.Error("Unable to scan results:", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
if id < 0 || id >= len(tests) {
|
||||||
if id == 1 {
|
t.Error("Bad row id: ", id)
|
||||||
seen += 1
|
continue
|
||||||
if !timestamp1.Equal(ts) {
|
|
||||||
t.Errorf("Value for id 1 should be %v, not %v", timestamp1, ts)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
seen++
|
||||||
if id == 2 {
|
if !tests[id].expected.Equal(ts) {
|
||||||
seen += 1
|
t.Errorf("Timestamp value for id %v should be %v, not %v", id, tests[id].expected, ts)
|
||||||
if !timestamp2.Equal(ts) {
|
|
||||||
t.Errorf("Value for id 2 should be %v, not %v", timestamp2, ts)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
if !tests[id].expected.Equal(dt) {
|
||||||
if id == 3 {
|
t.Errorf("Datetime value for id %v should be %v, not %v", id, tests[id].expected, dt)
|
||||||
seen += 1
|
|
||||||
if !ts.IsZero() {
|
|
||||||
t.Errorf("Value for id 3 should be the zero time, not %v", ts)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if id == 4 {
|
|
||||||
seen += 1
|
|
||||||
if !timestamp4.Equal(ts) {
|
|
||||||
t.Errorf("Value for id 4 should be %v, not %v", timestamp4, ts)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if seen != 4 {
|
if seen != len(tests) {
|
||||||
t.Error("Expected to see four timestamps")
|
t.Errorf("Expected to see %d rows", len(tests))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -414,130 +396,3 @@ func TestBoolean(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDateOnlyTimestamp(t *testing.T) {
|
|
||||||
// make sure that timestamps without times are extractable. these are generated when
|
|
||||||
// e.g., you use the sqlite `DATE()` built-in.
|
|
||||||
|
|
||||||
db, er := sql.Open("sqlite3", "db.sqlite")
|
|
||||||
if er != nil {
|
|
||||||
t.Fatal(er)
|
|
||||||
}
|
|
||||||
defer func() {
|
|
||||||
db.Close()
|
|
||||||
os.Remove("db.sqlite")
|
|
||||||
}()
|
|
||||||
|
|
||||||
_, er = db.Exec(`
|
|
||||||
CREATE TABLE test
|
|
||||||
( entry_id INTEGER PRIMARY KEY AUTOINCREMENT
|
|
||||||
, entry_published TIMESTAMP NOT NULL
|
|
||||||
)
|
|
||||||
`)
|
|
||||||
if er != nil {
|
|
||||||
t.Fatal(er)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, er = db.Exec(`
|
|
||||||
INSERT INTO test VALUES ( 1, '2012-11-04')
|
|
||||||
`)
|
|
||||||
if er != nil {
|
|
||||||
t.Fatal(er)
|
|
||||||
}
|
|
||||||
|
|
||||||
rows, er := db.Query("SELECT entry_id, entry_published FROM test")
|
|
||||||
if er != nil {
|
|
||||||
t.Fatal(er)
|
|
||||||
}
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
if !rows.Next() {
|
|
||||||
if er := rows.Err(); er != nil {
|
|
||||||
t.Fatal(er)
|
|
||||||
} else {
|
|
||||||
t.Fatalf("Unable to extract row containing date-only timestamp")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var entryId int64
|
|
||||||
var entryPublished time.Time
|
|
||||||
|
|
||||||
if er := rows.Scan(&entryId, &entryPublished); er != nil {
|
|
||||||
t.Fatal(er)
|
|
||||||
}
|
|
||||||
|
|
||||||
if entryId != 1 {
|
|
||||||
t.Errorf("EntryId does not match inserted value")
|
|
||||||
}
|
|
||||||
|
|
||||||
if entryPublished.Year() != 2012 || entryPublished.Month() != 11 || entryPublished.Day() != 4 {
|
|
||||||
t.Errorf("Extracted time does not match inserted value")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDatetime(t *testing.T) {
|
|
||||||
db, err := sql.Open("sqlite3", "./datetime.db")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Failed to open database:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
db.Close()
|
|
||||||
os.Remove("./datetime.db")
|
|
||||||
}()
|
|
||||||
|
|
||||||
_, err = db.Exec("DROP TABLE datetimetest")
|
|
||||||
_, err = db.Exec(`
|
|
||||||
CREATE TABLE datetimetest
|
|
||||||
( entry_id INTEGER
|
|
||||||
, my_datetime DATETIME
|
|
||||||
)
|
|
||||||
`)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Failed to create table:", err)
|
|
||||||
}
|
|
||||||
datetime := "2006-01-02 15:04:05.003"
|
|
||||||
_, err = db.Exec(`
|
|
||||||
INSERT INTO datetimetest(entry_id, my_datetime)
|
|
||||||
VALUES(1, ?)`, datetime)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Failed to insert datetime:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
rows, err := db.Query(
|
|
||||||
"SELECT entry_id, my_datetime FROM datetimetest ORDER BY entry_id ASC")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Unable to query datetimetest table:", err)
|
|
||||||
}
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
if !rows.Next() {
|
|
||||||
if er := rows.Err(); er != nil {
|
|
||||||
t.Fatal(er)
|
|
||||||
} else {
|
|
||||||
t.Fatalf("Unable to extract row containing datetime")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var entryId int
|
|
||||||
var myDatetime time.Time
|
|
||||||
|
|
||||||
if err := rows.Scan(&entryId, &myDatetime); err != nil {
|
|
||||||
t.Error("Unable to scan results:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if entryId != 1 {
|
|
||||||
t.Errorf("EntryId does not match inserted value")
|
|
||||||
}
|
|
||||||
|
|
||||||
if myDatetime.Year() != 2006 ||
|
|
||||||
myDatetime.Month() != 1 ||
|
|
||||||
myDatetime.Day() != 2 ||
|
|
||||||
myDatetime.Hour() != 15 ||
|
|
||||||
myDatetime.Minute() != 4 ||
|
|
||||||
myDatetime.Second() != 5 ||
|
|
||||||
myDatetime.Nanosecond() != 3000000 {
|
|
||||||
t.Errorf("Extracted time does not match inserted value")
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue