|
| 1 | +// Copyright 2017 The Go Authors. All rights reserved. |
| 2 | +// Use of this source code is governed by a BSD-style |
| 3 | +// license that can be found in the LICENSE file. |
| 4 | + |
| 5 | +package odbc |
| 6 | + |
| 7 | +import ( |
| 8 | + "database/sql" |
| 9 | + "flag" |
| 10 | + "fmt" |
| 11 | + "testing" |
| 12 | + "time" |
| 13 | +) |
| 14 | + |
| 15 | +var ( |
| 16 | + mysrv = flag.String("mysrv", "server", "mysql server name") |
| 17 | + mydb = flag.String("mydb", "dbname", "mysql database name") |
| 18 | + myuser = flag.String("myuser", "", "mysql user name") |
| 19 | + mypass = flag.String("mypass", "", "mysql password") |
| 20 | +) |
| 21 | + |
| 22 | +func mysqlConnect() (db *sql.DB, stmtCount int, err error) { |
| 23 | + // from https://dev.mysql.com/doc/connector-odbc/en/connector-odbc-configuration-connection-parameters.html |
| 24 | + conn := fmt.Sprintf("driver=mysql;server=%s;database=%s;user=%s;password=%s;", |
| 25 | + *mysrv, *mydb, *myuser, *mypass) |
| 26 | + db, err = sql.Open("odbc", conn) |
| 27 | + if err != nil { |
| 28 | + return nil, 0, err |
| 29 | + } |
| 30 | + stats := db.Driver().(*Driver).Stats |
| 31 | + return db, stats.StmtCount, nil |
| 32 | +} |
| 33 | + |
| 34 | +func TestMYSQLTime(t *testing.T) { |
| 35 | + db, sc, err := mysqlConnect() |
| 36 | + if err != nil { |
| 37 | + t.Fatal(err) |
| 38 | + } |
| 39 | + defer closeDB(t, db, sc, sc) |
| 40 | + |
| 41 | + db.Exec("drop table temp") |
| 42 | + exec(t, db, "create table temp(id int not null auto_increment primary key, time time)") |
| 43 | + now := time.Now() |
| 44 | + // SQL_TIME_STRUCT only supports hours, minutes and seconds |
| 45 | + now = time.Date(1, time.January, 1, now.Hour(), now.Minute(), now.Second(), 0, time.Local) |
| 46 | + _, err = db.Exec("insert into temp (time) values(?)", now) |
| 47 | + if err != nil { |
| 48 | + t.Fatal(err) |
| 49 | + } |
| 50 | + |
| 51 | + var ret time.Time |
| 52 | + if err := db.QueryRow("select time from temp where id = ?", 1).Scan(&ret); err != nil { |
| 53 | + t.Fatal(err) |
| 54 | + } |
| 55 | + if ret != now { |
| 56 | + t.Fatalf("unexpected return value: want=%v, is=%v", now, ret) |
| 57 | + } |
| 58 | + |
| 59 | + exec(t, db, "drop table temp") |
| 60 | +} |
0 commit comments