driver_test.go 39 KB

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