driver_test.go 20 KB

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