driver_test.go 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614
  1. // Go MySQL Driver - A MySQL-Driver for Go's database/sql package
  2. //
  3. // Copyright 2013 The Go-MySQL-Driver Authors. All rights reserved.
  4. //
  5. // This Source Code Form is subject to the terms of the Mozilla Public
  6. // License, v. 2.0. If a copy of the MPL was not distributed with this file,
  7. // You can obtain one at http://mozilla.org/MPL/2.0/.
  8. package mysql
  9. import (
  10. "crypto/tls"
  11. "database/sql"
  12. "database/sql/driver"
  13. "fmt"
  14. "io"
  15. "io/ioutil"
  16. "net"
  17. "net/url"
  18. "os"
  19. "strings"
  20. "sync"
  21. "sync/atomic"
  22. "testing"
  23. "time"
  24. )
  25. var (
  26. user string
  27. pass string
  28. prot string
  29. addr string
  30. dbname string
  31. dsn string
  32. netAddr string
  33. available bool
  34. )
  35. var (
  36. tDate = time.Date(2012, 6, 14, 0, 0, 0, 0, time.UTC)
  37. sDate = "2012-06-14"
  38. tDateTime = time.Date(2011, 11, 20, 21, 27, 37, 0, time.UTC)
  39. sDateTime = "2011-11-20 21:27:37"
  40. tDate0 = time.Time{}
  41. sDate0 = "0000-00-00"
  42. sDateTime0 = "0000-00-00 00:00:00"
  43. )
  44. // See https://github.com/go-sql-driver/mysql/wiki/Testing
  45. func init() {
  46. // get environment variables
  47. env := func(key, defaultValue string) string {
  48. if value := os.Getenv(key); value != "" {
  49. return value
  50. }
  51. return defaultValue
  52. }
  53. user = env("MYSQL_TEST_USER", "root")
  54. pass = env("MYSQL_TEST_PASS", "")
  55. prot = env("MYSQL_TEST_PROT", "tcp")
  56. addr = env("MYSQL_TEST_ADDR", "localhost:3306")
  57. dbname = env("MYSQL_TEST_DBNAME", "gotest")
  58. netAddr = fmt.Sprintf("%s(%s)", prot, addr)
  59. dsn = fmt.Sprintf("%s:%s@%s/%s?timeout=30s&strict=true", user, pass, netAddr, dbname)
  60. c, err := net.Dial(prot, addr)
  61. if err == nil {
  62. available = true
  63. c.Close()
  64. }
  65. }
  66. type DBTest struct {
  67. *testing.T
  68. db *sql.DB
  69. }
  70. func runTests(t *testing.T, dsn string, tests ...func(dbt *DBTest)) {
  71. if !available {
  72. t.Skipf("MySQL-Server not running on %s", netAddr)
  73. }
  74. db, err := sql.Open("mysql", dsn)
  75. if err != nil {
  76. t.Fatalf("Error connecting: %s", err.Error())
  77. }
  78. defer db.Close()
  79. db.Exec("DROP TABLE IF EXISTS test")
  80. dsn2 := dsn + "&interpolateParams=true"
  81. var db2 *sql.DB
  82. if _, err := parseDSN(dsn2); err != errInvalidDSNUnsafeCollation {
  83. db2, err = sql.Open("mysql", dsn2)
  84. if err != nil {
  85. t.Fatalf("Error connecting: %s", err.Error())
  86. }
  87. defer db2.Close()
  88. }
  89. dbt := &DBTest{t, db}
  90. dbt2 := &DBTest{t, db2}
  91. for _, test := range tests {
  92. test(dbt)
  93. dbt.db.Exec("DROP TABLE IF EXISTS test")
  94. if db2 != nil {
  95. test(dbt2)
  96. dbt2.db.Exec("DROP TABLE IF EXISTS test")
  97. }
  98. }
  99. }
  100. func (dbt *DBTest) fail(method, query string, err error) {
  101. if len(query) > 300 {
  102. query = "[query too large to print]"
  103. }
  104. dbt.Fatalf("Error on %s %s: %s", method, query, err.Error())
  105. }
  106. func (dbt *DBTest) mustExec(query string, args ...interface{}) (res sql.Result) {
  107. res, err := dbt.db.Exec(query, args...)
  108. if err != nil {
  109. dbt.fail("Exec", query, err)
  110. }
  111. return res
  112. }
  113. func (dbt *DBTest) mustQuery(query string, args ...interface{}) (rows *sql.Rows) {
  114. rows, err := dbt.db.Query(query, args...)
  115. if err != nil {
  116. dbt.fail("Query", query, err)
  117. }
  118. return rows
  119. }
  120. func TestEmptyQuery(t *testing.T) {
  121. runTests(t, dsn, func(dbt *DBTest) {
  122. // just a comment, no query
  123. rows := dbt.mustQuery("--")
  124. // will hang before #255
  125. if rows.Next() {
  126. dbt.Errorf("Next on rows must be false")
  127. }
  128. })
  129. }
  130. func TestCRUD(t *testing.T) {
  131. runTests(t, dsn, func(dbt *DBTest) {
  132. // Create Table
  133. dbt.mustExec("CREATE TABLE test (value BOOL)")
  134. // Test for unexpected data
  135. var out bool
  136. rows := dbt.mustQuery("SELECT * FROM test")
  137. if rows.Next() {
  138. dbt.Error("unexpected data in empty table")
  139. }
  140. // Create Data
  141. res := dbt.mustExec("INSERT INTO test VALUES (1)")
  142. count, err := res.RowsAffected()
  143. if err != nil {
  144. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  145. }
  146. if count != 1 {
  147. dbt.Fatalf("Expected 1 affected row, got %d", count)
  148. }
  149. id, err := res.LastInsertId()
  150. if err != nil {
  151. dbt.Fatalf("res.LastInsertId() returned error: %s", err.Error())
  152. }
  153. if id != 0 {
  154. dbt.Fatalf("Expected InsertID 0, got %d", id)
  155. }
  156. // Read
  157. rows = dbt.mustQuery("SELECT value FROM test")
  158. if rows.Next() {
  159. rows.Scan(&out)
  160. if true != out {
  161. dbt.Errorf("true != %t", out)
  162. }
  163. if rows.Next() {
  164. dbt.Error("unexpected data")
  165. }
  166. } else {
  167. dbt.Error("no data")
  168. }
  169. // Update
  170. res = dbt.mustExec("UPDATE test SET value = ? WHERE value = ?", false, true)
  171. count, err = res.RowsAffected()
  172. if err != nil {
  173. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  174. }
  175. if count != 1 {
  176. dbt.Fatalf("Expected 1 affected row, got %d", count)
  177. }
  178. // Check Update
  179. rows = dbt.mustQuery("SELECT value FROM test")
  180. if rows.Next() {
  181. rows.Scan(&out)
  182. if false != out {
  183. dbt.Errorf("false != %t", out)
  184. }
  185. if rows.Next() {
  186. dbt.Error("unexpected data")
  187. }
  188. } else {
  189. dbt.Error("no data")
  190. }
  191. // Delete
  192. res = dbt.mustExec("DELETE FROM test WHERE value = ?", false)
  193. count, err = res.RowsAffected()
  194. if err != nil {
  195. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  196. }
  197. if count != 1 {
  198. dbt.Fatalf("Expected 1 affected row, got %d", count)
  199. }
  200. // Check for unexpected rows
  201. res = dbt.mustExec("DELETE FROM test")
  202. count, err = res.RowsAffected()
  203. if err != nil {
  204. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  205. }
  206. if count != 0 {
  207. dbt.Fatalf("Expected 0 affected row, got %d", count)
  208. }
  209. })
  210. }
  211. func TestInt(t *testing.T) {
  212. runTests(t, dsn, func(dbt *DBTest) {
  213. types := [5]string{"TINYINT", "SMALLINT", "MEDIUMINT", "INT", "BIGINT"}
  214. in := int64(42)
  215. var out int64
  216. var rows *sql.Rows
  217. // SIGNED
  218. for _, v := range types {
  219. dbt.mustExec("CREATE TABLE test (value " + v + ")")
  220. dbt.mustExec("INSERT INTO test VALUES (?)", in)
  221. rows = dbt.mustQuery("SELECT value FROM test")
  222. if rows.Next() {
  223. rows.Scan(&out)
  224. if in != out {
  225. dbt.Errorf("%s: %d != %d", v, in, out)
  226. }
  227. } else {
  228. dbt.Errorf("%s: no data", v)
  229. }
  230. dbt.mustExec("DROP TABLE IF EXISTS test")
  231. }
  232. // UNSIGNED ZEROFILL
  233. for _, v := range types {
  234. dbt.mustExec("CREATE TABLE test (value " + v + " ZEROFILL)")
  235. dbt.mustExec("INSERT INTO test VALUES (?)", in)
  236. rows = dbt.mustQuery("SELECT value FROM test")
  237. if rows.Next() {
  238. rows.Scan(&out)
  239. if in != out {
  240. dbt.Errorf("%s ZEROFILL: %d != %d", v, in, out)
  241. }
  242. } else {
  243. dbt.Errorf("%s ZEROFILL: no data", v)
  244. }
  245. dbt.mustExec("DROP TABLE IF EXISTS test")
  246. }
  247. })
  248. }
  249. func TestFloat(t *testing.T) {
  250. runTests(t, dsn, func(dbt *DBTest) {
  251. types := [2]string{"FLOAT", "DOUBLE"}
  252. in := float32(42.23)
  253. var out float32
  254. var rows *sql.Rows
  255. for _, v := range types {
  256. dbt.mustExec("CREATE TABLE test (value " + v + ")")
  257. dbt.mustExec("INSERT INTO test VALUES (?)", in)
  258. rows = dbt.mustQuery("SELECT value FROM test")
  259. if rows.Next() {
  260. rows.Scan(&out)
  261. if in != out {
  262. dbt.Errorf("%s: %g != %g", v, in, out)
  263. }
  264. } else {
  265. dbt.Errorf("%s: no data", v)
  266. }
  267. dbt.mustExec("DROP TABLE IF EXISTS test")
  268. }
  269. })
  270. }
  271. func TestString(t *testing.T) {
  272. runTests(t, dsn, func(dbt *DBTest) {
  273. types := [6]string{"CHAR(255)", "VARCHAR(255)", "TINYTEXT", "TEXT", "MEDIUMTEXT", "LONGTEXT"}
  274. in := "κόσμε üöäßñóùéàâÿœ'îë Árvíztűrő いろはにほへとちりぬるを イロハニホヘト דג סקרן чащах น่าฟังเอย"
  275. var out string
  276. var rows *sql.Rows
  277. for _, v := range types {
  278. dbt.mustExec("CREATE TABLE test (value " + v + ") CHARACTER SET utf8")
  279. dbt.mustExec("INSERT INTO test VALUES (?)", in)
  280. rows = dbt.mustQuery("SELECT value FROM test")
  281. if rows.Next() {
  282. rows.Scan(&out)
  283. if in != out {
  284. dbt.Errorf("%s: %s != %s", v, in, out)
  285. }
  286. } else {
  287. dbt.Errorf("%s: no data", v)
  288. }
  289. dbt.mustExec("DROP TABLE IF EXISTS test")
  290. }
  291. // BLOB
  292. dbt.mustExec("CREATE TABLE test (id int, value BLOB) CHARACTER SET utf8")
  293. id := 2
  294. in = "Lorem ipsum dolor sit amet, consetetur sadipscing elitr, " +
  295. "sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, " +
  296. "sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. " +
  297. "Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet. " +
  298. "Lorem ipsum dolor sit amet, consetetur sadipscing elitr, " +
  299. "sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, " +
  300. "sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. " +
  301. "Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet."
  302. dbt.mustExec("INSERT INTO test VALUES (?, ?)", id, in)
  303. err := dbt.db.QueryRow("SELECT value FROM test WHERE id = ?", id).Scan(&out)
  304. if err != nil {
  305. dbt.Fatalf("Error on BLOB-Query: %s", err.Error())
  306. } else if out != in {
  307. dbt.Errorf("BLOB: %s != %s", in, out)
  308. }
  309. })
  310. }
  311. type timeTests struct {
  312. dbtype string
  313. tlayout string
  314. tests []timeTest
  315. }
  316. type timeTest struct {
  317. s string // leading "!": do not use t as value in queries
  318. t time.Time
  319. }
  320. type timeMode byte
  321. func (t timeMode) String() string {
  322. switch t {
  323. case binaryString:
  324. return "binary:string"
  325. case binaryTime:
  326. return "binary:time.Time"
  327. case textString:
  328. return "text:string"
  329. }
  330. panic("unsupported timeMode")
  331. }
  332. func (t timeMode) Binary() bool {
  333. switch t {
  334. case binaryString, binaryTime:
  335. return true
  336. }
  337. return false
  338. }
  339. const (
  340. binaryString timeMode = iota
  341. binaryTime
  342. textString
  343. )
  344. func (t timeTest) genQuery(dbtype string, mode timeMode) string {
  345. var inner string
  346. if mode.Binary() {
  347. inner = "?"
  348. } else {
  349. inner = `"%s"`
  350. }
  351. return `SELECT cast(` + inner + ` as ` + dbtype + `)`
  352. }
  353. func (t timeTest) run(dbt *DBTest, dbtype, tlayout string, mode timeMode) {
  354. var rows *sql.Rows
  355. query := t.genQuery(dbtype, mode)
  356. switch mode {
  357. case binaryString:
  358. rows = dbt.mustQuery(query, t.s)
  359. case binaryTime:
  360. rows = dbt.mustQuery(query, t.t)
  361. case textString:
  362. query = fmt.Sprintf(query, t.s)
  363. rows = dbt.mustQuery(query)
  364. default:
  365. panic("unsupported mode")
  366. }
  367. defer rows.Close()
  368. var err error
  369. if !rows.Next() {
  370. err = rows.Err()
  371. if err == nil {
  372. err = fmt.Errorf("no data")
  373. }
  374. dbt.Errorf("%s [%s]: %s", dbtype, mode, err)
  375. return
  376. }
  377. var dst interface{}
  378. err = rows.Scan(&dst)
  379. if err != nil {
  380. dbt.Errorf("%s [%s]: %s", dbtype, mode, err)
  381. return
  382. }
  383. switch val := dst.(type) {
  384. case []uint8:
  385. str := string(val)
  386. if str == t.s {
  387. return
  388. }
  389. if mode.Binary() && dbtype == "DATETIME" && len(str) == 26 && str[:19] == t.s {
  390. // a fix mainly for TravisCI:
  391. // accept full microsecond resolution in result for DATETIME columns
  392. // where the binary protocol was used
  393. return
  394. }
  395. dbt.Errorf("%s [%s] to string: expected %q, got %q",
  396. dbtype, mode,
  397. t.s, str,
  398. )
  399. case time.Time:
  400. if val == t.t {
  401. return
  402. }
  403. dbt.Errorf("%s [%s] to string: expected %q, got %q",
  404. dbtype, mode,
  405. t.s, val.Format(tlayout),
  406. )
  407. default:
  408. fmt.Printf("%#v\n", []interface{}{dbtype, tlayout, mode, t.s, t.t})
  409. dbt.Errorf("%s [%s]: unhandled type %T (is '%v')",
  410. dbtype, mode,
  411. val, val,
  412. )
  413. }
  414. }
  415. func TestDateTime(t *testing.T) {
  416. afterTime := func(t time.Time, d string) time.Time {
  417. dur, err := time.ParseDuration(d)
  418. if err != nil {
  419. panic(err)
  420. }
  421. return t.Add(dur)
  422. }
  423. // NOTE: MySQL rounds DATETIME(x) up - but that's not included in the tests
  424. format := "2006-01-02 15:04:05.999999"
  425. t0 := time.Time{}
  426. tstr0 := "0000-00-00 00:00:00.000000"
  427. testcases := []timeTests{
  428. {"DATE", format[:10], []timeTest{
  429. {t: time.Date(2011, 11, 20, 0, 0, 0, 0, time.UTC)},
  430. {t: t0, s: tstr0[:10]},
  431. }},
  432. {"DATETIME", format[:19], []timeTest{
  433. {t: time.Date(2011, 11, 20, 21, 27, 37, 0, time.UTC)},
  434. {t: t0, s: tstr0[:19]},
  435. }},
  436. {"DATETIME(0)", format[:21], []timeTest{
  437. {t: time.Date(2011, 11, 20, 21, 27, 37, 0, time.UTC)},
  438. {t: t0, s: tstr0[:19]},
  439. }},
  440. {"DATETIME(1)", format[:21], []timeTest{
  441. {t: time.Date(2011, 11, 20, 21, 27, 37, 100000000, time.UTC)},
  442. {t: t0, s: tstr0[:21]},
  443. }},
  444. {"DATETIME(6)", format, []timeTest{
  445. {t: time.Date(2011, 11, 20, 21, 27, 37, 123456000, time.UTC)},
  446. {t: t0, s: tstr0},
  447. }},
  448. {"TIME", format[11:19], []timeTest{
  449. {t: afterTime(t0, "12345s")},
  450. {s: "!-12:34:56"},
  451. {s: "!-838:59:59"},
  452. {s: "!838:59:59"},
  453. {t: t0, s: tstr0[11:19]},
  454. }},
  455. {"TIME(0)", format[11:19], []timeTest{
  456. {t: afterTime(t0, "12345s")},
  457. {s: "!-12:34:56"},
  458. {s: "!-838:59:59"},
  459. {s: "!838:59:59"},
  460. {t: t0, s: tstr0[11:19]},
  461. }},
  462. {"TIME(1)", format[11:21], []timeTest{
  463. {t: afterTime(t0, "12345600ms")},
  464. {s: "!-12:34:56.7"},
  465. {s: "!-838:59:58.9"},
  466. {s: "!838:59:58.9"},
  467. {t: t0, s: tstr0[11:21]},
  468. }},
  469. {"TIME(6)", format[11:], []timeTest{
  470. {t: afterTime(t0, "1234567890123000ns")},
  471. {s: "!-12:34:56.789012"},
  472. {s: "!-838:59:58.999999"},
  473. {s: "!838:59:58.999999"},
  474. {t: t0, s: tstr0[11:]},
  475. }},
  476. }
  477. dsns := []string{
  478. dsn + "&parseTime=true",
  479. dsn + "&parseTime=false",
  480. }
  481. for _, testdsn := range dsns {
  482. runTests(t, testdsn, func(dbt *DBTest) {
  483. microsecsSupported := false
  484. zeroDateSupported := false
  485. var rows *sql.Rows
  486. var err error
  487. rows, err = dbt.db.Query(`SELECT cast("00:00:00.1" as TIME(1)) = "00:00:00.1"`)
  488. if err == nil {
  489. rows.Scan(&microsecsSupported)
  490. rows.Close()
  491. }
  492. rows, err = dbt.db.Query(`SELECT cast("0000-00-00" as DATE) = "0000-00-00"`)
  493. if err == nil {
  494. rows.Scan(&zeroDateSupported)
  495. rows.Close()
  496. }
  497. for _, setups := range testcases {
  498. if t := setups.dbtype; !microsecsSupported && t[len(t)-1:] == ")" {
  499. // skip fractional second tests if unsupported by server
  500. continue
  501. }
  502. for _, setup := range setups.tests {
  503. allowBinTime := true
  504. if setup.s == "" {
  505. // fill time string whereever Go can reliable produce it
  506. setup.s = setup.t.Format(setups.tlayout)
  507. } else if setup.s[0] == '!' {
  508. // skip tests using setup.t as source in queries
  509. allowBinTime = false
  510. // fix setup.s - remove the "!"
  511. setup.s = setup.s[1:]
  512. }
  513. if !zeroDateSupported && setup.s == tstr0[:len(setup.s)] {
  514. // skip disallowed 0000-00-00 date
  515. continue
  516. }
  517. setup.run(dbt, setups.dbtype, setups.tlayout, textString)
  518. setup.run(dbt, setups.dbtype, setups.tlayout, binaryString)
  519. if allowBinTime {
  520. setup.run(dbt, setups.dbtype, setups.tlayout, binaryTime)
  521. }
  522. }
  523. }
  524. })
  525. }
  526. }
  527. func TestTimestampMicros(t *testing.T) {
  528. format := "2006-01-02 15:04:05.999999"
  529. f0 := format[:19]
  530. f1 := format[:21]
  531. f6 := format[:26]
  532. runTests(t, dsn, func(dbt *DBTest) {
  533. // check if microseconds are supported.
  534. // Do not use timestamp(x) for that check - before 5.5.6, x would mean display width
  535. // and not precision.
  536. // Se last paragraph at http://dev.mysql.com/doc/refman/5.6/en/fractional-seconds.html
  537. microsecsSupported := false
  538. if rows, err := dbt.db.Query(`SELECT cast("00:00:00.1" as TIME(1)) = "00:00:00.1"`); err == nil {
  539. rows.Scan(&microsecsSupported)
  540. rows.Close()
  541. }
  542. if !microsecsSupported {
  543. // skip test
  544. return
  545. }
  546. _, err := dbt.db.Exec(`
  547. CREATE TABLE test (
  548. value0 TIMESTAMP NOT NULL DEFAULT '` + f0 + `',
  549. value1 TIMESTAMP(1) NOT NULL DEFAULT '` + f1 + `',
  550. value6 TIMESTAMP(6) NOT NULL DEFAULT '` + f6 + `'
  551. )`,
  552. )
  553. if err != nil {
  554. dbt.Error(err)
  555. }
  556. defer dbt.mustExec("DROP TABLE IF EXISTS test")
  557. dbt.mustExec("INSERT INTO test SET value0=?, value1=?, value6=?", f0, f1, f6)
  558. var res0, res1, res6 string
  559. rows := dbt.mustQuery("SELECT * FROM test")
  560. if !rows.Next() {
  561. dbt.Errorf("test contained no selectable values")
  562. }
  563. err = rows.Scan(&res0, &res1, &res6)
  564. if err != nil {
  565. dbt.Error(err)
  566. }
  567. if res0 != f0 {
  568. dbt.Errorf("expected %q, got %q", f0, res0)
  569. }
  570. if res1 != f1 {
  571. dbt.Errorf("expected %q, got %q", f1, res1)
  572. }
  573. if res6 != f6 {
  574. dbt.Errorf("expected %q, got %q", f6, res6)
  575. }
  576. })
  577. }
  578. func TestNULL(t *testing.T) {
  579. runTests(t, dsn, func(dbt *DBTest) {
  580. nullStmt, err := dbt.db.Prepare("SELECT NULL")
  581. if err != nil {
  582. dbt.Fatal(err)
  583. }
  584. defer nullStmt.Close()
  585. nonNullStmt, err := dbt.db.Prepare("SELECT 1")
  586. if err != nil {
  587. dbt.Fatal(err)
  588. }
  589. defer nonNullStmt.Close()
  590. // NullBool
  591. var nb sql.NullBool
  592. // Invalid
  593. if err = nullStmt.QueryRow().Scan(&nb); err != nil {
  594. dbt.Fatal(err)
  595. }
  596. if nb.Valid {
  597. dbt.Error("Valid NullBool which should be invalid")
  598. }
  599. // Valid
  600. if err = nonNullStmt.QueryRow().Scan(&nb); err != nil {
  601. dbt.Fatal(err)
  602. }
  603. if !nb.Valid {
  604. dbt.Error("Invalid NullBool which should be valid")
  605. } else if nb.Bool != true {
  606. dbt.Errorf("Unexpected NullBool value: %t (should be true)", nb.Bool)
  607. }
  608. // NullFloat64
  609. var nf sql.NullFloat64
  610. // Invalid
  611. if err = nullStmt.QueryRow().Scan(&nf); err != nil {
  612. dbt.Fatal(err)
  613. }
  614. if nf.Valid {
  615. dbt.Error("Valid NullFloat64 which should be invalid")
  616. }
  617. // Valid
  618. if err = nonNullStmt.QueryRow().Scan(&nf); err != nil {
  619. dbt.Fatal(err)
  620. }
  621. if !nf.Valid {
  622. dbt.Error("Invalid NullFloat64 which should be valid")
  623. } else if nf.Float64 != float64(1) {
  624. dbt.Errorf("Unexpected NullFloat64 value: %f (should be 1.0)", nf.Float64)
  625. }
  626. // NullInt64
  627. var ni sql.NullInt64
  628. // Invalid
  629. if err = nullStmt.QueryRow().Scan(&ni); err != nil {
  630. dbt.Fatal(err)
  631. }
  632. if ni.Valid {
  633. dbt.Error("Valid NullInt64 which should be invalid")
  634. }
  635. // Valid
  636. if err = nonNullStmt.QueryRow().Scan(&ni); err != nil {
  637. dbt.Fatal(err)
  638. }
  639. if !ni.Valid {
  640. dbt.Error("Invalid NullInt64 which should be valid")
  641. } else if ni.Int64 != int64(1) {
  642. dbt.Errorf("Unexpected NullInt64 value: %d (should be 1)", ni.Int64)
  643. }
  644. // NullString
  645. var ns sql.NullString
  646. // Invalid
  647. if err = nullStmt.QueryRow().Scan(&ns); err != nil {
  648. dbt.Fatal(err)
  649. }
  650. if ns.Valid {
  651. dbt.Error("Valid NullString which should be invalid")
  652. }
  653. // Valid
  654. if err = nonNullStmt.QueryRow().Scan(&ns); err != nil {
  655. dbt.Fatal(err)
  656. }
  657. if !ns.Valid {
  658. dbt.Error("Invalid NullString which should be valid")
  659. } else if ns.String != `1` {
  660. dbt.Error("Unexpected NullString value:" + ns.String + " (should be `1`)")
  661. }
  662. // nil-bytes
  663. var b []byte
  664. // Read nil
  665. if err = nullStmt.QueryRow().Scan(&b); err != nil {
  666. dbt.Fatal(err)
  667. }
  668. if b != nil {
  669. dbt.Error("Non-nil []byte wich should be nil")
  670. }
  671. // Read non-nil
  672. if err = nonNullStmt.QueryRow().Scan(&b); err != nil {
  673. dbt.Fatal(err)
  674. }
  675. if b == nil {
  676. dbt.Error("Nil []byte wich should be non-nil")
  677. }
  678. // Insert nil
  679. b = nil
  680. success := false
  681. if err = dbt.db.QueryRow("SELECT ? IS NULL", b).Scan(&success); err != nil {
  682. dbt.Fatal(err)
  683. }
  684. if !success {
  685. dbt.Error("Inserting []byte(nil) as NULL failed")
  686. }
  687. // Check input==output with input==nil
  688. b = nil
  689. if err = dbt.db.QueryRow("SELECT ?", b).Scan(&b); err != nil {
  690. dbt.Fatal(err)
  691. }
  692. if b != nil {
  693. dbt.Error("Non-nil echo from nil input")
  694. }
  695. // Check input==output with input!=nil
  696. b = []byte("")
  697. if err = dbt.db.QueryRow("SELECT ?", b).Scan(&b); err != nil {
  698. dbt.Fatal(err)
  699. }
  700. if b == nil {
  701. dbt.Error("nil echo from non-nil input")
  702. }
  703. // Insert NULL
  704. dbt.mustExec("CREATE TABLE test (dummmy1 int, value int, dummy2 int)")
  705. dbt.mustExec("INSERT INTO test VALUES (?, ?, ?)", 1, nil, 2)
  706. var out interface{}
  707. rows := dbt.mustQuery("SELECT * FROM test")
  708. if rows.Next() {
  709. rows.Scan(&out)
  710. if out != nil {
  711. dbt.Errorf("%v != nil", out)
  712. }
  713. } else {
  714. dbt.Error("no data")
  715. }
  716. })
  717. }
  718. func TestLongData(t *testing.T) {
  719. runTests(t, dsn, func(dbt *DBTest) {
  720. var maxAllowedPacketSize int
  721. err := dbt.db.QueryRow("select @@max_allowed_packet").Scan(&maxAllowedPacketSize)
  722. if err != nil {
  723. dbt.Fatal(err)
  724. }
  725. maxAllowedPacketSize--
  726. // don't get too ambitious
  727. if maxAllowedPacketSize > 1<<25 {
  728. maxAllowedPacketSize = 1 << 25
  729. }
  730. dbt.mustExec("CREATE TABLE test (value LONGBLOB)")
  731. in := strings.Repeat(`a`, maxAllowedPacketSize+1)
  732. var out string
  733. var rows *sql.Rows
  734. // Long text data
  735. const nonDataQueryLen = 28 // length query w/o value
  736. inS := in[:maxAllowedPacketSize-nonDataQueryLen]
  737. dbt.mustExec("INSERT INTO test VALUES('" + inS + "')")
  738. rows = dbt.mustQuery("SELECT value FROM test")
  739. if rows.Next() {
  740. rows.Scan(&out)
  741. if inS != out {
  742. dbt.Fatalf("LONGBLOB: length in: %d, length out: %d", len(inS), len(out))
  743. }
  744. if rows.Next() {
  745. dbt.Error("LONGBLOB: unexpexted row")
  746. }
  747. } else {
  748. dbt.Fatalf("LONGBLOB: no data")
  749. }
  750. // Empty table
  751. dbt.mustExec("TRUNCATE TABLE test")
  752. // Long binary data
  753. dbt.mustExec("INSERT INTO test VALUES(?)", in)
  754. rows = dbt.mustQuery("SELECT value FROM test WHERE 1=?", 1)
  755. if rows.Next() {
  756. rows.Scan(&out)
  757. if in != out {
  758. dbt.Fatalf("LONGBLOB: length in: %d, length out: %d", len(in), len(out))
  759. }
  760. if rows.Next() {
  761. dbt.Error("LONGBLOB: unexpexted row")
  762. }
  763. } else {
  764. if err = rows.Err(); err != nil {
  765. dbt.Fatalf("LONGBLOB: no data (err: %s)", err.Error())
  766. } else {
  767. dbt.Fatal("LONGBLOB: no data (err: <nil>)")
  768. }
  769. }
  770. })
  771. }
  772. func TestLoadData(t *testing.T) {
  773. runTests(t, dsn, func(dbt *DBTest) {
  774. verifyLoadDataResult := func() {
  775. rows, err := dbt.db.Query("SELECT * FROM test")
  776. if err != nil {
  777. dbt.Fatal(err.Error())
  778. }
  779. i := 0
  780. values := [4]string{
  781. "a string",
  782. "a string containing a \t",
  783. "a string containing a \n",
  784. "a string containing both \t\n",
  785. }
  786. var id int
  787. var value string
  788. for rows.Next() {
  789. i++
  790. err = rows.Scan(&id, &value)
  791. if err != nil {
  792. dbt.Fatal(err.Error())
  793. }
  794. if i != id {
  795. dbt.Fatalf("%d != %d", i, id)
  796. }
  797. if values[i-1] != value {
  798. dbt.Fatalf("%q != %q", values[i-1], value)
  799. }
  800. }
  801. err = rows.Err()
  802. if err != nil {
  803. dbt.Fatal(err.Error())
  804. }
  805. if i != 4 {
  806. dbt.Fatalf("Rows count mismatch. Got %d, want 4", i)
  807. }
  808. }
  809. file, err := ioutil.TempFile("", "gotest")
  810. defer os.Remove(file.Name())
  811. if err != nil {
  812. dbt.Fatal(err)
  813. }
  814. file.WriteString("1\ta string\n2\ta string containing a \\t\n3\ta string containing a \\n\n4\ta string containing both \\t\\n\n")
  815. file.Close()
  816. dbt.db.Exec("DROP TABLE IF EXISTS test")
  817. dbt.mustExec("CREATE TABLE test (id INT NOT NULL PRIMARY KEY, value TEXT NOT NULL) CHARACTER SET utf8")
  818. // Local File
  819. RegisterLocalFile(file.Name())
  820. dbt.mustExec(fmt.Sprintf("LOAD DATA LOCAL INFILE %q INTO TABLE test", file.Name()))
  821. verifyLoadDataResult()
  822. // negative test
  823. _, err = dbt.db.Exec("LOAD DATA LOCAL INFILE 'doesnotexist' INTO TABLE test")
  824. if err == nil {
  825. dbt.Fatal("Load non-existent file didn't fail")
  826. } else if err.Error() != "Local File 'doesnotexist' is not registered. Use the DSN parameter 'allowAllFiles=true' to allow all files" {
  827. dbt.Fatal(err.Error())
  828. }
  829. // Empty table
  830. dbt.mustExec("TRUNCATE TABLE test")
  831. // Reader
  832. RegisterReaderHandler("test", func() io.Reader {
  833. file, err = os.Open(file.Name())
  834. if err != nil {
  835. dbt.Fatal(err)
  836. }
  837. return file
  838. })
  839. dbt.mustExec("LOAD DATA LOCAL INFILE 'Reader::test' INTO TABLE test")
  840. verifyLoadDataResult()
  841. // negative test
  842. _, err = dbt.db.Exec("LOAD DATA LOCAL INFILE 'Reader::doesnotexist' INTO TABLE test")
  843. if err == nil {
  844. dbt.Fatal("Load non-existent Reader didn't fail")
  845. } else if err.Error() != "Reader 'doesnotexist' is not registered" {
  846. dbt.Fatal(err.Error())
  847. }
  848. })
  849. }
  850. func TestFoundRows(t *testing.T) {
  851. runTests(t, dsn, func(dbt *DBTest) {
  852. dbt.mustExec("CREATE TABLE test (id INT NOT NULL ,data INT NOT NULL)")
  853. dbt.mustExec("INSERT INTO test (id, data) VALUES (0, 0),(0, 0),(1, 0),(1, 0),(1, 1)")
  854. res := dbt.mustExec("UPDATE test SET data = 1 WHERE id = 0")
  855. count, err := res.RowsAffected()
  856. if err != nil {
  857. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  858. }
  859. if count != 2 {
  860. dbt.Fatalf("Expected 2 affected rows, got %d", count)
  861. }
  862. res = dbt.mustExec("UPDATE test SET data = 1 WHERE id = 1")
  863. count, err = res.RowsAffected()
  864. if err != nil {
  865. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  866. }
  867. if count != 2 {
  868. dbt.Fatalf("Expected 2 affected rows, got %d", count)
  869. }
  870. })
  871. runTests(t, dsn+"&clientFoundRows=true", func(dbt *DBTest) {
  872. dbt.mustExec("CREATE TABLE test (id INT NOT NULL ,data INT NOT NULL)")
  873. dbt.mustExec("INSERT INTO test (id, data) VALUES (0, 0),(0, 0),(1, 0),(1, 0),(1, 1)")
  874. res := dbt.mustExec("UPDATE test SET data = 1 WHERE id = 0")
  875. count, err := res.RowsAffected()
  876. if err != nil {
  877. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  878. }
  879. if count != 2 {
  880. dbt.Fatalf("Expected 2 matched rows, got %d", count)
  881. }
  882. res = dbt.mustExec("UPDATE test SET data = 1 WHERE id = 1")
  883. count, err = res.RowsAffected()
  884. if err != nil {
  885. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  886. }
  887. if count != 3 {
  888. dbt.Fatalf("Expected 3 matched rows, got %d", count)
  889. }
  890. })
  891. }
  892. func TestStrict(t *testing.T) {
  893. // ALLOW_INVALID_DATES to get rid of stricter modes - we want to test for warnings, not errors
  894. relaxedDsn := dsn + "&sql_mode=ALLOW_INVALID_DATES"
  895. // make sure the MySQL version is recent enough with a separate connection
  896. // before running the test
  897. conn, err := MySQLDriver{}.Open(relaxedDsn)
  898. if conn != nil {
  899. conn.Close()
  900. }
  901. if me, ok := err.(*MySQLError); ok && me.Number == 1231 {
  902. // Error 1231: Variable 'sql_mode' can't be set to the value of 'ALLOW_INVALID_DATES'
  903. // => skip test, MySQL server version is too old
  904. return
  905. }
  906. runTests(t, relaxedDsn, func(dbt *DBTest) {
  907. dbt.mustExec("CREATE TABLE test (a TINYINT NOT NULL, b CHAR(4))")
  908. var queries = [...]struct {
  909. in string
  910. codes []string
  911. }{
  912. {"DROP TABLE IF EXISTS no_such_table", []string{"1051"}},
  913. {"INSERT INTO test VALUES(10,'mysql'),(NULL,'test'),(300,'Open Source')", []string{"1265", "1048", "1264", "1265"}},
  914. }
  915. var err error
  916. var checkWarnings = func(err error, mode string, idx int) {
  917. if err == nil {
  918. dbt.Errorf("Expected STRICT error on query [%s] %s", mode, queries[idx].in)
  919. }
  920. if warnings, ok := err.(MySQLWarnings); ok {
  921. var codes = make([]string, len(warnings))
  922. for i := range warnings {
  923. codes[i] = warnings[i].Code
  924. }
  925. if len(codes) != len(queries[idx].codes) {
  926. dbt.Errorf("Unexpected STRICT error count on query [%s] %s: Wanted %v, Got %v", mode, queries[idx].in, queries[idx].codes, codes)
  927. }
  928. for i := range warnings {
  929. if codes[i] != queries[idx].codes[i] {
  930. dbt.Errorf("Unexpected STRICT error codes on query [%s] %s: Wanted %v, Got %v", mode, queries[idx].in, queries[idx].codes, codes)
  931. return
  932. }
  933. }
  934. } else {
  935. dbt.Errorf("Unexpected error on query [%s] %s: %s", mode, queries[idx].in, err.Error())
  936. }
  937. }
  938. // text protocol
  939. for i := range queries {
  940. _, err = dbt.db.Exec(queries[i].in)
  941. checkWarnings(err, "text", i)
  942. }
  943. var stmt *sql.Stmt
  944. // binary protocol
  945. for i := range queries {
  946. stmt, err = dbt.db.Prepare(queries[i].in)
  947. if err != nil {
  948. dbt.Errorf("Error on preparing query %s: %s", queries[i].in, err.Error())
  949. }
  950. _, err = stmt.Exec()
  951. checkWarnings(err, "binary", i)
  952. err = stmt.Close()
  953. if err != nil {
  954. dbt.Errorf("Error on closing stmt for query %s: %s", queries[i].in, err.Error())
  955. }
  956. }
  957. })
  958. }
  959. func TestTLS(t *testing.T) {
  960. tlsTest := func(dbt *DBTest) {
  961. if err := dbt.db.Ping(); err != nil {
  962. if err == ErrNoTLS {
  963. dbt.Skip("Server does not support TLS")
  964. } else {
  965. dbt.Fatalf("Error on Ping: %s", err.Error())
  966. }
  967. }
  968. rows := dbt.mustQuery("SHOW STATUS LIKE 'Ssl_cipher'")
  969. var variable, value *sql.RawBytes
  970. for rows.Next() {
  971. if err := rows.Scan(&variable, &value); err != nil {
  972. dbt.Fatal(err.Error())
  973. }
  974. if value == nil {
  975. dbt.Fatal("No Cipher")
  976. }
  977. }
  978. }
  979. runTests(t, dsn+"&tls=skip-verify", tlsTest)
  980. // Verify that registering / using a custom cfg works
  981. RegisterTLSConfig("custom-skip-verify", &tls.Config{
  982. InsecureSkipVerify: true,
  983. })
  984. runTests(t, dsn+"&tls=custom-skip-verify", tlsTest)
  985. }
  986. func TestReuseClosedConnection(t *testing.T) {
  987. // this test does not use sql.database, it uses the driver directly
  988. if !available {
  989. t.Skipf("MySQL-Server not running on %s", netAddr)
  990. }
  991. md := &MySQLDriver{}
  992. conn, err := md.Open(dsn)
  993. if err != nil {
  994. t.Fatalf("Error connecting: %s", err.Error())
  995. }
  996. stmt, err := conn.Prepare("DO 1")
  997. if err != nil {
  998. t.Fatalf("Error preparing statement: %s", err.Error())
  999. }
  1000. _, err = stmt.Exec(nil)
  1001. if err != nil {
  1002. t.Fatalf("Error executing statement: %s", err.Error())
  1003. }
  1004. err = conn.Close()
  1005. if err != nil {
  1006. t.Fatalf("Error closing connection: %s", err.Error())
  1007. }
  1008. defer func() {
  1009. if err := recover(); err != nil {
  1010. t.Errorf("Panic after reusing a closed connection: %v", err)
  1011. }
  1012. }()
  1013. _, err = stmt.Exec(nil)
  1014. if err != nil && err != driver.ErrBadConn {
  1015. t.Errorf("Unexpected error '%s', expected '%s'",
  1016. err.Error(), driver.ErrBadConn.Error())
  1017. }
  1018. }
  1019. func TestCharset(t *testing.T) {
  1020. if !available {
  1021. t.Skipf("MySQL-Server not running on %s", netAddr)
  1022. }
  1023. mustSetCharset := func(charsetParam, expected string) {
  1024. runTests(t, dsn+"&"+charsetParam, func(dbt *DBTest) {
  1025. rows := dbt.mustQuery("SELECT @@character_set_connection")
  1026. defer rows.Close()
  1027. if !rows.Next() {
  1028. dbt.Fatalf("Error getting connection charset: %s", rows.Err())
  1029. }
  1030. var got string
  1031. rows.Scan(&got)
  1032. if got != expected {
  1033. dbt.Fatalf("Expected connection charset %s but got %s", expected, got)
  1034. }
  1035. })
  1036. }
  1037. // non utf8 test
  1038. mustSetCharset("charset=ascii", "ascii")
  1039. // when the first charset is invalid, use the second
  1040. mustSetCharset("charset=none,utf8", "utf8")
  1041. // when the first charset is valid, use it
  1042. mustSetCharset("charset=ascii,utf8", "ascii")
  1043. mustSetCharset("charset=utf8,ascii", "utf8")
  1044. }
  1045. func TestFailingCharset(t *testing.T) {
  1046. runTests(t, dsn+"&charset=none", func(dbt *DBTest) {
  1047. // run query to really establish connection...
  1048. _, err := dbt.db.Exec("SELECT 1")
  1049. if err == nil {
  1050. dbt.db.Close()
  1051. t.Fatalf("Connection must not succeed without a valid charset")
  1052. }
  1053. })
  1054. }
  1055. func TestCollation(t *testing.T) {
  1056. if !available {
  1057. t.Skipf("MySQL-Server not running on %s", netAddr)
  1058. }
  1059. defaultCollation := "utf8_general_ci"
  1060. testCollations := []string{
  1061. "", // do not set
  1062. defaultCollation, // driver default
  1063. "latin1_general_ci",
  1064. "binary",
  1065. "utf8_unicode_ci",
  1066. "cp1257_bin",
  1067. }
  1068. for _, collation := range testCollations {
  1069. var expected, tdsn string
  1070. if collation != "" {
  1071. tdsn = dsn + "&collation=" + collation
  1072. expected = collation
  1073. } else {
  1074. tdsn = dsn
  1075. expected = defaultCollation
  1076. }
  1077. runTests(t, tdsn, func(dbt *DBTest) {
  1078. var got string
  1079. if err := dbt.db.QueryRow("SELECT @@collation_connection").Scan(&got); err != nil {
  1080. dbt.Fatal(err)
  1081. }
  1082. if got != expected {
  1083. dbt.Fatalf("Expected connection collation %s but got %s", expected, got)
  1084. }
  1085. })
  1086. }
  1087. }
  1088. func TestRawBytesResultExceedsBuffer(t *testing.T) {
  1089. runTests(t, dsn, func(dbt *DBTest) {
  1090. // defaultBufSize from buffer.go
  1091. expected := strings.Repeat("abc", defaultBufSize)
  1092. rows := dbt.mustQuery("SELECT '" + expected + "'")
  1093. defer rows.Close()
  1094. if !rows.Next() {
  1095. dbt.Error("expected result, got none")
  1096. }
  1097. var result sql.RawBytes
  1098. rows.Scan(&result)
  1099. if expected != string(result) {
  1100. dbt.Error("result did not match expected value")
  1101. }
  1102. })
  1103. }
  1104. func TestTimezoneConversion(t *testing.T) {
  1105. zones := []string{"UTC", "US/Central", "US/Pacific", "Local"}
  1106. // Regression test for timezone handling
  1107. tzTest := func(dbt *DBTest) {
  1108. // Create table
  1109. dbt.mustExec("CREATE TABLE test (ts TIMESTAMP)")
  1110. // Insert local time into database (should be converted)
  1111. usCentral, _ := time.LoadLocation("US/Central")
  1112. reftime := time.Date(2014, 05, 30, 18, 03, 17, 0, time.UTC).In(usCentral)
  1113. dbt.mustExec("INSERT INTO test VALUE (?)", reftime)
  1114. // Retrieve time from DB
  1115. rows := dbt.mustQuery("SELECT ts FROM test")
  1116. if !rows.Next() {
  1117. dbt.Fatal("Didn't get any rows out")
  1118. }
  1119. var dbTime time.Time
  1120. err := rows.Scan(&dbTime)
  1121. if err != nil {
  1122. dbt.Fatal("Err", err)
  1123. }
  1124. // Check that dates match
  1125. if reftime.Unix() != dbTime.Unix() {
  1126. dbt.Errorf("Times don't match.\n")
  1127. dbt.Errorf(" Now(%v)=%v\n", usCentral, reftime)
  1128. dbt.Errorf(" Now(UTC)=%v\n", dbTime)
  1129. }
  1130. }
  1131. for _, tz := range zones {
  1132. runTests(t, dsn+"&parseTime=true&loc="+url.QueryEscape(tz), tzTest)
  1133. }
  1134. }
  1135. // Special cases
  1136. func TestRowsClose(t *testing.T) {
  1137. runTests(t, dsn, func(dbt *DBTest) {
  1138. rows, err := dbt.db.Query("SELECT 1")
  1139. if err != nil {
  1140. dbt.Fatal(err)
  1141. }
  1142. err = rows.Close()
  1143. if err != nil {
  1144. dbt.Fatal(err)
  1145. }
  1146. if rows.Next() {
  1147. dbt.Fatal("Unexpected row after rows.Close()")
  1148. }
  1149. err = rows.Err()
  1150. if err != nil {
  1151. dbt.Fatal(err)
  1152. }
  1153. })
  1154. }
  1155. // dangling statements
  1156. // http://code.google.com/p/go/issues/detail?id=3865
  1157. func TestCloseStmtBeforeRows(t *testing.T) {
  1158. runTests(t, dsn, func(dbt *DBTest) {
  1159. stmt, err := dbt.db.Prepare("SELECT 1")
  1160. if err != nil {
  1161. dbt.Fatal(err)
  1162. }
  1163. rows, err := stmt.Query()
  1164. if err != nil {
  1165. stmt.Close()
  1166. dbt.Fatal(err)
  1167. }
  1168. defer rows.Close()
  1169. err = stmt.Close()
  1170. if err != nil {
  1171. dbt.Fatal(err)
  1172. }
  1173. if !rows.Next() {
  1174. dbt.Fatal("Getting row failed")
  1175. } else {
  1176. err = rows.Err()
  1177. if err != nil {
  1178. dbt.Fatal(err)
  1179. }
  1180. var out bool
  1181. err = rows.Scan(&out)
  1182. if err != nil {
  1183. dbt.Fatalf("Error on rows.Scan(): %s", err.Error())
  1184. }
  1185. if out != true {
  1186. dbt.Errorf("true != %t", out)
  1187. }
  1188. }
  1189. })
  1190. }
  1191. // It is valid to have multiple Rows for the same Stmt
  1192. // http://code.google.com/p/go/issues/detail?id=3734
  1193. func TestStmtMultiRows(t *testing.T) {
  1194. runTests(t, dsn, func(dbt *DBTest) {
  1195. stmt, err := dbt.db.Prepare("SELECT 1 UNION SELECT 0")
  1196. if err != nil {
  1197. dbt.Fatal(err)
  1198. }
  1199. rows1, err := stmt.Query()
  1200. if err != nil {
  1201. stmt.Close()
  1202. dbt.Fatal(err)
  1203. }
  1204. defer rows1.Close()
  1205. rows2, err := stmt.Query()
  1206. if err != nil {
  1207. stmt.Close()
  1208. dbt.Fatal(err)
  1209. }
  1210. defer rows2.Close()
  1211. var out bool
  1212. // 1
  1213. if !rows1.Next() {
  1214. dbt.Fatal("1st rows1.Next failed")
  1215. } else {
  1216. err = rows1.Err()
  1217. if err != nil {
  1218. dbt.Fatal(err)
  1219. }
  1220. err = rows1.Scan(&out)
  1221. if err != nil {
  1222. dbt.Fatalf("Error on rows.Scan(): %s", err.Error())
  1223. }
  1224. if out != true {
  1225. dbt.Errorf("true != %t", out)
  1226. }
  1227. }
  1228. if !rows2.Next() {
  1229. dbt.Fatal("1st rows2.Next failed")
  1230. } else {
  1231. err = rows2.Err()
  1232. if err != nil {
  1233. dbt.Fatal(err)
  1234. }
  1235. err = rows2.Scan(&out)
  1236. if err != nil {
  1237. dbt.Fatalf("Error on rows.Scan(): %s", err.Error())
  1238. }
  1239. if out != true {
  1240. dbt.Errorf("true != %t", out)
  1241. }
  1242. }
  1243. // 2
  1244. if !rows1.Next() {
  1245. dbt.Fatal("2nd rows1.Next failed")
  1246. } else {
  1247. err = rows1.Err()
  1248. if err != nil {
  1249. dbt.Fatal(err)
  1250. }
  1251. err = rows1.Scan(&out)
  1252. if err != nil {
  1253. dbt.Fatalf("Error on rows.Scan(): %s", err.Error())
  1254. }
  1255. if out != false {
  1256. dbt.Errorf("false != %t", out)
  1257. }
  1258. if rows1.Next() {
  1259. dbt.Fatal("Unexpected row on rows1")
  1260. }
  1261. err = rows1.Close()
  1262. if err != nil {
  1263. dbt.Fatal(err)
  1264. }
  1265. }
  1266. if !rows2.Next() {
  1267. dbt.Fatal("2nd rows2.Next failed")
  1268. } else {
  1269. err = rows2.Err()
  1270. if err != nil {
  1271. dbt.Fatal(err)
  1272. }
  1273. err = rows2.Scan(&out)
  1274. if err != nil {
  1275. dbt.Fatalf("Error on rows.Scan(): %s", err.Error())
  1276. }
  1277. if out != false {
  1278. dbt.Errorf("false != %t", out)
  1279. }
  1280. if rows2.Next() {
  1281. dbt.Fatal("Unexpected row on rows2")
  1282. }
  1283. err = rows2.Close()
  1284. if err != nil {
  1285. dbt.Fatal(err)
  1286. }
  1287. }
  1288. })
  1289. }
  1290. // Regression test for
  1291. // * more than 32 NULL parameters (issue 209)
  1292. // * more parameters than fit into the buffer (issue 201)
  1293. func TestPreparedManyCols(t *testing.T) {
  1294. const numParams = defaultBufSize
  1295. runTests(t, dsn, func(dbt *DBTest) {
  1296. query := "SELECT ?" + strings.Repeat(",?", numParams-1)
  1297. stmt, err := dbt.db.Prepare(query)
  1298. if err != nil {
  1299. dbt.Fatal(err)
  1300. }
  1301. defer stmt.Close()
  1302. // create more parameters than fit into the buffer
  1303. // which will take nil-values
  1304. params := make([]interface{}, numParams)
  1305. rows, err := stmt.Query(params...)
  1306. if err != nil {
  1307. stmt.Close()
  1308. dbt.Fatal(err)
  1309. }
  1310. defer rows.Close()
  1311. })
  1312. }
  1313. func TestConcurrent(t *testing.T) {
  1314. if enabled, _ := readBool(os.Getenv("MYSQL_TEST_CONCURRENT")); !enabled {
  1315. t.Skip("MYSQL_TEST_CONCURRENT env var not set")
  1316. }
  1317. runTests(t, dsn, func(dbt *DBTest) {
  1318. var max int
  1319. err := dbt.db.QueryRow("SELECT @@max_connections").Scan(&max)
  1320. if err != nil {
  1321. dbt.Fatalf("%s", err.Error())
  1322. }
  1323. dbt.Logf("Testing up to %d concurrent connections \r\n", max)
  1324. var remaining, succeeded int32 = int32(max), 0
  1325. var wg sync.WaitGroup
  1326. wg.Add(max)
  1327. var fatalError string
  1328. var once sync.Once
  1329. fatalf := func(s string, vals ...interface{}) {
  1330. once.Do(func() {
  1331. fatalError = fmt.Sprintf(s, vals...)
  1332. })
  1333. }
  1334. for i := 0; i < max; i++ {
  1335. go func(id int) {
  1336. defer wg.Done()
  1337. tx, err := dbt.db.Begin()
  1338. atomic.AddInt32(&remaining, -1)
  1339. if err != nil {
  1340. if err.Error() != "Error 1040: Too many connections" {
  1341. fatalf("Error on Conn %d: %s", id, err.Error())
  1342. }
  1343. return
  1344. }
  1345. // keep the connection busy until all connections are open
  1346. for remaining > 0 {
  1347. if _, err = tx.Exec("DO 1"); err != nil {
  1348. fatalf("Error on Conn %d: %s", id, err.Error())
  1349. return
  1350. }
  1351. }
  1352. if err = tx.Commit(); err != nil {
  1353. fatalf("Error on Conn %d: %s", id, err.Error())
  1354. return
  1355. }
  1356. // everything went fine with this connection
  1357. atomic.AddInt32(&succeeded, 1)
  1358. }(i)
  1359. }
  1360. // wait until all conections are open
  1361. wg.Wait()
  1362. if fatalError != "" {
  1363. dbt.Fatal(fatalError)
  1364. }
  1365. dbt.Logf("Reached %d concurrent connections\r\n", succeeded)
  1366. })
  1367. }
  1368. // Tests custom dial functions
  1369. func TestCustomDial(t *testing.T) {
  1370. if !available {
  1371. t.Skipf("MySQL-Server not running on %s", netAddr)
  1372. }
  1373. // our custom dial function which justs wraps net.Dial here
  1374. RegisterDial("mydial", func(addr string) (net.Conn, error) {
  1375. return net.Dial(prot, addr)
  1376. })
  1377. db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@mydial(%s)/%s?timeout=30s&strict=true", user, pass, addr, dbname))
  1378. if err != nil {
  1379. t.Fatalf("Error connecting: %s", err.Error())
  1380. }
  1381. defer db.Close()
  1382. if _, err = db.Exec("DO 1"); err != nil {
  1383. t.Fatalf("Connection failed: %s", err.Error())
  1384. }
  1385. }
  1386. func TestSqlInjection(t *testing.T) {
  1387. createTest := func(arg string) func(dbt *DBTest) {
  1388. return func(dbt *DBTest) {
  1389. dbt.mustExec("CREATE TABLE test (v INTEGER)")
  1390. dbt.mustExec("INSERT INTO test VALUES (?)", 1)
  1391. var v int
  1392. // NULL can't be equal to anything, the idea here is to inject query so it returns row
  1393. // This test verifies that escapeQuotes and escapeBackslash are working properly
  1394. err := dbt.db.QueryRow("SELECT v FROM test WHERE NULL = ?", arg).Scan(&v)
  1395. if err == sql.ErrNoRows {
  1396. return // success, sql injection failed
  1397. } else if err == nil {
  1398. dbt.Errorf("Sql injection successful with arg: %s", arg)
  1399. } else {
  1400. dbt.Errorf("Error running query with arg: %s; err: %s", arg, err.Error())
  1401. }
  1402. }
  1403. }
  1404. dsns := []string{
  1405. dsn,
  1406. dsn + "&sql_mode=NO_BACKSLASH_ESCAPES",
  1407. }
  1408. for _, testdsn := range dsns {
  1409. runTests(t, testdsn, createTest("1 OR 1=1"))
  1410. runTests(t, testdsn, createTest("' OR '1'='1"))
  1411. }
  1412. }
  1413. // Test if inserted data is correctly retrieved after being escaped
  1414. func TestInsertRetrieveEscapedData(t *testing.T) {
  1415. testData := func(dbt *DBTest) {
  1416. dbt.mustExec("CREATE TABLE test (v VARCHAR(255))")
  1417. // All sequences that are escaped by escapeQuotes and escapeBackslash
  1418. v := "foo \x00\n\r\x1a\"'\\"
  1419. dbt.mustExec("INSERT INTO test VALUES (?)", v)
  1420. var out string
  1421. err := dbt.db.QueryRow("SELECT v FROM test").Scan(&out)
  1422. if err != nil {
  1423. dbt.Fatalf("%s", err.Error())
  1424. }
  1425. if out != v {
  1426. dbt.Errorf("%q != %q", out, v)
  1427. }
  1428. }
  1429. dsns := []string{
  1430. dsn,
  1431. dsn + "&sql_mode=NO_BACKSLASH_ESCAPES",
  1432. }
  1433. for _, testdsn := range dsns {
  1434. runTests(t, testdsn, testData)
  1435. }
  1436. }