driver_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949
  1. package mysql
  2. import (
  3. "database/sql"
  4. "fmt"
  5. "io"
  6. "io/ioutil"
  7. "net"
  8. "os"
  9. "strings"
  10. "sync"
  11. "testing"
  12. )
  13. var (
  14. dsn string
  15. netAddr string
  16. run bool
  17. once sync.Once
  18. )
  19. // See https://github.com/go-sql-driver/mysql/wiki/Testing
  20. func getEnv() bool {
  21. once.Do(func() {
  22. user := os.Getenv("MYSQL_TEST_USER")
  23. if user == "" {
  24. user = "root"
  25. }
  26. pass := os.Getenv("MYSQL_TEST_PASS")
  27. prot := os.Getenv("MYSQL_TEST_PROT")
  28. if prot == "" {
  29. prot = "tcp"
  30. }
  31. addr := os.Getenv("MYSQL_TEST_ADDR")
  32. if addr == "" {
  33. addr = "localhost:3306"
  34. }
  35. dbname := os.Getenv("MYSQL_TEST_DBNAME")
  36. if dbname == "" {
  37. dbname = "gotest"
  38. }
  39. netAddr = fmt.Sprintf("%s(%s)", prot, addr)
  40. dsn = fmt.Sprintf("%s:%s@%s/%s?timeout=30s&charset=utf8", user, pass, netAddr, dbname)
  41. c, err := net.Dial(prot, addr)
  42. if err == nil {
  43. run = true
  44. c.Close()
  45. }
  46. })
  47. return run
  48. }
  49. func mustExec(t *testing.T, db *sql.DB, query string, args ...interface{}) (res sql.Result) {
  50. res, err := db.Exec(query, args...)
  51. if err != nil {
  52. if len(query) > 300 {
  53. query = "[query too large to print]"
  54. }
  55. t.Fatalf("Error on Exec %q: %v", query, err)
  56. }
  57. return
  58. }
  59. func mustQuery(t *testing.T, db *sql.DB, query string, args ...interface{}) (rows *sql.Rows) {
  60. rows, err := db.Query(query, args...)
  61. if err != nil {
  62. if len(query) > 300 {
  63. query = "[query too large to print]"
  64. }
  65. t.Fatalf("Error on Query %q: %v", query, err)
  66. }
  67. return
  68. }
  69. func TestCRUD(t *testing.T) {
  70. if !getEnv() {
  71. t.Logf("MySQL-Server not running on %s. Skipping TestCRUD", netAddr)
  72. return
  73. }
  74. db, err := sql.Open("mysql", dsn)
  75. if err != nil {
  76. t.Fatalf("Error connecting: %v", err)
  77. }
  78. defer db.Close()
  79. mustExec(t, db, "DROP TABLE IF EXISTS test")
  80. // Create Table
  81. mustExec(t, db, "CREATE TABLE test (value BOOL)")
  82. // Test for unexpected data
  83. var out bool
  84. rows := mustQuery(t, db, ("SELECT * FROM test"))
  85. if rows.Next() {
  86. t.Error("unexpected data in empty table")
  87. }
  88. // Create Data
  89. res := mustExec(t, db, ("INSERT INTO test VALUES (1)"))
  90. count, err := res.RowsAffected()
  91. if err != nil {
  92. t.Fatalf("res.RowsAffected() returned error: %v", err)
  93. }
  94. if count != 1 {
  95. t.Fatalf("Expected 1 affected row, got %d", count)
  96. }
  97. id, err := res.LastInsertId()
  98. if err != nil {
  99. t.Fatalf("res.LastInsertId() returned error: %v", err)
  100. }
  101. if id != 0 {
  102. t.Fatalf("Expected InsertID 0, got %d", id)
  103. }
  104. // Read
  105. rows = mustQuery(t, db, ("SELECT value FROM test"))
  106. if rows.Next() {
  107. rows.Scan(&out)
  108. if true != out {
  109. t.Errorf("true != %t", out)
  110. }
  111. if rows.Next() {
  112. t.Error("unexpected data")
  113. }
  114. } else {
  115. t.Error("no data")
  116. }
  117. // Update
  118. res = mustExec(t, db, "UPDATE test SET value = ? WHERE value = ?", false, true)
  119. count, err = res.RowsAffected()
  120. if err != nil {
  121. t.Fatalf("res.RowsAffected() returned error: %v", err)
  122. }
  123. if count != 1 {
  124. t.Fatalf("Expected 1 affected row, got %d", count)
  125. }
  126. // Check Update
  127. rows = mustQuery(t, db, ("SELECT value FROM test"))
  128. if rows.Next() {
  129. rows.Scan(&out)
  130. if false != out {
  131. t.Errorf("false != %t", out)
  132. }
  133. if rows.Next() {
  134. t.Error("unexpected data")
  135. }
  136. } else {
  137. t.Error("no data")
  138. }
  139. // Delete
  140. res = mustExec(t, db, "DELETE FROM test WHERE value = ?", false)
  141. count, err = res.RowsAffected()
  142. if err != nil {
  143. t.Fatalf("res.RowsAffected() returned error: %v", err)
  144. }
  145. if count != 1 {
  146. t.Fatalf("Expected 1 affected row, got %d", count)
  147. }
  148. // Check for unexpected rows
  149. res = mustExec(t, db, "DELETE FROM test")
  150. count, err = res.RowsAffected()
  151. if err != nil {
  152. t.Fatalf("res.RowsAffected() returned error: %v", err)
  153. }
  154. if count != 0 {
  155. t.Fatalf("Expected 0 affected row, got %d", count)
  156. }
  157. }
  158. func TestInt(t *testing.T) {
  159. if !getEnv() {
  160. t.Logf("MySQL-Server not running on %s. Skipping TestInt", netAddr)
  161. return
  162. }
  163. db, err := sql.Open("mysql", dsn)
  164. if err != nil {
  165. t.Fatalf("Error connecting: %v", err)
  166. }
  167. defer db.Close()
  168. mustExec(t, db, "DROP TABLE IF EXISTS test")
  169. types := [5]string{"TINYINT", "SMALLINT", "MEDIUMINT", "INT", "BIGINT"}
  170. in := int64(42)
  171. var out int64
  172. var rows *sql.Rows
  173. // SIGNED
  174. for _, v := range types {
  175. mustExec(t, db, "CREATE TABLE test (value "+v+")")
  176. mustExec(t, db, ("INSERT INTO test VALUES (?)"), in)
  177. rows = mustQuery(t, db, ("SELECT value FROM test"))
  178. if rows.Next() {
  179. rows.Scan(&out)
  180. if in != out {
  181. t.Errorf("%s: %d != %d", v, in, out)
  182. }
  183. } else {
  184. t.Errorf("%s: no data", v)
  185. }
  186. mustExec(t, db, "DROP TABLE IF EXISTS test")
  187. }
  188. // UNSIGNED ZEROFILL
  189. for _, v := range types {
  190. mustExec(t, db, "CREATE TABLE test (value "+v+" ZEROFILL)")
  191. mustExec(t, db, ("INSERT INTO test VALUES (?)"), in)
  192. rows = mustQuery(t, db, ("SELECT value FROM test"))
  193. if rows.Next() {
  194. rows.Scan(&out)
  195. if in != out {
  196. t.Errorf("%s ZEROFILL: %d != %d", v, in, out)
  197. }
  198. } else {
  199. t.Errorf("%s ZEROFILL: no data", v)
  200. }
  201. mustExec(t, db, "DROP TABLE IF EXISTS test")
  202. }
  203. }
  204. func TestFloat(t *testing.T) {
  205. if !getEnv() {
  206. t.Logf("MySQL-Server not running on %s. Skipping TestFloat", netAddr)
  207. return
  208. }
  209. db, err := sql.Open("mysql", dsn)
  210. if err != nil {
  211. t.Fatalf("Error connecting: %v", err)
  212. }
  213. defer db.Close()
  214. mustExec(t, db, "DROP TABLE IF EXISTS test")
  215. types := [2]string{"FLOAT", "DOUBLE"}
  216. in := float32(42.23)
  217. var out float32
  218. var rows *sql.Rows
  219. for _, v := range types {
  220. mustExec(t, db, "CREATE TABLE test (value "+v+")")
  221. mustExec(t, db, ("INSERT INTO test VALUES (?)"), in)
  222. rows = mustQuery(t, db, ("SELECT value FROM test"))
  223. if rows.Next() {
  224. rows.Scan(&out)
  225. if in != out {
  226. t.Errorf("%s: %g != %g", v, in, out)
  227. }
  228. } else {
  229. t.Errorf("%s: no data", v)
  230. }
  231. mustExec(t, db, "DROP TABLE IF EXISTS test")
  232. }
  233. }
  234. func TestString(t *testing.T) {
  235. if !getEnv() {
  236. t.Logf("MySQL-Server not running on %s. Skipping TestString", netAddr)
  237. return
  238. }
  239. db, err := sql.Open("mysql", dsn)
  240. if err != nil {
  241. t.Fatalf("Error connecting: %v", err)
  242. }
  243. defer db.Close()
  244. mustExec(t, db, "DROP TABLE IF EXISTS test")
  245. types := [6]string{"CHAR(255)", "VARCHAR(255)", "TINYTEXT", "TEXT", "MEDIUMTEXT", "LONGTEXT"}
  246. in := "κόσμε üöäßñóùéàâÿœ'îë Árvíztűrő いろはにほへとちりぬるを イロハニホヘト דג סקרן чащах น่าฟังเอย"
  247. var out string
  248. var rows *sql.Rows
  249. for _, v := range types {
  250. mustExec(t, db, "CREATE TABLE test (value "+v+") CHARACTER SET utf8 COLLATE utf8_unicode_ci")
  251. mustExec(t, db, ("INSERT INTO test VALUES (?)"), in)
  252. rows = mustQuery(t, db, ("SELECT value FROM test"))
  253. if rows.Next() {
  254. rows.Scan(&out)
  255. if in != out {
  256. t.Errorf("%s: %s != %s", v, in, out)
  257. }
  258. } else {
  259. t.Errorf("%s: no data", v)
  260. }
  261. mustExec(t, db, "DROP TABLE IF EXISTS test")
  262. }
  263. // BLOB
  264. mustExec(t, db, "CREATE TABLE test (id int, value BLOB) CHARACTER SET utf8 COLLATE utf8_unicode_ci")
  265. id := 2
  266. in = "Lorem ipsum dolor sit amet, consetetur sadipscing elitr, " +
  267. "sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, " +
  268. "sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. " +
  269. "Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet. " +
  270. "Lorem ipsum dolor sit amet, consetetur sadipscing elitr, " +
  271. "sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, " +
  272. "sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. " +
  273. "Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet."
  274. mustExec(t, db, ("INSERT INTO test VALUES (?, ?)"), id, in)
  275. err = db.QueryRow("SELECT value FROM test WHERE id = ?", id).Scan(&out)
  276. if err != nil {
  277. t.Fatalf("Error on BLOB-Query: %v", err)
  278. } else if out != in {
  279. t.Errorf("BLOB: %s != %s", in, out)
  280. }
  281. return
  282. }
  283. func TestDateTime(t *testing.T) {
  284. if !getEnv() {
  285. t.Logf("MySQL-Server not running on %s. Skipping TestString", netAddr)
  286. return
  287. }
  288. db, err := sql.Open("mysql", dsn)
  289. if err != nil {
  290. t.Fatalf("Error connecting: %v", err)
  291. }
  292. defer db.Close()
  293. mustExec(t, db, "DROP TABLE IF EXISTS test")
  294. types := [...]string{"DATE", "DATETIME"}
  295. in := [...]string{"2012-06-14", "2011-11-20 21:27:37"}
  296. var out string
  297. var rows *sql.Rows
  298. for i, v := range types {
  299. mustExec(t, db, "CREATE TABLE test (value "+v+") CHARACTER SET utf8 COLLATE utf8_unicode_ci")
  300. mustExec(t, db, ("INSERT INTO test VALUES (?)"), in[i])
  301. rows = mustQuery(t, db, ("SELECT value FROM test"))
  302. if rows.Next() {
  303. rows.Scan(&out)
  304. if in[i] != out {
  305. t.Errorf("%s: %s != %s", v, in[i], out)
  306. }
  307. } else {
  308. t.Errorf("%s: no data", v)
  309. }
  310. mustExec(t, db, "DROP TABLE IF EXISTS test")
  311. }
  312. }
  313. func TestNULL(t *testing.T) {
  314. if !getEnv() {
  315. t.Logf("MySQL-Server not running on %s. Skipping TestNULL", netAddr)
  316. return
  317. }
  318. db, err := sql.Open("mysql", dsn)
  319. if err != nil {
  320. t.Fatalf("Error connecting: %v", err)
  321. }
  322. defer db.Close()
  323. nullStmt, err := db.Prepare("SELECT NULL")
  324. if err != nil {
  325. t.Fatal(err)
  326. }
  327. defer nullStmt.Close()
  328. nonNullStmt, err := db.Prepare("SELECT 1")
  329. if err != nil {
  330. t.Fatal(err)
  331. }
  332. defer nonNullStmt.Close()
  333. // NullBool
  334. var nb sql.NullBool
  335. // Invalid
  336. err = nullStmt.QueryRow().Scan(&nb)
  337. if err != nil {
  338. t.Fatal(err)
  339. }
  340. if nb.Valid {
  341. t.Error("Valid NullBool which should be invalid")
  342. }
  343. // Valid
  344. err = nonNullStmt.QueryRow().Scan(&nb)
  345. if err != nil {
  346. t.Fatal(err)
  347. }
  348. if !nb.Valid {
  349. t.Error("Invalid NullBool which should be valid")
  350. } else if nb.Bool != true {
  351. t.Errorf("Unexpected NullBool value: %t (should be true)", nb.Bool)
  352. }
  353. // NullFloat64
  354. var nf sql.NullFloat64
  355. // Invalid
  356. err = nullStmt.QueryRow().Scan(&nf)
  357. if err != nil {
  358. t.Fatal(err)
  359. }
  360. if nf.Valid {
  361. t.Error("Valid NullFloat64 which should be invalid")
  362. }
  363. // Valid
  364. err = nonNullStmt.QueryRow().Scan(&nf)
  365. if err != nil {
  366. t.Fatal(err)
  367. }
  368. if !nf.Valid {
  369. t.Error("Invalid NullFloat64 which should be valid")
  370. } else if nf.Float64 != float64(1) {
  371. t.Errorf("Unexpected NullFloat64 value: %f (should be 1.0)", nf.Float64)
  372. }
  373. // NullInt64
  374. var ni sql.NullInt64
  375. // Invalid
  376. err = nullStmt.QueryRow().Scan(&ni)
  377. if err != nil {
  378. t.Fatal(err)
  379. }
  380. if ni.Valid {
  381. t.Error("Valid NullInt64 which should be invalid")
  382. }
  383. // Valid
  384. err = nonNullStmt.QueryRow().Scan(&ni)
  385. if err != nil {
  386. t.Fatal(err)
  387. }
  388. if !ni.Valid {
  389. t.Error("Invalid NullInt64 which should be valid")
  390. } else if ni.Int64 != int64(1) {
  391. t.Errorf("Unexpected NullInt64 value: %d (should be 1)", ni.Int64)
  392. }
  393. // NullString
  394. var ns sql.NullString
  395. // Invalid
  396. err = nullStmt.QueryRow().Scan(&ns)
  397. if err != nil {
  398. t.Fatal(err)
  399. }
  400. if ns.Valid {
  401. t.Error("Valid NullString which should be invalid")
  402. }
  403. // Valid
  404. err = nonNullStmt.QueryRow().Scan(&ns)
  405. if err != nil {
  406. t.Fatal(err)
  407. }
  408. if !ns.Valid {
  409. t.Error("Invalid NullString which should be valid")
  410. } else if ns.String != `1` {
  411. t.Error("Unexpected NullString value:" + ns.String + " (should be `1`)")
  412. }
  413. // Insert NULL
  414. mustExec(t, db, "CREATE TABLE test (dummmy1 int, value int, dummy2 int)")
  415. mustExec(t, db, ("INSERT INTO test VALUES (?, ?, ?)"), 1, nil, 2)
  416. var out interface{}
  417. rows := mustQuery(t, db, ("SELECT * FROM test"))
  418. if rows.Next() {
  419. rows.Scan(&out)
  420. if out != nil {
  421. t.Errorf("%v != nil", out)
  422. }
  423. } else {
  424. t.Error("no data")
  425. }
  426. mustExec(t, db, "DROP TABLE IF EXISTS test")
  427. }
  428. func TestLongData(t *testing.T) {
  429. if !getEnv() {
  430. t.Logf("MySQL-Server not running on %s. Skipping TestLongData", netAddr)
  431. return
  432. }
  433. db, err := sql.Open("mysql", dsn)
  434. if err != nil {
  435. t.Fatalf("Error connecting: %v", err)
  436. }
  437. defer db.Close()
  438. var maxAllowedPacketSize int
  439. err = db.QueryRow("select @@max_allowed_packet").Scan(&maxAllowedPacketSize)
  440. if err != nil {
  441. t.Fatal(err)
  442. }
  443. maxAllowedPacketSize--
  444. // don't get too ambitious
  445. if maxAllowedPacketSize > 1<<25 {
  446. maxAllowedPacketSize = 1 << 25
  447. }
  448. mustExec(t, db, "DROP TABLE IF EXISTS test")
  449. mustExec(t, db, "CREATE TABLE test (value LONGBLOB) CHARACTER SET utf8 COLLATE utf8_unicode_ci")
  450. in := strings.Repeat(`0`, maxAllowedPacketSize+1)
  451. var out string
  452. var rows *sql.Rows
  453. // Long text data
  454. const nonDataQueryLen = 28 // length query w/o value
  455. inS := in[:maxAllowedPacketSize-nonDataQueryLen]
  456. mustExec(t, db, "INSERT INTO test VALUES('"+inS+"')")
  457. rows = mustQuery(t, db, "SELECT value FROM test")
  458. if rows.Next() {
  459. rows.Scan(&out)
  460. if inS != out {
  461. t.Fatalf("LONGBLOB: length in: %d, length out: %d", len(inS), len(out))
  462. }
  463. if rows.Next() {
  464. t.Error("LONGBLOB: unexpexted row")
  465. }
  466. } else {
  467. t.Fatalf("LONGBLOB: no data")
  468. }
  469. // Empty table
  470. mustExec(t, db, "TRUNCATE TABLE test")
  471. // Long binary data
  472. mustExec(t, db, "INSERT INTO test VALUES(?)", in)
  473. rows = mustQuery(t, db, "SELECT value FROM test WHERE 1=?", 1)
  474. if rows.Next() {
  475. rows.Scan(&out)
  476. if in != out {
  477. t.Fatalf("LONGBLOB: length in: %d, length out: %d", len(in), len(out))
  478. }
  479. if rows.Next() {
  480. t.Error("LONGBLOB: unexpexted row")
  481. }
  482. } else {
  483. t.Fatalf("LONGBLOB: no data")
  484. }
  485. mustExec(t, db, "DROP TABLE IF EXISTS test")
  486. }
  487. func verifyLoadDataResult(t *testing.T, db *sql.DB) {
  488. rows, err := db.Query("SELECT * FROM test")
  489. if err != nil {
  490. t.Fatal(err.Error())
  491. }
  492. i := 0
  493. values := [4]string{
  494. "a string",
  495. "a string containing a \t",
  496. "a string containing a \n",
  497. "a string containing both \t\n",
  498. }
  499. var id int
  500. var value string
  501. for rows.Next() {
  502. i++
  503. err = rows.Scan(&id, &value)
  504. if err != nil {
  505. t.Fatal(err.Error())
  506. }
  507. if i != id {
  508. t.Fatalf("%d != %d", i, id)
  509. }
  510. if values[i-1] != value {
  511. t.Fatalf("%s != %s", values[i-1], value)
  512. }
  513. }
  514. err = rows.Err()
  515. if err != nil {
  516. t.Fatal(err.Error())
  517. }
  518. if i != 4 {
  519. t.Fatalf("Rows count mismatch. Got %d, want 4", i)
  520. }
  521. }
  522. func TestLoadData(t *testing.T) {
  523. if !getEnv() {
  524. t.Logf("MySQL-Server not running on %s. Skipping TestLoadData", netAddr)
  525. return
  526. }
  527. db, err := sql.Open("mysql", dsn)
  528. if err != nil {
  529. t.Fatalf("Error connecting: %v", err)
  530. }
  531. defer db.Close()
  532. file, err := ioutil.TempFile("", "gotest")
  533. defer os.Remove(file.Name())
  534. if err != nil {
  535. t.Fatal(err)
  536. }
  537. 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")
  538. file.Close()
  539. mustExec(t, db, "DROP TABLE IF EXISTS test")
  540. mustExec(t, db, "CREATE TABLE test (id INT NOT NULL PRIMARY KEY, value TEXT NOT NULL) CHARACTER SET utf8 COLLATE utf8_unicode_ci")
  541. // Local File
  542. RegisterLocalFile(file.Name())
  543. mustExec(t, db, "LOAD DATA LOCAL INFILE '"+file.Name()+"' INTO TABLE test")
  544. verifyLoadDataResult(t, db)
  545. // negative test
  546. _, err = db.Exec("LOAD DATA LOCAL INFILE 'doesnotexist' INTO TABLE test")
  547. if err == nil {
  548. t.Fatal("Load non-existent file didn't fail")
  549. } else if err.Error() != "Local File 'doesnotexist' is not registered. Use the DSN parameter 'allowAllFiles=true' to allow all files" {
  550. t.Fatal(err.Error())
  551. }
  552. // Empty table
  553. mustExec(t, db, "TRUNCATE TABLE test")
  554. // Reader
  555. RegisterReaderHandler("test", func() io.Reader {
  556. file, err = os.Open(file.Name())
  557. if err != nil {
  558. t.Fatal(err)
  559. }
  560. return file
  561. })
  562. mustExec(t, db, "LOAD DATA LOCAL INFILE 'Reader::test' INTO TABLE test")
  563. verifyLoadDataResult(t, db)
  564. // negative test
  565. _, err = db.Exec("LOAD DATA LOCAL INFILE 'Reader::doesnotexist' INTO TABLE test")
  566. if err == nil {
  567. t.Fatal("Load non-existent Reader didn't fail")
  568. } else if err.Error() != "Reader 'doesnotexist' is not registered" {
  569. t.Fatal(err.Error())
  570. }
  571. mustExec(t, db, "DROP TABLE IF EXISTS test")
  572. }
  573. // Special cases
  574. func TestRowsClose(t *testing.T) {
  575. if !getEnv() {
  576. t.Logf("MySQL-Server not running on %s. Skipping TestRowsClose", netAddr)
  577. return
  578. }
  579. db, err := sql.Open("mysql", dsn)
  580. if err != nil {
  581. t.Fatalf("Error connecting: %v", err)
  582. }
  583. defer db.Close()
  584. rows, err := db.Query("SELECT 1")
  585. if err != nil {
  586. t.Fatal(err)
  587. }
  588. err = rows.Close()
  589. if err != nil {
  590. t.Fatal(err)
  591. }
  592. if rows.Next() {
  593. t.Fatal("Unexpected row after rows.Close()")
  594. }
  595. err = rows.Err()
  596. if err != nil {
  597. t.Fatal(err)
  598. }
  599. }
  600. // dangling statements
  601. // http://code.google.com/p/go/issues/detail?id=3865
  602. func TestCloseStmtBeforeRows(t *testing.T) {
  603. if !getEnv() {
  604. t.Logf("MySQL-Server not running on %s. Skipping TestCloseStmtBeforeRows", netAddr)
  605. return
  606. }
  607. db, err := sql.Open("mysql", dsn)
  608. if err != nil {
  609. t.Fatalf("Error connecting: %v", err)
  610. }
  611. defer db.Close()
  612. stmt, err := db.Prepare("SELECT 1")
  613. if err != nil {
  614. t.Fatal(err)
  615. }
  616. rows, err := stmt.Query()
  617. if err != nil {
  618. stmt.Close()
  619. t.Fatal(err)
  620. }
  621. defer rows.Close()
  622. err = stmt.Close()
  623. if err != nil {
  624. t.Fatal(err)
  625. }
  626. if !rows.Next() {
  627. t.Fatal("Getting row failed")
  628. } else {
  629. err = rows.Err()
  630. if err != nil {
  631. t.Fatal(err)
  632. }
  633. var out bool
  634. err = rows.Scan(&out)
  635. if err != nil {
  636. t.Fatalf("Error on rows.Scan(): %v", err)
  637. }
  638. if out != true {
  639. t.Errorf("true != %t", out)
  640. }
  641. }
  642. }
  643. // It is valid to have multiple Rows for the same Stmt
  644. // http://code.google.com/p/go/issues/detail?id=3734
  645. func TestStmtMultiRows(t *testing.T) {
  646. if !getEnv() {
  647. t.Logf("MySQL-Server not running on %s. Skipping TestStmtMultiRows", netAddr)
  648. return
  649. }
  650. db, err := sql.Open("mysql", dsn)
  651. if err != nil {
  652. t.Fatalf("Error connecting: %v", err)
  653. }
  654. defer db.Close()
  655. stmt, err := db.Prepare("SELECT 1 UNION SELECT 0")
  656. if err != nil {
  657. t.Fatal(err)
  658. }
  659. rows1, err := stmt.Query()
  660. if err != nil {
  661. stmt.Close()
  662. t.Fatal(err)
  663. }
  664. defer rows1.Close()
  665. rows2, err := stmt.Query()
  666. if err != nil {
  667. stmt.Close()
  668. t.Fatal(err)
  669. }
  670. defer rows2.Close()
  671. var out bool
  672. // 1
  673. if !rows1.Next() {
  674. t.Fatal("1st rows1.Next failed")
  675. } else {
  676. err = rows1.Err()
  677. if err != nil {
  678. t.Fatal(err)
  679. }
  680. err = rows1.Scan(&out)
  681. if err != nil {
  682. t.Fatalf("Error on rows.Scan(): %v", err)
  683. }
  684. if out != true {
  685. t.Errorf("true != %t", out)
  686. }
  687. }
  688. if !rows2.Next() {
  689. t.Fatal("1st rows2.Next failed")
  690. } else {
  691. err = rows2.Err()
  692. if err != nil {
  693. t.Fatal(err)
  694. }
  695. err = rows2.Scan(&out)
  696. if err != nil {
  697. t.Fatalf("Error on rows.Scan(): %v", err)
  698. }
  699. if out != true {
  700. t.Errorf("true != %t", out)
  701. }
  702. }
  703. // 2
  704. if !rows1.Next() {
  705. t.Fatal("2nd rows1.Next failed")
  706. } else {
  707. err = rows1.Err()
  708. if err != nil {
  709. t.Fatal(err)
  710. }
  711. err = rows1.Scan(&out)
  712. if err != nil {
  713. t.Fatalf("Error on rows.Scan(): %v", err)
  714. }
  715. if out != false {
  716. t.Errorf("false != %t", out)
  717. }
  718. if rows1.Next() {
  719. t.Fatal("Unexpected row on rows1")
  720. }
  721. err = rows1.Close()
  722. if err != nil {
  723. t.Fatal(err)
  724. }
  725. }
  726. if !rows2.Next() {
  727. t.Fatal("2nd rows2.Next failed")
  728. } else {
  729. err = rows2.Err()
  730. if err != nil {
  731. t.Fatal(err)
  732. }
  733. err = rows2.Scan(&out)
  734. if err != nil {
  735. t.Fatalf("Error on rows.Scan(): %v", err)
  736. }
  737. if out != false {
  738. t.Errorf("false != %t", out)
  739. }
  740. if rows2.Next() {
  741. t.Fatal("Unexpected row on rows2")
  742. }
  743. err = rows2.Close()
  744. if err != nil {
  745. t.Fatal(err)
  746. }
  747. }
  748. }
  749. func TestConcurrent(t *testing.T) {
  750. if os.Getenv("MYSQL_TEST_CONCURRENT") != "1" {
  751. t.Log("CONCURRENT env var not set. Skipping TestConcurrent")
  752. return
  753. }
  754. if !getEnv() {
  755. t.Logf("MySQL-Server not running on %s. Skipping TestConcurrent", netAddr)
  756. return
  757. }
  758. db, err := sql.Open("mysql", dsn)
  759. if err != nil {
  760. t.Fatalf("Error connecting: %v", err)
  761. }
  762. defer db.Close()
  763. var max int
  764. err = db.QueryRow("SELECT @@max_connections").Scan(&max)
  765. if err != nil {
  766. t.Fatalf("%v", err)
  767. }
  768. t.Logf("Testing %d concurrent connections \r\n", max)
  769. canStop := false
  770. c := make(chan struct{}, max)
  771. for i := 0; i < max; i++ {
  772. go func() {
  773. tx, err := db.Begin()
  774. if err != nil {
  775. canStop = true
  776. t.Fatalf("Error on Con %d: %s", i, err.Error())
  777. }
  778. c <- struct{}{}
  779. for !canStop {
  780. _, err := tx.Exec("SELECT 1")
  781. if err != nil {
  782. canStop = true
  783. t.Fatalf("Error on Con %d: %s", i, err.Error())
  784. }
  785. }
  786. err = tx.Commit()
  787. if err != nil {
  788. canStop = true
  789. t.Fatalf("Error on Con %d: %s", i, err.Error())
  790. }
  791. }()
  792. }
  793. for i := 0; i < max; i++ {
  794. <-c
  795. }
  796. canStop = true
  797. }