| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947 |
- package mysql
- import (
- "database/sql"
- "fmt"
- "io/ioutil"
- "net"
- "os"
- "strings"
- "sync"
- "testing"
- )
- var (
- dsn string
- netAddr string
- run bool
- once sync.Once
- )
- // See https://github.com/Go-SQL-Driver/MySQL/wiki/Testing
- func getEnv() bool {
- once.Do(func() {
- user := os.Getenv("MYSQL_TEST_USER")
- if user == "" {
- user = "root"
- }
- pass := os.Getenv("MYSQL_TEST_PASS")
- prot := os.Getenv("MYSQL_TEST_PROT")
- if prot == "" {
- prot = "tcp"
- }
- addr := os.Getenv("MYSQL_TEST_ADDR")
- if addr == "" {
- addr = "localhost:3306"
- }
- dbname := os.Getenv("MYSQL_TEST_DBNAME")
- if dbname == "" {
- dbname = "gotest"
- }
- netAddr = fmt.Sprintf("%s(%s)", prot, addr)
- dsn = fmt.Sprintf("%s:%s@%s/%s?timeout=30s&charset=utf8", user, pass, netAddr, dbname)
- c, err := net.Dial(prot, addr)
- if err == nil {
- run = true
- c.Close()
- }
- })
- return run
- }
- func mustExec(t *testing.T, db *sql.DB, query string, args ...interface{}) (res sql.Result) {
- res, err := db.Exec(query, args...)
- if err != nil {
- if len(query) > 300 {
- query = "[query too large to print]"
- }
- t.Fatalf("Error on Exec %q: %v", query, err)
- }
- return
- }
- func mustQuery(t *testing.T, db *sql.DB, query string, args ...interface{}) (rows *sql.Rows) {
- rows, err := db.Query(query, args...)
- if err != nil {
- if len(query) > 300 {
- query = "[query too large to print]"
- }
- t.Fatalf("Error on Query %q: %v", query, err)
- }
- return
- }
- func TestCRUD(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestCRUD", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- // Create Table
- mustExec(t, db, "CREATE TABLE test (value BOOL)")
- // Test for unexpected data
- var out bool
- rows := mustQuery(t, db, ("SELECT * FROM test"))
- if rows.Next() {
- t.Error("unexpected data in empty table")
- }
- // Create Data
- res := mustExec(t, db, ("INSERT INTO test VALUES (1)"))
- count, err := res.RowsAffected()
- if err != nil {
- t.Fatalf("res.RowsAffected() returned error: %v", err)
- }
- if count != 1 {
- t.Fatalf("Expected 1 affected row, got %d", count)
- }
- id, err := res.LastInsertId()
- if err != nil {
- t.Fatalf("res.LastInsertId() returned error: %v", err)
- }
- if id != 0 {
- t.Fatalf("Expected InsertID 0, got %d", id)
- }
- // Read
- rows = mustQuery(t, db, ("SELECT value FROM test"))
- if rows.Next() {
- rows.Scan(&out)
- if true != out {
- t.Errorf("true != %t", out)
- }
- if rows.Next() {
- t.Error("unexpected data")
- }
- } else {
- t.Error("no data")
- }
- // Update
- res = mustExec(t, db, "UPDATE test SET value = ? WHERE value = ?", false, true)
- count, err = res.RowsAffected()
- if err != nil {
- t.Fatalf("res.RowsAffected() returned error: %v", err)
- }
- if count != 1 {
- t.Fatalf("Expected 1 affected row, got %d", count)
- }
- // Check Update
- rows = mustQuery(t, db, ("SELECT value FROM test"))
- if rows.Next() {
- rows.Scan(&out)
- if false != out {
- t.Errorf("false != %t", out)
- }
- if rows.Next() {
- t.Error("unexpected data")
- }
- } else {
- t.Error("no data")
- }
- // Delete
- res = mustExec(t, db, "DELETE FROM test WHERE value = ?", false)
- count, err = res.RowsAffected()
- if err != nil {
- t.Fatalf("res.RowsAffected() returned error: %v", err)
- }
- if count != 1 {
- t.Fatalf("Expected 1 affected row, got %d", count)
- }
- // Check for unexpected rows
- res = mustExec(t, db, "DELETE FROM test")
- count, err = res.RowsAffected()
- if err != nil {
- t.Fatalf("res.RowsAffected() returned error: %v", err)
- }
- if count != 0 {
- t.Fatalf("Expected 0 affected row, got %d", count)
- }
- }
- func TestInt(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestInt", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- types := [5]string{"TINYINT", "SMALLINT", "MEDIUMINT", "INT", "BIGINT"}
- in := int64(42)
- var out int64
- var rows *sql.Rows
- // SIGNED
- for _, v := range types {
- mustExec(t, db, "CREATE TABLE test (value "+v+")")
- mustExec(t, db, ("INSERT INTO test VALUES (?)"), in)
- rows = mustQuery(t, db, ("SELECT value FROM test"))
- if rows.Next() {
- rows.Scan(&out)
- if in != out {
- t.Errorf("%s: %d != %d", v, in, out)
- }
- } else {
- t.Errorf("%s: no data", v)
- }
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- }
- // UNSIGNED ZEROFILL
- for _, v := range types {
- mustExec(t, db, "CREATE TABLE test (value "+v+" ZEROFILL)")
- mustExec(t, db, ("INSERT INTO test VALUES (?)"), in)
- rows = mustQuery(t, db, ("SELECT value FROM test"))
- if rows.Next() {
- rows.Scan(&out)
- if in != out {
- t.Errorf("%s ZEROFILL: %d != %d", v, in, out)
- }
- } else {
- t.Errorf("%s ZEROFILL: no data", v)
- }
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- }
- }
- func TestFloat(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestFloat", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- types := [2]string{"FLOAT", "DOUBLE"}
- in := float32(42.23)
- var out float32
- var rows *sql.Rows
- for _, v := range types {
- mustExec(t, db, "CREATE TABLE test (value "+v+")")
- mustExec(t, db, ("INSERT INTO test VALUES (?)"), in)
- rows = mustQuery(t, db, ("SELECT value FROM test"))
- if rows.Next() {
- rows.Scan(&out)
- if in != out {
- t.Errorf("%s: %g != %g", v, in, out)
- }
- } else {
- t.Errorf("%s: no data", v)
- }
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- }
- }
- func TestString(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestString", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- types := [6]string{"CHAR(255)", "VARCHAR(255)", "TINYTEXT", "TEXT", "MEDIUMTEXT", "LONGTEXT"}
- in := "κόσμε üöäßñóùéàâÿœ'îë Árvíztűrő いろはにほへとちりぬるを イロハニホヘト דג סקרן чащах น่าฟังเอย"
- var out string
- var rows *sql.Rows
- for _, v := range types {
- mustExec(t, db, "CREATE TABLE test (value "+v+") CHARACTER SET utf8 COLLATE utf8_unicode_ci")
- mustExec(t, db, ("INSERT INTO test VALUES (?)"), in)
- rows = mustQuery(t, db, ("SELECT value FROM test"))
- if rows.Next() {
- rows.Scan(&out)
- if in != out {
- t.Errorf("%s: %s != %s", v, in, out)
- }
- } else {
- t.Errorf("%s: no data", v)
- }
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- }
- // BLOB
- mustExec(t, db, "CREATE TABLE test (id int, value BLOB) CHARACTER SET utf8 COLLATE utf8_unicode_ci")
- id := 2
- in = "Lorem ipsum dolor sit amet, consetetur sadipscing elitr, " +
- "sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, " +
- "sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. " +
- "Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet. " +
- "Lorem ipsum dolor sit amet, consetetur sadipscing elitr, " +
- "sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, " +
- "sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. " +
- "Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet."
- mustExec(t, db, ("INSERT INTO test VALUES (?, ?)"), id, in)
- err = db.QueryRow("SELECT value FROM test WHERE id = ?", id).Scan(&out)
- if err != nil {
- t.Fatalf("Error on BLOB-Query: %v", err)
- } else if out != in {
- t.Errorf("BLOB: %s != %s", in, out)
- }
- return
- }
- func TestDateTime(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestString", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- types := [...]string{"DATE", "DATETIME"}
- in := [...]string{"2012-06-14", "2011-11-20 21:27:37"}
- var out string
- var rows *sql.Rows
- for i, v := range types {
- mustExec(t, db, "CREATE TABLE test (value "+v+") CHARACTER SET utf8 COLLATE utf8_unicode_ci")
- mustExec(t, db, ("INSERT INTO test VALUES (?)"), in[i])
- rows = mustQuery(t, db, ("SELECT value FROM test"))
- if rows.Next() {
- rows.Scan(&out)
- if in[i] != out {
- t.Errorf("%s: %s != %s", v, in[i], out)
- }
- } else {
- t.Errorf("%s: no data", v)
- }
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- }
- }
- func TestNULL(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestNULL", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- nullStmt, err := db.Prepare("SELECT NULL")
- if err != nil {
- t.Fatal(err)
- }
- defer nullStmt.Close()
- nonNullStmt, err := db.Prepare("SELECT 1")
- if err != nil {
- t.Fatal(err)
- }
- defer nonNullStmt.Close()
- // NullBool
- var nb sql.NullBool
- // Invalid
- err = nullStmt.QueryRow().Scan(&nb)
- if err != nil {
- t.Fatal(err)
- }
- if nb.Valid {
- t.Error("Valid NullBool which should be invalid")
- }
- // Valid
- err = nonNullStmt.QueryRow().Scan(&nb)
- if err != nil {
- t.Fatal(err)
- }
- if !nb.Valid {
- t.Error("Invalid NullBool which should be valid")
- } else if nb.Bool != true {
- t.Errorf("Unexpected NullBool value: %t (should be true)", nb.Bool)
- }
- // NullFloat64
- var nf sql.NullFloat64
- // Invalid
- err = nullStmt.QueryRow().Scan(&nf)
- if err != nil {
- t.Fatal(err)
- }
- if nf.Valid {
- t.Error("Valid NullFloat64 which should be invalid")
- }
- // Valid
- err = nonNullStmt.QueryRow().Scan(&nf)
- if err != nil {
- t.Fatal(err)
- }
- if !nf.Valid {
- t.Error("Invalid NullFloat64 which should be valid")
- } else if nf.Float64 != float64(1) {
- t.Errorf("Unexpected NullFloat64 value: %f (should be 1.0)", nf.Float64)
- }
- // NullInt64
- var ni sql.NullInt64
- // Invalid
- err = nullStmt.QueryRow().Scan(&ni)
- if err != nil {
- t.Fatal(err)
- }
- if ni.Valid {
- t.Error("Valid NullInt64 which should be invalid")
- }
- // Valid
- err = nonNullStmt.QueryRow().Scan(&ni)
- if err != nil {
- t.Fatal(err)
- }
- if !ni.Valid {
- t.Error("Invalid NullInt64 which should be valid")
- } else if ni.Int64 != int64(1) {
- t.Errorf("Unexpected NullInt64 value: %d (should be 1)", ni.Int64)
- }
- // NullString
- var ns sql.NullString
- // Invalid
- err = nullStmt.QueryRow().Scan(&ns)
- if err != nil {
- t.Fatal(err)
- }
- if ns.Valid {
- t.Error("Valid NullString which should be invalid")
- }
- // Valid
- err = nonNullStmt.QueryRow().Scan(&ns)
- if err != nil {
- t.Fatal(err)
- }
- if !ns.Valid {
- t.Error("Invalid NullString which should be valid")
- } else if ns.String != `1` {
- t.Error("Unexpected NullString value:" + ns.String + " (should be `1`)")
- }
- // Insert NULL
- mustExec(t, db, "CREATE TABLE test (dummmy1 int, value int, dummy2 int)")
- mustExec(t, db, ("INSERT INTO test VALUES (?, ?, ?)"), 1, nil, 2)
- var out interface{}
- rows := mustQuery(t, db, ("SELECT * FROM test"))
- if rows.Next() {
- rows.Scan(&out)
- if out != nil {
- t.Errorf("%v != nil", out)
- }
- } else {
- t.Error("no data")
- }
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- }
- func TestLongData(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestLongData", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- var maxAllowedPacketSize int
- err = db.QueryRow("select @@max_allowed_packet").Scan(&maxAllowedPacketSize)
- if err != nil {
- t.Fatal(err)
- }
- maxAllowedPacketSize--
- // don't get too ambitious
- if maxAllowedPacketSize > 1<<25 {
- maxAllowedPacketSize = 1 << 25
- }
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- mustExec(t, db, "CREATE TABLE test (value LONGBLOB) CHARACTER SET utf8 COLLATE utf8_unicode_ci")
- in := strings.Repeat(`0`, maxAllowedPacketSize+1)
- var out string
- var rows *sql.Rows
- // Long text data
- const nonDataQueryLen = 28 // length query w/o value
- inS := in[:maxAllowedPacketSize-nonDataQueryLen]
- mustExec(t, db, "INSERT INTO test VALUES('"+inS+"')")
- rows = mustQuery(t, db, "SELECT value FROM test")
- if rows.Next() {
- rows.Scan(&out)
- if inS != out {
- t.Fatalf("LONGBLOB: length in: %d, length out: %d", len(inS), len(out))
- }
- if rows.Next() {
- t.Error("LONGBLOB: unexpexted row")
- }
- } else {
- t.Fatalf("LONGBLOB: no data")
- }
- // Empty table
- mustExec(t, db, "TRUNCATE TABLE test")
- // Long binary data
- mustExec(t, db, "INSERT INTO test VALUES(?)", in)
- rows = mustQuery(t, db, "SELECT value FROM test WHERE 1=?", 1)
- if rows.Next() {
- rows.Scan(&out)
- if in != out {
- t.Fatalf("LONGBLOB: length in: %d, length out: %d", len(in), len(out))
- }
- if rows.Next() {
- t.Error("LONGBLOB: unexpexted row")
- }
- } else {
- t.Fatalf("LONGBLOB: no data")
- }
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- }
- func verifyLoadDataResult(t *testing.T, db *sql.DB) {
- rows, err := db.Query("SELECT * FROM test")
- if err != nil {
- t.Fatal(err.Error())
- }
- i := 0
- values := [4]string{
- "a string",
- "a string containing a \t",
- "a string containing a \n",
- "a string containing both \t\n",
- }
- var id int
- var value string
- for rows.Next() {
- i++
- err = rows.Scan(&id, &value)
- if err != nil {
- t.Fatal(err.Error())
- }
- if i != id {
- t.Fatalf("%d != %d", i, id)
- }
- if values[i-1] != value {
- t.Fatalf("%s != %s", values[i-1], value)
- }
- }
- err = rows.Err()
- if err != nil {
- t.Fatal(err.Error())
- }
- if i != 4 {
- t.Fatalf("Rows count mismatch. Got %d, want 4", i)
- }
- }
- func TestLoadData(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestLoadData", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- file, err := ioutil.TempFile("", "gotest")
- defer os.Remove(file.Name())
- if err != nil {
- t.Fatal(err)
- }
- 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")
- file.Close()
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- mustExec(t, db, "CREATE TABLE test (id INT NOT NULL PRIMARY KEY, value TEXT NOT NULL) CHARACTER SET utf8 COLLATE utf8_unicode_ci")
- // Local File
- RegisterLocalFile(file.Name())
- mustExec(t, db, "LOAD DATA LOCAL INFILE '"+file.Name()+"' INTO TABLE test")
- verifyLoadDataResult(t, db)
- // negative test
- _, err = db.Exec("LOAD DATA LOCAL INFILE 'doesnotexist' INTO TABLE test")
- if err == nil {
- t.Fatal("Load non-existent file didn't fail")
- } else if err.Error() != "Local File 'doesnotexist' is not registered. Use the DSN parameter 'allowAllFiles=true' to allow all files" {
- t.Fatal(err.Error())
- }
- // Empty table
- mustExec(t, db, "TRUNCATE TABLE test")
- // Reader
- file, err = os.Open(file.Name())
- if err != nil {
- t.Fatal(err)
- }
- RegisterReader("test", file)
- mustExec(t, db, "LOAD DATA LOCAL INFILE 'Reader::test' INTO TABLE test")
- verifyLoadDataResult(t, db)
- // negative test
- _, err = db.Exec("LOAD DATA LOCAL INFILE 'Reader::doesnotexist' INTO TABLE test")
- if err == nil {
- t.Fatal("Load non-existent Reader didn't fail")
- } else if err.Error() != "Reader 'doesnotexist' is not registered" {
- t.Fatal(err.Error())
- }
- file.Close()
- mustExec(t, db, "DROP TABLE IF EXISTS test")
- }
- // Special cases
- func TestRowsClose(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestRowsClose", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- rows, err := db.Query("SELECT 1")
- if err != nil {
- t.Fatal(err)
- }
- err = rows.Close()
- if err != nil {
- t.Fatal(err)
- }
- if rows.Next() {
- t.Fatal("Unexpected row after rows.Close()")
- }
- err = rows.Err()
- if err != nil {
- t.Fatal(err)
- }
- }
- // dangling statements
- // http://code.google.com/p/go/issues/detail?id=3865
- func TestCloseStmtBeforeRows(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestCloseStmtBeforeRows", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- stmt, err := db.Prepare("SELECT 1")
- if err != nil {
- t.Fatal(err)
- }
- rows, err := stmt.Query()
- if err != nil {
- stmt.Close()
- t.Fatal(err)
- }
- defer rows.Close()
- err = stmt.Close()
- if err != nil {
- t.Fatal(err)
- }
- if !rows.Next() {
- t.Fatal("Getting row failed")
- } else {
- err = rows.Err()
- if err != nil {
- t.Fatal(err)
- }
- var out bool
- err = rows.Scan(&out)
- if err != nil {
- t.Fatalf("Error on rows.Scan(): %v", err)
- }
- if out != true {
- t.Errorf("true != %t", out)
- }
- }
- }
- // It is valid to have multiple Rows for the same Stmt
- // http://code.google.com/p/go/issues/detail?id=3734
- func TestStmtMultiRows(t *testing.T) {
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestStmtMultiRows", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- stmt, err := db.Prepare("SELECT 1 UNION SELECT 0")
- if err != nil {
- t.Fatal(err)
- }
- rows1, err := stmt.Query()
- if err != nil {
- stmt.Close()
- t.Fatal(err)
- }
- defer rows1.Close()
- rows2, err := stmt.Query()
- if err != nil {
- stmt.Close()
- t.Fatal(err)
- }
- defer rows2.Close()
- var out bool
- // 1
- if !rows1.Next() {
- t.Fatal("1st rows1.Next failed")
- } else {
- err = rows1.Err()
- if err != nil {
- t.Fatal(err)
- }
- err = rows1.Scan(&out)
- if err != nil {
- t.Fatalf("Error on rows.Scan(): %v", err)
- }
- if out != true {
- t.Errorf("true != %t", out)
- }
- }
- if !rows2.Next() {
- t.Fatal("1st rows2.Next failed")
- } else {
- err = rows2.Err()
- if err != nil {
- t.Fatal(err)
- }
- err = rows2.Scan(&out)
- if err != nil {
- t.Fatalf("Error on rows.Scan(): %v", err)
- }
- if out != true {
- t.Errorf("true != %t", out)
- }
- }
- // 2
- if !rows1.Next() {
- t.Fatal("2nd rows1.Next failed")
- } else {
- err = rows1.Err()
- if err != nil {
- t.Fatal(err)
- }
- err = rows1.Scan(&out)
- if err != nil {
- t.Fatalf("Error on rows.Scan(): %v", err)
- }
- if out != false {
- t.Errorf("false != %t", out)
- }
- if rows1.Next() {
- t.Fatal("Unexpected row on rows1")
- }
- err = rows1.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- if !rows2.Next() {
- t.Fatal("2nd rows2.Next failed")
- } else {
- err = rows2.Err()
- if err != nil {
- t.Fatal(err)
- }
- err = rows2.Scan(&out)
- if err != nil {
- t.Fatalf("Error on rows.Scan(): %v", err)
- }
- if out != false {
- t.Errorf("false != %t", out)
- }
- if rows2.Next() {
- t.Fatal("Unexpected row on rows2")
- }
- err = rows2.Close()
- if err != nil {
- t.Fatal(err)
- }
- }
- }
- func TestConcurrent(t *testing.T) {
- if os.Getenv("MYSQL_TEST_CONCURRENT") != "1" {
- t.Log("CONCURRENT env var not set. Skipping TestConcurrent")
- return
- }
- if !getEnv() {
- t.Logf("MySQL-Server not running on %s. Skipping TestConcurrent", netAddr)
- return
- }
- db, err := sql.Open("mysql", dsn)
- if err != nil {
- t.Fatalf("Error connecting: %v", err)
- }
- defer db.Close()
- var max int
- err = db.QueryRow("SELECT @@max_connections").Scan(&max)
- if err != nil {
- t.Fatalf("%v", err)
- }
- t.Logf("Testing %d concurrent connections \r\n", max)
- canStop := false
- c := make(chan struct{}, max)
- for i := 0; i < max; i++ {
- go func() {
- tx, err := db.Begin()
- if err != nil {
- canStop = true
- t.Fatalf("Error on Con %d: %s", i, err.Error())
- }
- c <- struct{}{}
- for !canStop {
- _, err := tx.Exec("SELECT 1")
- if err != nil {
- canStop = true
- t.Fatalf("Error on Con %d: %s", i, err.Error())
- }
- }
- err = tx.Commit()
- if err != nil {
- canStop = true
- t.Fatalf("Error on Con %d: %s", i, err.Error())
- }
- }()
- }
- for i := 0; i < max; i++ {
- <-c
- }
- canStop = true
- }
|