driver_test.go 39 KB

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