driver_test.go 77 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862
  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. "bytes"
  11. "context"
  12. "crypto/tls"
  13. "database/sql"
  14. "database/sql/driver"
  15. "fmt"
  16. "io"
  17. "io/ioutil"
  18. "log"
  19. "math"
  20. "net"
  21. "net/url"
  22. "os"
  23. "reflect"
  24. "strings"
  25. "sync"
  26. "sync/atomic"
  27. "testing"
  28. "time"
  29. )
  30. // Ensure that all the driver interfaces are implemented
  31. var (
  32. _ driver.Rows = &binaryRows{}
  33. _ driver.Rows = &textRows{}
  34. )
  35. var (
  36. user string
  37. pass string
  38. prot string
  39. addr string
  40. dbname string
  41. dsn string
  42. netAddr string
  43. available bool
  44. )
  45. var (
  46. tDate = time.Date(2012, 6, 14, 0, 0, 0, 0, time.UTC)
  47. sDate = "2012-06-14"
  48. tDateTime = time.Date(2011, 11, 20, 21, 27, 37, 0, time.UTC)
  49. sDateTime = "2011-11-20 21:27:37"
  50. tDate0 = time.Time{}
  51. sDate0 = "0000-00-00"
  52. sDateTime0 = "0000-00-00 00:00:00"
  53. )
  54. // See https://github.com/go-sql-driver/mysql/wiki/Testing
  55. func init() {
  56. // get environment variables
  57. env := func(key, defaultValue string) string {
  58. if value := os.Getenv(key); value != "" {
  59. return value
  60. }
  61. return defaultValue
  62. }
  63. user = env("MYSQL_TEST_USER", "root")
  64. pass = env("MYSQL_TEST_PASS", "")
  65. prot = env("MYSQL_TEST_PROT", "tcp")
  66. addr = env("MYSQL_TEST_ADDR", "localhost:3306")
  67. dbname = env("MYSQL_TEST_DBNAME", "gotest")
  68. netAddr = fmt.Sprintf("%s(%s)", prot, addr)
  69. dsn = fmt.Sprintf("%s:%s@%s/%s?timeout=30s", user, pass, netAddr, dbname)
  70. c, err := net.Dial(prot, addr)
  71. if err == nil {
  72. available = true
  73. c.Close()
  74. }
  75. }
  76. type DBTest struct {
  77. *testing.T
  78. db *sql.DB
  79. }
  80. func runTestsWithMultiStatement(t *testing.T, dsn string, tests ...func(dbt *DBTest)) {
  81. if !available {
  82. t.Skipf("MySQL server not running on %s", netAddr)
  83. }
  84. dsn += "&multiStatements=true"
  85. var db *sql.DB
  86. if _, err := ParseDSN(dsn); err != errInvalidDSNUnsafeCollation {
  87. db, err = sql.Open("mysql", dsn)
  88. if err != nil {
  89. t.Fatalf("error connecting: %s", err.Error())
  90. }
  91. defer db.Close()
  92. }
  93. dbt := &DBTest{t, db}
  94. for _, test := range tests {
  95. test(dbt)
  96. dbt.db.Exec("DROP TABLE IF EXISTS test")
  97. }
  98. }
  99. func runTests(t *testing.T, dsn string, tests ...func(dbt *DBTest)) {
  100. if !available {
  101. t.Skipf("MySQL server not running on %s", netAddr)
  102. }
  103. db, err := sql.Open("mysql", dsn)
  104. if err != nil {
  105. t.Fatalf("error connecting: %s", err.Error())
  106. }
  107. defer db.Close()
  108. db.Exec("DROP TABLE IF EXISTS test")
  109. dsn2 := dsn + "&interpolateParams=true"
  110. var db2 *sql.DB
  111. if _, err := ParseDSN(dsn2); err != errInvalidDSNUnsafeCollation {
  112. db2, err = sql.Open("mysql", dsn2)
  113. if err != nil {
  114. t.Fatalf("error connecting: %s", err.Error())
  115. }
  116. defer db2.Close()
  117. }
  118. dsn3 := dsn + "&multiStatements=true"
  119. var db3 *sql.DB
  120. if _, err := ParseDSN(dsn3); err != errInvalidDSNUnsafeCollation {
  121. db3, err = sql.Open("mysql", dsn3)
  122. if err != nil {
  123. t.Fatalf("error connecting: %s", err.Error())
  124. }
  125. defer db3.Close()
  126. }
  127. dbt := &DBTest{t, db}
  128. dbt2 := &DBTest{t, db2}
  129. dbt3 := &DBTest{t, db3}
  130. for _, test := range tests {
  131. test(dbt)
  132. dbt.db.Exec("DROP TABLE IF EXISTS test")
  133. if db2 != nil {
  134. test(dbt2)
  135. dbt2.db.Exec("DROP TABLE IF EXISTS test")
  136. }
  137. if db3 != nil {
  138. test(dbt3)
  139. dbt3.db.Exec("DROP TABLE IF EXISTS test")
  140. }
  141. }
  142. }
  143. func (dbt *DBTest) fail(method, query string, err error) {
  144. if len(query) > 300 {
  145. query = "[query too large to print]"
  146. }
  147. dbt.Fatalf("error on %s %s: %s", method, query, err.Error())
  148. }
  149. func (dbt *DBTest) mustExec(query string, args ...interface{}) (res sql.Result) {
  150. res, err := dbt.db.Exec(query, args...)
  151. if err != nil {
  152. dbt.fail("exec", query, err)
  153. }
  154. return res
  155. }
  156. func (dbt *DBTest) mustQuery(query string, args ...interface{}) (rows *sql.Rows) {
  157. rows, err := dbt.db.Query(query, args...)
  158. if err != nil {
  159. dbt.fail("query", query, err)
  160. }
  161. return rows
  162. }
  163. func maybeSkip(t *testing.T, err error, skipErrno uint16) {
  164. mySQLErr, ok := err.(*MySQLError)
  165. if !ok {
  166. return
  167. }
  168. if mySQLErr.Number == skipErrno {
  169. t.Skipf("skipping test for error: %v", err)
  170. }
  171. }
  172. func TestEmptyQuery(t *testing.T) {
  173. runTests(t, dsn, func(dbt *DBTest) {
  174. // just a comment, no query
  175. rows := dbt.mustQuery("--")
  176. // will hang before #255
  177. if rows.Next() {
  178. dbt.Errorf("next on rows must be false")
  179. }
  180. })
  181. }
  182. func TestCRUD(t *testing.T) {
  183. runTests(t, dsn, func(dbt *DBTest) {
  184. // Create Table
  185. dbt.mustExec("CREATE TABLE test (value BOOL)")
  186. // Test for unexpected data
  187. var out bool
  188. rows := dbt.mustQuery("SELECT * FROM test")
  189. if rows.Next() {
  190. dbt.Error("unexpected data in empty table")
  191. }
  192. // Create Data
  193. res := dbt.mustExec("INSERT INTO test VALUES (1)")
  194. count, err := res.RowsAffected()
  195. if err != nil {
  196. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  197. }
  198. if count != 1 {
  199. dbt.Fatalf("expected 1 affected row, got %d", count)
  200. }
  201. id, err := res.LastInsertId()
  202. if err != nil {
  203. dbt.Fatalf("res.LastInsertId() returned error: %s", err.Error())
  204. }
  205. if id != 0 {
  206. dbt.Fatalf("expected InsertId 0, got %d", id)
  207. }
  208. // Read
  209. rows = dbt.mustQuery("SELECT value FROM test")
  210. if rows.Next() {
  211. rows.Scan(&out)
  212. if true != out {
  213. dbt.Errorf("true != %t", out)
  214. }
  215. if rows.Next() {
  216. dbt.Error("unexpected data")
  217. }
  218. } else {
  219. dbt.Error("no data")
  220. }
  221. // Update
  222. res = dbt.mustExec("UPDATE test SET value = ? WHERE value = ?", false, true)
  223. count, err = res.RowsAffected()
  224. if err != nil {
  225. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  226. }
  227. if count != 1 {
  228. dbt.Fatalf("expected 1 affected row, got %d", count)
  229. }
  230. // Check Update
  231. rows = dbt.mustQuery("SELECT value FROM test")
  232. if rows.Next() {
  233. rows.Scan(&out)
  234. if false != out {
  235. dbt.Errorf("false != %t", out)
  236. }
  237. if rows.Next() {
  238. dbt.Error("unexpected data")
  239. }
  240. } else {
  241. dbt.Error("no data")
  242. }
  243. // Delete
  244. res = dbt.mustExec("DELETE FROM test WHERE value = ?", false)
  245. count, err = res.RowsAffected()
  246. if err != nil {
  247. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  248. }
  249. if count != 1 {
  250. dbt.Fatalf("expected 1 affected row, got %d", count)
  251. }
  252. // Check for unexpected rows
  253. res = dbt.mustExec("DELETE FROM test")
  254. count, err = res.RowsAffected()
  255. if err != nil {
  256. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  257. }
  258. if count != 0 {
  259. dbt.Fatalf("expected 0 affected row, got %d", count)
  260. }
  261. })
  262. }
  263. func TestMultiQuery(t *testing.T) {
  264. runTestsWithMultiStatement(t, dsn, func(dbt *DBTest) {
  265. // Create Table
  266. dbt.mustExec("CREATE TABLE `test` (`id` int(11) NOT NULL, `value` int(11) NOT NULL) ")
  267. // Create Data
  268. res := dbt.mustExec("INSERT INTO test VALUES (1, 1)")
  269. count, err := res.RowsAffected()
  270. if err != nil {
  271. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  272. }
  273. if count != 1 {
  274. dbt.Fatalf("expected 1 affected row, got %d", count)
  275. }
  276. // Update
  277. res = dbt.mustExec("UPDATE test SET value = 3 WHERE id = 1; UPDATE test SET value = 4 WHERE id = 1; UPDATE test SET value = 5 WHERE id = 1;")
  278. count, err = res.RowsAffected()
  279. if err != nil {
  280. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  281. }
  282. if count != 1 {
  283. dbt.Fatalf("expected 1 affected row, got %d", count)
  284. }
  285. // Read
  286. var out int
  287. rows := dbt.mustQuery("SELECT value FROM test WHERE id=1;")
  288. if rows.Next() {
  289. rows.Scan(&out)
  290. if 5 != out {
  291. dbt.Errorf("5 != %d", out)
  292. }
  293. if rows.Next() {
  294. dbt.Error("unexpected data")
  295. }
  296. } else {
  297. dbt.Error("no data")
  298. }
  299. })
  300. }
  301. func TestInt(t *testing.T) {
  302. runTests(t, dsn, func(dbt *DBTest) {
  303. types := [5]string{"TINYINT", "SMALLINT", "MEDIUMINT", "INT", "BIGINT"}
  304. in := int64(42)
  305. var out int64
  306. var rows *sql.Rows
  307. // SIGNED
  308. for _, v := range types {
  309. dbt.mustExec("CREATE TABLE test (value " + v + ")")
  310. dbt.mustExec("INSERT INTO test VALUES (?)", in)
  311. rows = dbt.mustQuery("SELECT value FROM test")
  312. if rows.Next() {
  313. rows.Scan(&out)
  314. if in != out {
  315. dbt.Errorf("%s: %d != %d", v, in, out)
  316. }
  317. } else {
  318. dbt.Errorf("%s: no data", v)
  319. }
  320. dbt.mustExec("DROP TABLE IF EXISTS test")
  321. }
  322. // UNSIGNED ZEROFILL
  323. for _, v := range types {
  324. dbt.mustExec("CREATE TABLE test (value " + v + " ZEROFILL)")
  325. dbt.mustExec("INSERT INTO test VALUES (?)", in)
  326. rows = dbt.mustQuery("SELECT value FROM test")
  327. if rows.Next() {
  328. rows.Scan(&out)
  329. if in != out {
  330. dbt.Errorf("%s ZEROFILL: %d != %d", v, in, out)
  331. }
  332. } else {
  333. dbt.Errorf("%s ZEROFILL: no data", v)
  334. }
  335. dbt.mustExec("DROP TABLE IF EXISTS test")
  336. }
  337. })
  338. }
  339. func TestFloat32(t *testing.T) {
  340. runTests(t, dsn, func(dbt *DBTest) {
  341. types := [2]string{"FLOAT", "DOUBLE"}
  342. in := float32(42.23)
  343. var out float32
  344. var rows *sql.Rows
  345. for _, v := range types {
  346. dbt.mustExec("CREATE TABLE test (value " + v + ")")
  347. dbt.mustExec("INSERT INTO test VALUES (?)", in)
  348. rows = dbt.mustQuery("SELECT value FROM test")
  349. if rows.Next() {
  350. rows.Scan(&out)
  351. if in != out {
  352. dbt.Errorf("%s: %g != %g", v, in, out)
  353. }
  354. } else {
  355. dbt.Errorf("%s: no data", v)
  356. }
  357. dbt.mustExec("DROP TABLE IF EXISTS test")
  358. }
  359. })
  360. }
  361. func TestFloat64(t *testing.T) {
  362. runTests(t, dsn, func(dbt *DBTest) {
  363. types := [2]string{"FLOAT", "DOUBLE"}
  364. var expected float64 = 42.23
  365. var out float64
  366. var rows *sql.Rows
  367. for _, v := range types {
  368. dbt.mustExec("CREATE TABLE test (value " + v + ")")
  369. dbt.mustExec("INSERT INTO test VALUES (42.23)")
  370. rows = dbt.mustQuery("SELECT value FROM test")
  371. if rows.Next() {
  372. rows.Scan(&out)
  373. if expected != out {
  374. dbt.Errorf("%s: %g != %g", v, expected, out)
  375. }
  376. } else {
  377. dbt.Errorf("%s: no data", v)
  378. }
  379. dbt.mustExec("DROP TABLE IF EXISTS test")
  380. }
  381. })
  382. }
  383. func TestFloat64Placeholder(t *testing.T) {
  384. runTests(t, dsn, func(dbt *DBTest) {
  385. types := [2]string{"FLOAT", "DOUBLE"}
  386. var expected float64 = 42.23
  387. var out float64
  388. var rows *sql.Rows
  389. for _, v := range types {
  390. dbt.mustExec("CREATE TABLE test (id int, value " + v + ")")
  391. dbt.mustExec("INSERT INTO test VALUES (1, 42.23)")
  392. rows = dbt.mustQuery("SELECT value FROM test WHERE id = ?", 1)
  393. if rows.Next() {
  394. rows.Scan(&out)
  395. if expected != out {
  396. dbt.Errorf("%s: %g != %g", v, expected, out)
  397. }
  398. } else {
  399. dbt.Errorf("%s: no data", v)
  400. }
  401. dbt.mustExec("DROP TABLE IF EXISTS test")
  402. }
  403. })
  404. }
  405. func TestString(t *testing.T) {
  406. runTests(t, dsn, func(dbt *DBTest) {
  407. types := [6]string{"CHAR(255)", "VARCHAR(255)", "TINYTEXT", "TEXT", "MEDIUMTEXT", "LONGTEXT"}
  408. in := "κόσμε üöäßñóùéàâÿœ'îë Árvíztűrő いろはにほへとちりぬるを イロハニホヘト דג סקרן чащах น่าฟังเอย"
  409. var out string
  410. var rows *sql.Rows
  411. for _, v := range types {
  412. dbt.mustExec("CREATE TABLE test (value " + v + ") CHARACTER SET utf8")
  413. dbt.mustExec("INSERT INTO test VALUES (?)", in)
  414. rows = dbt.mustQuery("SELECT value FROM test")
  415. if rows.Next() {
  416. rows.Scan(&out)
  417. if in != out {
  418. dbt.Errorf("%s: %s != %s", v, in, out)
  419. }
  420. } else {
  421. dbt.Errorf("%s: no data", v)
  422. }
  423. dbt.mustExec("DROP TABLE IF EXISTS test")
  424. }
  425. // BLOB
  426. dbt.mustExec("CREATE TABLE test (id int, value BLOB) CHARACTER SET utf8")
  427. id := 2
  428. in = "Lorem ipsum dolor sit amet, consetetur sadipscing elitr, " +
  429. "sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, " +
  430. "sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. " +
  431. "Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet. " +
  432. "Lorem ipsum dolor sit amet, consetetur sadipscing elitr, " +
  433. "sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, " +
  434. "sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. " +
  435. "Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet."
  436. dbt.mustExec("INSERT INTO test VALUES (?, ?)", id, in)
  437. err := dbt.db.QueryRow("SELECT value FROM test WHERE id = ?", id).Scan(&out)
  438. if err != nil {
  439. dbt.Fatalf("Error on BLOB-Query: %s", err.Error())
  440. } else if out != in {
  441. dbt.Errorf("BLOB: %s != %s", in, out)
  442. }
  443. })
  444. }
  445. func TestRawBytes(t *testing.T) {
  446. runTests(t, dsn, func(dbt *DBTest) {
  447. v1 := []byte("aaa")
  448. v2 := []byte("bbb")
  449. rows := dbt.mustQuery("SELECT ?, ?", v1, v2)
  450. if rows.Next() {
  451. var o1, o2 sql.RawBytes
  452. if err := rows.Scan(&o1, &o2); err != nil {
  453. dbt.Errorf("Got error: %v", err)
  454. }
  455. if !bytes.Equal(v1, o1) {
  456. dbt.Errorf("expected %v, got %v", v1, o1)
  457. }
  458. if !bytes.Equal(v2, o2) {
  459. dbt.Errorf("expected %v, got %v", v2, o2)
  460. }
  461. // https://github.com/go-sql-driver/mysql/issues/765
  462. // Appending to RawBytes shouldn't overwrite next RawBytes.
  463. o1 = append(o1, "xyzzy"...)
  464. if !bytes.Equal(v2, o2) {
  465. dbt.Errorf("expected %v, got %v", v2, o2)
  466. }
  467. } else {
  468. dbt.Errorf("no data")
  469. }
  470. })
  471. }
  472. type testValuer struct {
  473. value string
  474. }
  475. func (tv testValuer) Value() (driver.Value, error) {
  476. return tv.value, nil
  477. }
  478. func TestValuer(t *testing.T) {
  479. runTests(t, dsn, func(dbt *DBTest) {
  480. in := testValuer{"a_value"}
  481. var out string
  482. var rows *sql.Rows
  483. dbt.mustExec("CREATE TABLE test (value VARCHAR(255)) CHARACTER SET utf8")
  484. dbt.mustExec("INSERT INTO test VALUES (?)", in)
  485. rows = dbt.mustQuery("SELECT value FROM test")
  486. if rows.Next() {
  487. rows.Scan(&out)
  488. if in.value != out {
  489. dbt.Errorf("Valuer: %v != %s", in, out)
  490. }
  491. } else {
  492. dbt.Errorf("Valuer: no data")
  493. }
  494. dbt.mustExec("DROP TABLE IF EXISTS test")
  495. })
  496. }
  497. type testValuerWithValidation struct {
  498. value string
  499. }
  500. func (tv testValuerWithValidation) Value() (driver.Value, error) {
  501. if len(tv.value) == 0 {
  502. return nil, fmt.Errorf("Invalid string valuer. Value must not be empty")
  503. }
  504. return tv.value, nil
  505. }
  506. func TestValuerWithValidation(t *testing.T) {
  507. runTests(t, dsn, func(dbt *DBTest) {
  508. in := testValuerWithValidation{"a_value"}
  509. var out string
  510. var rows *sql.Rows
  511. dbt.mustExec("CREATE TABLE testValuer (value VARCHAR(255)) CHARACTER SET utf8")
  512. dbt.mustExec("INSERT INTO testValuer VALUES (?)", in)
  513. rows = dbt.mustQuery("SELECT value FROM testValuer")
  514. defer rows.Close()
  515. if rows.Next() {
  516. rows.Scan(&out)
  517. if in.value != out {
  518. dbt.Errorf("Valuer: %v != %s", in, out)
  519. }
  520. } else {
  521. dbt.Errorf("Valuer: no data")
  522. }
  523. if _, err := dbt.db.Exec("INSERT INTO testValuer VALUES (?)", testValuerWithValidation{""}); err == nil {
  524. dbt.Errorf("Failed to check valuer error")
  525. }
  526. if _, err := dbt.db.Exec("INSERT INTO testValuer VALUES (?)", nil); err != nil {
  527. dbt.Errorf("Failed to check nil")
  528. }
  529. if _, err := dbt.db.Exec("INSERT INTO testValuer VALUES (?)", map[string]bool{}); err == nil {
  530. dbt.Errorf("Failed to check not valuer")
  531. }
  532. dbt.mustExec("DROP TABLE IF EXISTS testValuer")
  533. })
  534. }
  535. type timeTests struct {
  536. dbtype string
  537. tlayout string
  538. tests []timeTest
  539. }
  540. type timeTest struct {
  541. s string // leading "!": do not use t as value in queries
  542. t time.Time
  543. }
  544. type timeMode byte
  545. func (t timeMode) String() string {
  546. switch t {
  547. case binaryString:
  548. return "binary:string"
  549. case binaryTime:
  550. return "binary:time.Time"
  551. case textString:
  552. return "text:string"
  553. }
  554. panic("unsupported timeMode")
  555. }
  556. func (t timeMode) Binary() bool {
  557. switch t {
  558. case binaryString, binaryTime:
  559. return true
  560. }
  561. return false
  562. }
  563. const (
  564. binaryString timeMode = iota
  565. binaryTime
  566. textString
  567. )
  568. func (t timeTest) genQuery(dbtype string, mode timeMode) string {
  569. var inner string
  570. if mode.Binary() {
  571. inner = "?"
  572. } else {
  573. inner = `"%s"`
  574. }
  575. return `SELECT cast(` + inner + ` as ` + dbtype + `)`
  576. }
  577. func (t timeTest) run(dbt *DBTest, dbtype, tlayout string, mode timeMode) {
  578. var rows *sql.Rows
  579. query := t.genQuery(dbtype, mode)
  580. switch mode {
  581. case binaryString:
  582. rows = dbt.mustQuery(query, t.s)
  583. case binaryTime:
  584. rows = dbt.mustQuery(query, t.t)
  585. case textString:
  586. query = fmt.Sprintf(query, t.s)
  587. rows = dbt.mustQuery(query)
  588. default:
  589. panic("unsupported mode")
  590. }
  591. defer rows.Close()
  592. var err error
  593. if !rows.Next() {
  594. err = rows.Err()
  595. if err == nil {
  596. err = fmt.Errorf("no data")
  597. }
  598. dbt.Errorf("%s [%s]: %s", dbtype, mode, err)
  599. return
  600. }
  601. var dst interface{}
  602. err = rows.Scan(&dst)
  603. if err != nil {
  604. dbt.Errorf("%s [%s]: %s", dbtype, mode, err)
  605. return
  606. }
  607. switch val := dst.(type) {
  608. case []uint8:
  609. str := string(val)
  610. if str == t.s {
  611. return
  612. }
  613. if mode.Binary() && dbtype == "DATETIME" && len(str) == 26 && str[:19] == t.s {
  614. // a fix mainly for TravisCI:
  615. // accept full microsecond resolution in result for DATETIME columns
  616. // where the binary protocol was used
  617. return
  618. }
  619. dbt.Errorf("%s [%s] to string: expected %q, got %q",
  620. dbtype, mode,
  621. t.s, str,
  622. )
  623. case time.Time:
  624. if val == t.t {
  625. return
  626. }
  627. dbt.Errorf("%s [%s] to string: expected %q, got %q",
  628. dbtype, mode,
  629. t.s, val.Format(tlayout),
  630. )
  631. default:
  632. fmt.Printf("%#v\n", []interface{}{dbtype, tlayout, mode, t.s, t.t})
  633. dbt.Errorf("%s [%s]: unhandled type %T (is '%v')",
  634. dbtype, mode,
  635. val, val,
  636. )
  637. }
  638. }
  639. func TestDateTime(t *testing.T) {
  640. afterTime := func(t time.Time, d string) time.Time {
  641. dur, err := time.ParseDuration(d)
  642. if err != nil {
  643. panic(err)
  644. }
  645. return t.Add(dur)
  646. }
  647. // NOTE: MySQL rounds DATETIME(x) up - but that's not included in the tests
  648. format := "2006-01-02 15:04:05.999999"
  649. t0 := time.Time{}
  650. tstr0 := "0000-00-00 00:00:00.000000"
  651. testcases := []timeTests{
  652. {"DATE", format[:10], []timeTest{
  653. {t: time.Date(2011, 11, 20, 0, 0, 0, 0, time.UTC)},
  654. {t: t0, s: tstr0[:10]},
  655. }},
  656. {"DATETIME", format[:19], []timeTest{
  657. {t: time.Date(2011, 11, 20, 21, 27, 37, 0, time.UTC)},
  658. {t: t0, s: tstr0[:19]},
  659. }},
  660. {"DATETIME(0)", format[:21], []timeTest{
  661. {t: time.Date(2011, 11, 20, 21, 27, 37, 0, time.UTC)},
  662. {t: t0, s: tstr0[:19]},
  663. }},
  664. {"DATETIME(1)", format[:21], []timeTest{
  665. {t: time.Date(2011, 11, 20, 21, 27, 37, 100000000, time.UTC)},
  666. {t: t0, s: tstr0[:21]},
  667. }},
  668. {"DATETIME(6)", format, []timeTest{
  669. {t: time.Date(2011, 11, 20, 21, 27, 37, 123456000, time.UTC)},
  670. {t: t0, s: tstr0},
  671. }},
  672. {"TIME", format[11:19], []timeTest{
  673. {t: afterTime(t0, "12345s")},
  674. {s: "!-12:34:56"},
  675. {s: "!-838:59:59"},
  676. {s: "!838:59:59"},
  677. {t: t0, s: tstr0[11:19]},
  678. }},
  679. {"TIME(0)", format[11:19], []timeTest{
  680. {t: afterTime(t0, "12345s")},
  681. {s: "!-12:34:56"},
  682. {s: "!-838:59:59"},
  683. {s: "!838:59:59"},
  684. {t: t0, s: tstr0[11:19]},
  685. }},
  686. {"TIME(1)", format[11:21], []timeTest{
  687. {t: afterTime(t0, "12345600ms")},
  688. {s: "!-12:34:56.7"},
  689. {s: "!-838:59:58.9"},
  690. {s: "!838:59:58.9"},
  691. {t: t0, s: tstr0[11:21]},
  692. }},
  693. {"TIME(6)", format[11:], []timeTest{
  694. {t: afterTime(t0, "1234567890123000ns")},
  695. {s: "!-12:34:56.789012"},
  696. {s: "!-838:59:58.999999"},
  697. {s: "!838:59:58.999999"},
  698. {t: t0, s: tstr0[11:]},
  699. }},
  700. }
  701. dsns := []string{
  702. dsn + "&parseTime=true",
  703. dsn + "&parseTime=false",
  704. }
  705. for _, testdsn := range dsns {
  706. runTests(t, testdsn, func(dbt *DBTest) {
  707. microsecsSupported := false
  708. zeroDateSupported := false
  709. var rows *sql.Rows
  710. var err error
  711. rows, err = dbt.db.Query(`SELECT cast("00:00:00.1" as TIME(1)) = "00:00:00.1"`)
  712. if err == nil {
  713. rows.Scan(&microsecsSupported)
  714. rows.Close()
  715. }
  716. rows, err = dbt.db.Query(`SELECT cast("0000-00-00" as DATE) = "0000-00-00"`)
  717. if err == nil {
  718. rows.Scan(&zeroDateSupported)
  719. rows.Close()
  720. }
  721. for _, setups := range testcases {
  722. if t := setups.dbtype; !microsecsSupported && t[len(t)-1:] == ")" {
  723. // skip fractional second tests if unsupported by server
  724. continue
  725. }
  726. for _, setup := range setups.tests {
  727. allowBinTime := true
  728. if setup.s == "" {
  729. // fill time string wherever Go can reliable produce it
  730. setup.s = setup.t.Format(setups.tlayout)
  731. } else if setup.s[0] == '!' {
  732. // skip tests using setup.t as source in queries
  733. allowBinTime = false
  734. // fix setup.s - remove the "!"
  735. setup.s = setup.s[1:]
  736. }
  737. if !zeroDateSupported && setup.s == tstr0[:len(setup.s)] {
  738. // skip disallowed 0000-00-00 date
  739. continue
  740. }
  741. setup.run(dbt, setups.dbtype, setups.tlayout, textString)
  742. setup.run(dbt, setups.dbtype, setups.tlayout, binaryString)
  743. if allowBinTime {
  744. setup.run(dbt, setups.dbtype, setups.tlayout, binaryTime)
  745. }
  746. }
  747. }
  748. })
  749. }
  750. }
  751. func TestTimestampMicros(t *testing.T) {
  752. format := "2006-01-02 15:04:05.999999"
  753. f0 := format[:19]
  754. f1 := format[:21]
  755. f6 := format[:26]
  756. runTests(t, dsn, func(dbt *DBTest) {
  757. // check if microseconds are supported.
  758. // Do not use timestamp(x) for that check - before 5.5.6, x would mean display width
  759. // and not precision.
  760. // Se last paragraph at http://dev.mysql.com/doc/refman/5.6/en/fractional-seconds.html
  761. microsecsSupported := false
  762. if rows, err := dbt.db.Query(`SELECT cast("00:00:00.1" as TIME(1)) = "00:00:00.1"`); err == nil {
  763. rows.Scan(&microsecsSupported)
  764. rows.Close()
  765. }
  766. if !microsecsSupported {
  767. // skip test
  768. return
  769. }
  770. _, err := dbt.db.Exec(`
  771. CREATE TABLE test (
  772. value0 TIMESTAMP NOT NULL DEFAULT '` + f0 + `',
  773. value1 TIMESTAMP(1) NOT NULL DEFAULT '` + f1 + `',
  774. value6 TIMESTAMP(6) NOT NULL DEFAULT '` + f6 + `'
  775. )`,
  776. )
  777. if err != nil {
  778. dbt.Error(err)
  779. }
  780. defer dbt.mustExec("DROP TABLE IF EXISTS test")
  781. dbt.mustExec("INSERT INTO test SET value0=?, value1=?, value6=?", f0, f1, f6)
  782. var res0, res1, res6 string
  783. rows := dbt.mustQuery("SELECT * FROM test")
  784. if !rows.Next() {
  785. dbt.Errorf("test contained no selectable values")
  786. }
  787. err = rows.Scan(&res0, &res1, &res6)
  788. if err != nil {
  789. dbt.Error(err)
  790. }
  791. if res0 != f0 {
  792. dbt.Errorf("expected %q, got %q", f0, res0)
  793. }
  794. if res1 != f1 {
  795. dbt.Errorf("expected %q, got %q", f1, res1)
  796. }
  797. if res6 != f6 {
  798. dbt.Errorf("expected %q, got %q", f6, res6)
  799. }
  800. })
  801. }
  802. func TestNULL(t *testing.T) {
  803. runTests(t, dsn, func(dbt *DBTest) {
  804. nullStmt, err := dbt.db.Prepare("SELECT NULL")
  805. if err != nil {
  806. dbt.Fatal(err)
  807. }
  808. defer nullStmt.Close()
  809. nonNullStmt, err := dbt.db.Prepare("SELECT 1")
  810. if err != nil {
  811. dbt.Fatal(err)
  812. }
  813. defer nonNullStmt.Close()
  814. // NullBool
  815. var nb sql.NullBool
  816. // Invalid
  817. if err = nullStmt.QueryRow().Scan(&nb); err != nil {
  818. dbt.Fatal(err)
  819. }
  820. if nb.Valid {
  821. dbt.Error("valid NullBool which should be invalid")
  822. }
  823. // Valid
  824. if err = nonNullStmt.QueryRow().Scan(&nb); err != nil {
  825. dbt.Fatal(err)
  826. }
  827. if !nb.Valid {
  828. dbt.Error("invalid NullBool which should be valid")
  829. } else if nb.Bool != true {
  830. dbt.Errorf("Unexpected NullBool value: %t (should be true)", nb.Bool)
  831. }
  832. // NullFloat64
  833. var nf sql.NullFloat64
  834. // Invalid
  835. if err = nullStmt.QueryRow().Scan(&nf); err != nil {
  836. dbt.Fatal(err)
  837. }
  838. if nf.Valid {
  839. dbt.Error("valid NullFloat64 which should be invalid")
  840. }
  841. // Valid
  842. if err = nonNullStmt.QueryRow().Scan(&nf); err != nil {
  843. dbt.Fatal(err)
  844. }
  845. if !nf.Valid {
  846. dbt.Error("invalid NullFloat64 which should be valid")
  847. } else if nf.Float64 != float64(1) {
  848. dbt.Errorf("unexpected NullFloat64 value: %f (should be 1.0)", nf.Float64)
  849. }
  850. // NullInt64
  851. var ni sql.NullInt64
  852. // Invalid
  853. if err = nullStmt.QueryRow().Scan(&ni); err != nil {
  854. dbt.Fatal(err)
  855. }
  856. if ni.Valid {
  857. dbt.Error("valid NullInt64 which should be invalid")
  858. }
  859. // Valid
  860. if err = nonNullStmt.QueryRow().Scan(&ni); err != nil {
  861. dbt.Fatal(err)
  862. }
  863. if !ni.Valid {
  864. dbt.Error("invalid NullInt64 which should be valid")
  865. } else if ni.Int64 != int64(1) {
  866. dbt.Errorf("unexpected NullInt64 value: %d (should be 1)", ni.Int64)
  867. }
  868. // NullString
  869. var ns sql.NullString
  870. // Invalid
  871. if err = nullStmt.QueryRow().Scan(&ns); err != nil {
  872. dbt.Fatal(err)
  873. }
  874. if ns.Valid {
  875. dbt.Error("valid NullString which should be invalid")
  876. }
  877. // Valid
  878. if err = nonNullStmt.QueryRow().Scan(&ns); err != nil {
  879. dbt.Fatal(err)
  880. }
  881. if !ns.Valid {
  882. dbt.Error("invalid NullString which should be valid")
  883. } else if ns.String != `1` {
  884. dbt.Error("unexpected NullString value:" + ns.String + " (should be `1`)")
  885. }
  886. // nil-bytes
  887. var b []byte
  888. // Read nil
  889. if err = nullStmt.QueryRow().Scan(&b); err != nil {
  890. dbt.Fatal(err)
  891. }
  892. if b != nil {
  893. dbt.Error("non-nil []byte which should be nil")
  894. }
  895. // Read non-nil
  896. if err = nonNullStmt.QueryRow().Scan(&b); err != nil {
  897. dbt.Fatal(err)
  898. }
  899. if b == nil {
  900. dbt.Error("nil []byte which should be non-nil")
  901. }
  902. // Insert nil
  903. b = nil
  904. success := false
  905. if err = dbt.db.QueryRow("SELECT ? IS NULL", b).Scan(&success); err != nil {
  906. dbt.Fatal(err)
  907. }
  908. if !success {
  909. dbt.Error("inserting []byte(nil) as NULL failed")
  910. }
  911. // Check input==output with input==nil
  912. b = nil
  913. if err = dbt.db.QueryRow("SELECT ?", b).Scan(&b); err != nil {
  914. dbt.Fatal(err)
  915. }
  916. if b != nil {
  917. dbt.Error("non-nil echo from nil input")
  918. }
  919. // Check input==output with input!=nil
  920. b = []byte("")
  921. if err = dbt.db.QueryRow("SELECT ?", b).Scan(&b); err != nil {
  922. dbt.Fatal(err)
  923. }
  924. if b == nil {
  925. dbt.Error("nil echo from non-nil input")
  926. }
  927. // Insert NULL
  928. dbt.mustExec("CREATE TABLE test (dummmy1 int, value int, dummy2 int)")
  929. dbt.mustExec("INSERT INTO test VALUES (?, ?, ?)", 1, nil, 2)
  930. var out interface{}
  931. rows := dbt.mustQuery("SELECT * FROM test")
  932. if rows.Next() {
  933. rows.Scan(&out)
  934. if out != nil {
  935. dbt.Errorf("%v != nil", out)
  936. }
  937. } else {
  938. dbt.Error("no data")
  939. }
  940. })
  941. }
  942. func TestUint64(t *testing.T) {
  943. const (
  944. u0 = uint64(0)
  945. uall = ^u0
  946. uhigh = uall >> 1
  947. utop = ^uhigh
  948. s0 = int64(0)
  949. sall = ^s0
  950. shigh = int64(uhigh)
  951. stop = ^shigh
  952. )
  953. runTests(t, dsn, func(dbt *DBTest) {
  954. stmt, err := dbt.db.Prepare(`SELECT ?, ?, ? ,?, ?, ?, ?, ?`)
  955. if err != nil {
  956. dbt.Fatal(err)
  957. }
  958. defer stmt.Close()
  959. row := stmt.QueryRow(
  960. u0, uhigh, utop, uall,
  961. s0, shigh, stop, sall,
  962. )
  963. var ua, ub, uc, ud uint64
  964. var sa, sb, sc, sd int64
  965. err = row.Scan(&ua, &ub, &uc, &ud, &sa, &sb, &sc, &sd)
  966. if err != nil {
  967. dbt.Fatal(err)
  968. }
  969. switch {
  970. case ua != u0,
  971. ub != uhigh,
  972. uc != utop,
  973. ud != uall,
  974. sa != s0,
  975. sb != shigh,
  976. sc != stop,
  977. sd != sall:
  978. dbt.Fatal("unexpected result value")
  979. }
  980. })
  981. }
  982. func TestLongData(t *testing.T) {
  983. runTests(t, dsn+"&maxAllowedPacket=0", func(dbt *DBTest) {
  984. var maxAllowedPacketSize int
  985. err := dbt.db.QueryRow("select @@max_allowed_packet").Scan(&maxAllowedPacketSize)
  986. if err != nil {
  987. dbt.Fatal(err)
  988. }
  989. maxAllowedPacketSize--
  990. // don't get too ambitious
  991. if maxAllowedPacketSize > 1<<25 {
  992. maxAllowedPacketSize = 1 << 25
  993. }
  994. dbt.mustExec("CREATE TABLE test (value LONGBLOB)")
  995. in := strings.Repeat(`a`, maxAllowedPacketSize+1)
  996. var out string
  997. var rows *sql.Rows
  998. // Long text data
  999. const nonDataQueryLen = 28 // length query w/o value
  1000. inS := in[:maxAllowedPacketSize-nonDataQueryLen]
  1001. dbt.mustExec("INSERT INTO test VALUES('" + inS + "')")
  1002. rows = dbt.mustQuery("SELECT value FROM test")
  1003. if rows.Next() {
  1004. rows.Scan(&out)
  1005. if inS != out {
  1006. dbt.Fatalf("LONGBLOB: length in: %d, length out: %d", len(inS), len(out))
  1007. }
  1008. if rows.Next() {
  1009. dbt.Error("LONGBLOB: unexpexted row")
  1010. }
  1011. } else {
  1012. dbt.Fatalf("LONGBLOB: no data")
  1013. }
  1014. // Empty table
  1015. dbt.mustExec("TRUNCATE TABLE test")
  1016. // Long binary data
  1017. dbt.mustExec("INSERT INTO test VALUES(?)", in)
  1018. rows = dbt.mustQuery("SELECT value FROM test WHERE 1=?", 1)
  1019. if rows.Next() {
  1020. rows.Scan(&out)
  1021. if in != out {
  1022. dbt.Fatalf("LONGBLOB: length in: %d, length out: %d", len(in), len(out))
  1023. }
  1024. if rows.Next() {
  1025. dbt.Error("LONGBLOB: unexpexted row")
  1026. }
  1027. } else {
  1028. if err = rows.Err(); err != nil {
  1029. dbt.Fatalf("LONGBLOB: no data (err: %s)", err.Error())
  1030. } else {
  1031. dbt.Fatal("LONGBLOB: no data (err: <nil>)")
  1032. }
  1033. }
  1034. })
  1035. }
  1036. func TestLoadData(t *testing.T) {
  1037. runTests(t, dsn, func(dbt *DBTest) {
  1038. verifyLoadDataResult := func() {
  1039. rows, err := dbt.db.Query("SELECT * FROM test")
  1040. if err != nil {
  1041. dbt.Fatal(err.Error())
  1042. }
  1043. i := 0
  1044. values := [4]string{
  1045. "a string",
  1046. "a string containing a \t",
  1047. "a string containing a \n",
  1048. "a string containing both \t\n",
  1049. }
  1050. var id int
  1051. var value string
  1052. for rows.Next() {
  1053. i++
  1054. err = rows.Scan(&id, &value)
  1055. if err != nil {
  1056. dbt.Fatal(err.Error())
  1057. }
  1058. if i != id {
  1059. dbt.Fatalf("%d != %d", i, id)
  1060. }
  1061. if values[i-1] != value {
  1062. dbt.Fatalf("%q != %q", values[i-1], value)
  1063. }
  1064. }
  1065. err = rows.Err()
  1066. if err != nil {
  1067. dbt.Fatal(err.Error())
  1068. }
  1069. if i != 4 {
  1070. dbt.Fatalf("rows count mismatch. Got %d, want 4", i)
  1071. }
  1072. }
  1073. dbt.db.Exec("DROP TABLE IF EXISTS test")
  1074. dbt.mustExec("CREATE TABLE test (id INT NOT NULL PRIMARY KEY, value TEXT NOT NULL) CHARACTER SET utf8")
  1075. // Local File
  1076. file, err := ioutil.TempFile("", "gotest")
  1077. defer os.Remove(file.Name())
  1078. if err != nil {
  1079. dbt.Fatal(err)
  1080. }
  1081. RegisterLocalFile(file.Name())
  1082. // Try first with empty file
  1083. dbt.mustExec(fmt.Sprintf("LOAD DATA LOCAL INFILE %q INTO TABLE test", file.Name()))
  1084. var count int
  1085. err = dbt.db.QueryRow("SELECT COUNT(*) FROM test").Scan(&count)
  1086. if err != nil {
  1087. dbt.Fatal(err.Error())
  1088. }
  1089. if count != 0 {
  1090. dbt.Fatalf("unexpected row count: got %d, want 0", count)
  1091. }
  1092. // Then fille File with data and try to load it
  1093. 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")
  1094. file.Close()
  1095. dbt.mustExec(fmt.Sprintf("LOAD DATA LOCAL INFILE %q INTO TABLE test", file.Name()))
  1096. verifyLoadDataResult()
  1097. // Try with non-existing file
  1098. _, err = dbt.db.Exec("LOAD DATA LOCAL INFILE 'doesnotexist' INTO TABLE test")
  1099. if err == nil {
  1100. dbt.Fatal("load non-existent file didn't fail")
  1101. } else if err.Error() != "local file 'doesnotexist' is not registered" {
  1102. dbt.Fatal(err.Error())
  1103. }
  1104. // Empty table
  1105. dbt.mustExec("TRUNCATE TABLE test")
  1106. // Reader
  1107. RegisterReaderHandler("test", func() io.Reader {
  1108. file, err = os.Open(file.Name())
  1109. if err != nil {
  1110. dbt.Fatal(err)
  1111. }
  1112. return file
  1113. })
  1114. dbt.mustExec("LOAD DATA LOCAL INFILE 'Reader::test' INTO TABLE test")
  1115. verifyLoadDataResult()
  1116. // negative test
  1117. _, err = dbt.db.Exec("LOAD DATA LOCAL INFILE 'Reader::doesnotexist' INTO TABLE test")
  1118. if err == nil {
  1119. dbt.Fatal("load non-existent Reader didn't fail")
  1120. } else if err.Error() != "Reader 'doesnotexist' is not registered" {
  1121. dbt.Fatal(err.Error())
  1122. }
  1123. })
  1124. }
  1125. func TestFoundRows(t *testing.T) {
  1126. runTests(t, dsn, func(dbt *DBTest) {
  1127. dbt.mustExec("CREATE TABLE test (id INT NOT NULL ,data INT NOT NULL)")
  1128. dbt.mustExec("INSERT INTO test (id, data) VALUES (0, 0),(0, 0),(1, 0),(1, 0),(1, 1)")
  1129. res := dbt.mustExec("UPDATE test SET data = 1 WHERE id = 0")
  1130. count, err := res.RowsAffected()
  1131. if err != nil {
  1132. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  1133. }
  1134. if count != 2 {
  1135. dbt.Fatalf("Expected 2 affected rows, got %d", count)
  1136. }
  1137. res = dbt.mustExec("UPDATE test SET data = 1 WHERE id = 1")
  1138. count, err = res.RowsAffected()
  1139. if err != nil {
  1140. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  1141. }
  1142. if count != 2 {
  1143. dbt.Fatalf("Expected 2 affected rows, got %d", count)
  1144. }
  1145. })
  1146. runTests(t, dsn+"&clientFoundRows=true", func(dbt *DBTest) {
  1147. dbt.mustExec("CREATE TABLE test (id INT NOT NULL ,data INT NOT NULL)")
  1148. dbt.mustExec("INSERT INTO test (id, data) VALUES (0, 0),(0, 0),(1, 0),(1, 0),(1, 1)")
  1149. res := dbt.mustExec("UPDATE test SET data = 1 WHERE id = 0")
  1150. count, err := res.RowsAffected()
  1151. if err != nil {
  1152. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  1153. }
  1154. if count != 2 {
  1155. dbt.Fatalf("Expected 2 matched rows, got %d", count)
  1156. }
  1157. res = dbt.mustExec("UPDATE test SET data = 1 WHERE id = 1")
  1158. count, err = res.RowsAffected()
  1159. if err != nil {
  1160. dbt.Fatalf("res.RowsAffected() returned error: %s", err.Error())
  1161. }
  1162. if count != 3 {
  1163. dbt.Fatalf("Expected 3 matched rows, got %d", count)
  1164. }
  1165. })
  1166. }
  1167. func TestTLS(t *testing.T) {
  1168. tlsTest := func(dbt *DBTest) {
  1169. if err := dbt.db.Ping(); err != nil {
  1170. if err == ErrNoTLS {
  1171. dbt.Skip("server does not support TLS")
  1172. } else {
  1173. dbt.Fatalf("error on Ping: %s", err.Error())
  1174. }
  1175. }
  1176. rows := dbt.mustQuery("SHOW STATUS LIKE 'Ssl_cipher'")
  1177. var variable, value *sql.RawBytes
  1178. for rows.Next() {
  1179. if err := rows.Scan(&variable, &value); err != nil {
  1180. dbt.Fatal(err.Error())
  1181. }
  1182. if value == nil {
  1183. dbt.Fatal("no Cipher")
  1184. }
  1185. }
  1186. }
  1187. runTests(t, dsn+"&tls=skip-verify", tlsTest)
  1188. // Verify that registering / using a custom cfg works
  1189. RegisterTLSConfig("custom-skip-verify", &tls.Config{
  1190. InsecureSkipVerify: true,
  1191. })
  1192. runTests(t, dsn+"&tls=custom-skip-verify", tlsTest)
  1193. }
  1194. func TestReuseClosedConnection(t *testing.T) {
  1195. // this test does not use sql.database, it uses the driver directly
  1196. if !available {
  1197. t.Skipf("MySQL server not running on %s", netAddr)
  1198. }
  1199. md := &MySQLDriver{}
  1200. conn, err := md.Open(dsn)
  1201. if err != nil {
  1202. t.Fatalf("error connecting: %s", err.Error())
  1203. }
  1204. stmt, err := conn.Prepare("DO 1")
  1205. if err != nil {
  1206. t.Fatalf("error preparing statement: %s", err.Error())
  1207. }
  1208. _, err = stmt.Exec(nil)
  1209. if err != nil {
  1210. t.Fatalf("error executing statement: %s", err.Error())
  1211. }
  1212. err = conn.Close()
  1213. if err != nil {
  1214. t.Fatalf("error closing connection: %s", err.Error())
  1215. }
  1216. defer func() {
  1217. if err := recover(); err != nil {
  1218. t.Errorf("panic after reusing a closed connection: %v", err)
  1219. }
  1220. }()
  1221. _, err = stmt.Exec(nil)
  1222. if err != nil && err != driver.ErrBadConn {
  1223. t.Errorf("unexpected error '%s', expected '%s'",
  1224. err.Error(), driver.ErrBadConn.Error())
  1225. }
  1226. }
  1227. func TestCharset(t *testing.T) {
  1228. if !available {
  1229. t.Skipf("MySQL server not running on %s", netAddr)
  1230. }
  1231. mustSetCharset := func(charsetParam, expected string) {
  1232. runTests(t, dsn+"&"+charsetParam, func(dbt *DBTest) {
  1233. rows := dbt.mustQuery("SELECT @@character_set_connection")
  1234. defer rows.Close()
  1235. if !rows.Next() {
  1236. dbt.Fatalf("error getting connection charset: %s", rows.Err())
  1237. }
  1238. var got string
  1239. rows.Scan(&got)
  1240. if got != expected {
  1241. dbt.Fatalf("expected connection charset %s but got %s", expected, got)
  1242. }
  1243. })
  1244. }
  1245. // non utf8 test
  1246. mustSetCharset("charset=ascii", "ascii")
  1247. // when the first charset is invalid, use the second
  1248. mustSetCharset("charset=none,utf8", "utf8")
  1249. // when the first charset is valid, use it
  1250. mustSetCharset("charset=ascii,utf8", "ascii")
  1251. mustSetCharset("charset=utf8,ascii", "utf8")
  1252. }
  1253. func TestFailingCharset(t *testing.T) {
  1254. runTests(t, dsn+"&charset=none", func(dbt *DBTest) {
  1255. // run query to really establish connection...
  1256. _, err := dbt.db.Exec("SELECT 1")
  1257. if err == nil {
  1258. dbt.db.Close()
  1259. t.Fatalf("connection must not succeed without a valid charset")
  1260. }
  1261. })
  1262. }
  1263. func TestCollation(t *testing.T) {
  1264. if !available {
  1265. t.Skipf("MySQL server not running on %s", netAddr)
  1266. }
  1267. defaultCollation := "utf8_general_ci"
  1268. testCollations := []string{
  1269. "", // do not set
  1270. defaultCollation, // driver default
  1271. "latin1_general_ci",
  1272. "binary",
  1273. "utf8_unicode_ci",
  1274. "cp1257_bin",
  1275. }
  1276. for _, collation := range testCollations {
  1277. var expected, tdsn string
  1278. if collation != "" {
  1279. tdsn = dsn + "&collation=" + collation
  1280. expected = collation
  1281. } else {
  1282. tdsn = dsn
  1283. expected = defaultCollation
  1284. }
  1285. runTests(t, tdsn, func(dbt *DBTest) {
  1286. var got string
  1287. if err := dbt.db.QueryRow("SELECT @@collation_connection").Scan(&got); err != nil {
  1288. dbt.Fatal(err)
  1289. }
  1290. if got != expected {
  1291. dbt.Fatalf("expected connection collation %s but got %s", expected, got)
  1292. }
  1293. })
  1294. }
  1295. }
  1296. func TestColumnsWithAlias(t *testing.T) {
  1297. runTests(t, dsn+"&columnsWithAlias=true", func(dbt *DBTest) {
  1298. rows := dbt.mustQuery("SELECT 1 AS A")
  1299. defer rows.Close()
  1300. cols, _ := rows.Columns()
  1301. if len(cols) != 1 {
  1302. t.Fatalf("expected 1 column, got %d", len(cols))
  1303. }
  1304. if cols[0] != "A" {
  1305. t.Fatalf("expected column name \"A\", got \"%s\"", cols[0])
  1306. }
  1307. rows.Close()
  1308. rows = dbt.mustQuery("SELECT * FROM (SELECT 1 AS one) AS A")
  1309. cols, _ = rows.Columns()
  1310. if len(cols) != 1 {
  1311. t.Fatalf("expected 1 column, got %d", len(cols))
  1312. }
  1313. if cols[0] != "A.one" {
  1314. t.Fatalf("expected column name \"A.one\", got \"%s\"", cols[0])
  1315. }
  1316. })
  1317. }
  1318. func TestRawBytesResultExceedsBuffer(t *testing.T) {
  1319. runTests(t, dsn, func(dbt *DBTest) {
  1320. // defaultBufSize from buffer.go
  1321. expected := strings.Repeat("abc", defaultBufSize)
  1322. rows := dbt.mustQuery("SELECT '" + expected + "'")
  1323. defer rows.Close()
  1324. if !rows.Next() {
  1325. dbt.Error("expected result, got none")
  1326. }
  1327. var result sql.RawBytes
  1328. rows.Scan(&result)
  1329. if expected != string(result) {
  1330. dbt.Error("result did not match expected value")
  1331. }
  1332. })
  1333. }
  1334. func TestTimezoneConversion(t *testing.T) {
  1335. zones := []string{"UTC", "US/Central", "US/Pacific", "Local"}
  1336. // Regression test for timezone handling
  1337. tzTest := func(dbt *DBTest) {
  1338. // Create table
  1339. dbt.mustExec("CREATE TABLE test (ts TIMESTAMP)")
  1340. // Insert local time into database (should be converted)
  1341. usCentral, _ := time.LoadLocation("US/Central")
  1342. reftime := time.Date(2014, 05, 30, 18, 03, 17, 0, time.UTC).In(usCentral)
  1343. dbt.mustExec("INSERT INTO test VALUE (?)", reftime)
  1344. // Retrieve time from DB
  1345. rows := dbt.mustQuery("SELECT ts FROM test")
  1346. if !rows.Next() {
  1347. dbt.Fatal("did not get any rows out")
  1348. }
  1349. var dbTime time.Time
  1350. err := rows.Scan(&dbTime)
  1351. if err != nil {
  1352. dbt.Fatal("Err", err)
  1353. }
  1354. // Check that dates match
  1355. if reftime.Unix() != dbTime.Unix() {
  1356. dbt.Errorf("times do not match.\n")
  1357. dbt.Errorf(" Now(%v)=%v\n", usCentral, reftime)
  1358. dbt.Errorf(" Now(UTC)=%v\n", dbTime)
  1359. }
  1360. }
  1361. for _, tz := range zones {
  1362. runTests(t, dsn+"&parseTime=true&loc="+url.QueryEscape(tz), tzTest)
  1363. }
  1364. }
  1365. // Special cases
  1366. func TestRowsClose(t *testing.T) {
  1367. runTests(t, dsn, func(dbt *DBTest) {
  1368. rows, err := dbt.db.Query("SELECT 1")
  1369. if err != nil {
  1370. dbt.Fatal(err)
  1371. }
  1372. err = rows.Close()
  1373. if err != nil {
  1374. dbt.Fatal(err)
  1375. }
  1376. if rows.Next() {
  1377. dbt.Fatal("unexpected row after rows.Close()")
  1378. }
  1379. err = rows.Err()
  1380. if err != nil {
  1381. dbt.Fatal(err)
  1382. }
  1383. })
  1384. }
  1385. // dangling statements
  1386. // http://code.google.com/p/go/issues/detail?id=3865
  1387. func TestCloseStmtBeforeRows(t *testing.T) {
  1388. runTests(t, dsn, func(dbt *DBTest) {
  1389. stmt, err := dbt.db.Prepare("SELECT 1")
  1390. if err != nil {
  1391. dbt.Fatal(err)
  1392. }
  1393. rows, err := stmt.Query()
  1394. if err != nil {
  1395. stmt.Close()
  1396. dbt.Fatal(err)
  1397. }
  1398. defer rows.Close()
  1399. err = stmt.Close()
  1400. if err != nil {
  1401. dbt.Fatal(err)
  1402. }
  1403. if !rows.Next() {
  1404. dbt.Fatal("getting row failed")
  1405. } else {
  1406. err = rows.Err()
  1407. if err != nil {
  1408. dbt.Fatal(err)
  1409. }
  1410. var out bool
  1411. err = rows.Scan(&out)
  1412. if err != nil {
  1413. dbt.Fatalf("error on rows.Scan(): %s", err.Error())
  1414. }
  1415. if out != true {
  1416. dbt.Errorf("true != %t", out)
  1417. }
  1418. }
  1419. })
  1420. }
  1421. // It is valid to have multiple Rows for the same Stmt
  1422. // http://code.google.com/p/go/issues/detail?id=3734
  1423. func TestStmtMultiRows(t *testing.T) {
  1424. runTests(t, dsn, func(dbt *DBTest) {
  1425. stmt, err := dbt.db.Prepare("SELECT 1 UNION SELECT 0")
  1426. if err != nil {
  1427. dbt.Fatal(err)
  1428. }
  1429. rows1, err := stmt.Query()
  1430. if err != nil {
  1431. stmt.Close()
  1432. dbt.Fatal(err)
  1433. }
  1434. defer rows1.Close()
  1435. rows2, err := stmt.Query()
  1436. if err != nil {
  1437. stmt.Close()
  1438. dbt.Fatal(err)
  1439. }
  1440. defer rows2.Close()
  1441. var out bool
  1442. // 1
  1443. if !rows1.Next() {
  1444. dbt.Fatal("first rows1.Next failed")
  1445. } else {
  1446. err = rows1.Err()
  1447. if err != nil {
  1448. dbt.Fatal(err)
  1449. }
  1450. err = rows1.Scan(&out)
  1451. if err != nil {
  1452. dbt.Fatalf("error on rows.Scan(): %s", err.Error())
  1453. }
  1454. if out != true {
  1455. dbt.Errorf("true != %t", out)
  1456. }
  1457. }
  1458. if !rows2.Next() {
  1459. dbt.Fatal("first rows2.Next failed")
  1460. } else {
  1461. err = rows2.Err()
  1462. if err != nil {
  1463. dbt.Fatal(err)
  1464. }
  1465. err = rows2.Scan(&out)
  1466. if err != nil {
  1467. dbt.Fatalf("error on rows.Scan(): %s", err.Error())
  1468. }
  1469. if out != true {
  1470. dbt.Errorf("true != %t", out)
  1471. }
  1472. }
  1473. // 2
  1474. if !rows1.Next() {
  1475. dbt.Fatal("second rows1.Next failed")
  1476. } else {
  1477. err = rows1.Err()
  1478. if err != nil {
  1479. dbt.Fatal(err)
  1480. }
  1481. err = rows1.Scan(&out)
  1482. if err != nil {
  1483. dbt.Fatalf("error on rows.Scan(): %s", err.Error())
  1484. }
  1485. if out != false {
  1486. dbt.Errorf("false != %t", out)
  1487. }
  1488. if rows1.Next() {
  1489. dbt.Fatal("unexpected row on rows1")
  1490. }
  1491. err = rows1.Close()
  1492. if err != nil {
  1493. dbt.Fatal(err)
  1494. }
  1495. }
  1496. if !rows2.Next() {
  1497. dbt.Fatal("second rows2.Next failed")
  1498. } else {
  1499. err = rows2.Err()
  1500. if err != nil {
  1501. dbt.Fatal(err)
  1502. }
  1503. err = rows2.Scan(&out)
  1504. if err != nil {
  1505. dbt.Fatalf("error on rows.Scan(): %s", err.Error())
  1506. }
  1507. if out != false {
  1508. dbt.Errorf("false != %t", out)
  1509. }
  1510. if rows2.Next() {
  1511. dbt.Fatal("unexpected row on rows2")
  1512. }
  1513. err = rows2.Close()
  1514. if err != nil {
  1515. dbt.Fatal(err)
  1516. }
  1517. }
  1518. })
  1519. }
  1520. // Regression test for
  1521. // * more than 32 NULL parameters (issue 209)
  1522. // * more parameters than fit into the buffer (issue 201)
  1523. // * parameters * 64 > max_allowed_packet (issue 734)
  1524. func TestPreparedManyCols(t *testing.T) {
  1525. numParams := 65535
  1526. runTests(t, dsn, func(dbt *DBTest) {
  1527. query := "SELECT ?" + strings.Repeat(",?", numParams-1)
  1528. stmt, err := dbt.db.Prepare(query)
  1529. if err != nil {
  1530. dbt.Fatal(err)
  1531. }
  1532. defer stmt.Close()
  1533. // create more parameters than fit into the buffer
  1534. // which will take nil-values
  1535. params := make([]interface{}, numParams)
  1536. rows, err := stmt.Query(params...)
  1537. if err != nil {
  1538. dbt.Fatal(err)
  1539. }
  1540. rows.Close()
  1541. // Create 0byte string which we can't send via STMT_LONG_DATA.
  1542. for i := 0; i < numParams; i++ {
  1543. params[i] = ""
  1544. }
  1545. rows, err = stmt.Query(params...)
  1546. if err != nil {
  1547. dbt.Fatal(err)
  1548. }
  1549. rows.Close()
  1550. })
  1551. }
  1552. func TestConcurrent(t *testing.T) {
  1553. if enabled, _ := readBool(os.Getenv("MYSQL_TEST_CONCURRENT")); !enabled {
  1554. t.Skip("MYSQL_TEST_CONCURRENT env var not set")
  1555. }
  1556. runTests(t, dsn, func(dbt *DBTest) {
  1557. var max int
  1558. err := dbt.db.QueryRow("SELECT @@max_connections").Scan(&max)
  1559. if err != nil {
  1560. dbt.Fatalf("%s", err.Error())
  1561. }
  1562. dbt.Logf("testing up to %d concurrent connections \r\n", max)
  1563. var remaining, succeeded int32 = int32(max), 0
  1564. var wg sync.WaitGroup
  1565. wg.Add(max)
  1566. var fatalError string
  1567. var once sync.Once
  1568. fatalf := func(s string, vals ...interface{}) {
  1569. once.Do(func() {
  1570. fatalError = fmt.Sprintf(s, vals...)
  1571. })
  1572. }
  1573. for i := 0; i < max; i++ {
  1574. go func(id int) {
  1575. defer wg.Done()
  1576. tx, err := dbt.db.Begin()
  1577. atomic.AddInt32(&remaining, -1)
  1578. if err != nil {
  1579. if err.Error() != "Error 1040: Too many connections" {
  1580. fatalf("error on conn %d: %s", id, err.Error())
  1581. }
  1582. return
  1583. }
  1584. // keep the connection busy until all connections are open
  1585. for remaining > 0 {
  1586. if _, err = tx.Exec("DO 1"); err != nil {
  1587. fatalf("error on conn %d: %s", id, err.Error())
  1588. return
  1589. }
  1590. }
  1591. if err = tx.Commit(); err != nil {
  1592. fatalf("error on conn %d: %s", id, err.Error())
  1593. return
  1594. }
  1595. // everything went fine with this connection
  1596. atomic.AddInt32(&succeeded, 1)
  1597. }(i)
  1598. }
  1599. // wait until all conections are open
  1600. wg.Wait()
  1601. if fatalError != "" {
  1602. dbt.Fatal(fatalError)
  1603. }
  1604. dbt.Logf("reached %d concurrent connections\r\n", succeeded)
  1605. })
  1606. }
  1607. // Tests custom dial functions
  1608. func TestCustomDial(t *testing.T) {
  1609. if !available {
  1610. t.Skipf("MySQL server not running on %s", netAddr)
  1611. }
  1612. // our custom dial function which justs wraps net.Dial here
  1613. RegisterDial("mydial", func(addr string) (net.Conn, error) {
  1614. return net.Dial(prot, addr)
  1615. })
  1616. db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@mydial(%s)/%s?timeout=30s", user, pass, addr, dbname))
  1617. if err != nil {
  1618. t.Fatalf("error connecting: %s", err.Error())
  1619. }
  1620. defer db.Close()
  1621. if _, err = db.Exec("DO 1"); err != nil {
  1622. t.Fatalf("connection failed: %s", err.Error())
  1623. }
  1624. }
  1625. func TestSQLInjection(t *testing.T) {
  1626. createTest := func(arg string) func(dbt *DBTest) {
  1627. return func(dbt *DBTest) {
  1628. dbt.mustExec("CREATE TABLE test (v INTEGER)")
  1629. dbt.mustExec("INSERT INTO test VALUES (?)", 1)
  1630. var v int
  1631. // NULL can't be equal to anything, the idea here is to inject query so it returns row
  1632. // This test verifies that escapeQuotes and escapeBackslash are working properly
  1633. err := dbt.db.QueryRow("SELECT v FROM test WHERE NULL = ?", arg).Scan(&v)
  1634. if err == sql.ErrNoRows {
  1635. return // success, sql injection failed
  1636. } else if err == nil {
  1637. dbt.Errorf("sql injection successful with arg: %s", arg)
  1638. } else {
  1639. dbt.Errorf("error running query with arg: %s; err: %s", arg, err.Error())
  1640. }
  1641. }
  1642. }
  1643. dsns := []string{
  1644. dsn,
  1645. dsn + "&sql_mode='NO_BACKSLASH_ESCAPES'",
  1646. }
  1647. for _, testdsn := range dsns {
  1648. runTests(t, testdsn, createTest("1 OR 1=1"))
  1649. runTests(t, testdsn, createTest("' OR '1'='1"))
  1650. }
  1651. }
  1652. // Test if inserted data is correctly retrieved after being escaped
  1653. func TestInsertRetrieveEscapedData(t *testing.T) {
  1654. testData := func(dbt *DBTest) {
  1655. dbt.mustExec("CREATE TABLE test (v VARCHAR(255))")
  1656. // All sequences that are escaped by escapeQuotes and escapeBackslash
  1657. v := "foo \x00\n\r\x1a\"'\\"
  1658. dbt.mustExec("INSERT INTO test VALUES (?)", v)
  1659. var out string
  1660. err := dbt.db.QueryRow("SELECT v FROM test").Scan(&out)
  1661. if err != nil {
  1662. dbt.Fatalf("%s", err.Error())
  1663. }
  1664. if out != v {
  1665. dbt.Errorf("%q != %q", out, v)
  1666. }
  1667. }
  1668. dsns := []string{
  1669. dsn,
  1670. dsn + "&sql_mode='NO_BACKSLASH_ESCAPES'",
  1671. }
  1672. for _, testdsn := range dsns {
  1673. runTests(t, testdsn, testData)
  1674. }
  1675. }
  1676. func TestUnixSocketAuthFail(t *testing.T) {
  1677. runTests(t, dsn, func(dbt *DBTest) {
  1678. // Save the current logger so we can restore it.
  1679. oldLogger := errLog
  1680. // Set a new logger so we can capture its output.
  1681. buffer := bytes.NewBuffer(make([]byte, 0, 64))
  1682. newLogger := log.New(buffer, "prefix: ", 0)
  1683. SetLogger(newLogger)
  1684. // Restore the logger.
  1685. defer SetLogger(oldLogger)
  1686. // Make a new DSN that uses the MySQL socket file and a bad password, which
  1687. // we can make by simply appending any character to the real password.
  1688. badPass := pass + "x"
  1689. socket := ""
  1690. if prot == "unix" {
  1691. socket = addr
  1692. } else {
  1693. // Get socket file from MySQL.
  1694. err := dbt.db.QueryRow("SELECT @@socket").Scan(&socket)
  1695. if err != nil {
  1696. t.Fatalf("error on SELECT @@socket: %s", err.Error())
  1697. }
  1698. }
  1699. t.Logf("socket: %s", socket)
  1700. badDSN := fmt.Sprintf("%s:%s@unix(%s)/%s?timeout=30s", user, badPass, socket, dbname)
  1701. db, err := sql.Open("mysql", badDSN)
  1702. if err != nil {
  1703. t.Fatalf("error connecting: %s", err.Error())
  1704. }
  1705. defer db.Close()
  1706. // Connect to MySQL for real. This will cause an auth failure.
  1707. err = db.Ping()
  1708. if err == nil {
  1709. t.Error("expected Ping() to return an error")
  1710. }
  1711. // The driver should not log anything.
  1712. if actual := buffer.String(); actual != "" {
  1713. t.Errorf("expected no output, got %q", actual)
  1714. }
  1715. })
  1716. }
  1717. // See Issue #422
  1718. func TestInterruptBySignal(t *testing.T) {
  1719. runTestsWithMultiStatement(t, dsn, func(dbt *DBTest) {
  1720. dbt.mustExec(`
  1721. DROP PROCEDURE IF EXISTS test_signal;
  1722. CREATE PROCEDURE test_signal(ret INT)
  1723. BEGIN
  1724. SELECT ret;
  1725. SIGNAL SQLSTATE
  1726. '45001'
  1727. SET
  1728. MESSAGE_TEXT = "an error",
  1729. MYSQL_ERRNO = 45001;
  1730. END
  1731. `)
  1732. defer dbt.mustExec("DROP PROCEDURE test_signal")
  1733. var val int
  1734. // text protocol
  1735. rows, err := dbt.db.Query("CALL test_signal(42)")
  1736. if err != nil {
  1737. dbt.Fatalf("error on text query: %s", err.Error())
  1738. }
  1739. for rows.Next() {
  1740. if err := rows.Scan(&val); err != nil {
  1741. dbt.Error(err)
  1742. } else if val != 42 {
  1743. dbt.Errorf("expected val to be 42")
  1744. }
  1745. }
  1746. // binary protocol
  1747. rows, err = dbt.db.Query("CALL test_signal(?)", 42)
  1748. if err != nil {
  1749. dbt.Fatalf("error on binary query: %s", err.Error())
  1750. }
  1751. for rows.Next() {
  1752. if err := rows.Scan(&val); err != nil {
  1753. dbt.Error(err)
  1754. } else if val != 42 {
  1755. dbt.Errorf("expected val to be 42")
  1756. }
  1757. }
  1758. })
  1759. }
  1760. func TestColumnsReusesSlice(t *testing.T) {
  1761. rows := mysqlRows{
  1762. rs: resultSet{
  1763. columns: []mysqlField{
  1764. {
  1765. tableName: "test",
  1766. name: "A",
  1767. },
  1768. {
  1769. tableName: "test",
  1770. name: "B",
  1771. },
  1772. },
  1773. },
  1774. }
  1775. allocs := testing.AllocsPerRun(1, func() {
  1776. cols := rows.Columns()
  1777. if len(cols) != 2 {
  1778. t.Fatalf("expected 2 columns, got %d", len(cols))
  1779. }
  1780. })
  1781. if allocs != 0 {
  1782. t.Fatalf("expected 0 allocations, got %d", int(allocs))
  1783. }
  1784. if rows.rs.columnNames == nil {
  1785. t.Fatalf("expected columnNames to be set, got nil")
  1786. }
  1787. }
  1788. func TestRejectReadOnly(t *testing.T) {
  1789. runTests(t, dsn, func(dbt *DBTest) {
  1790. // Create Table
  1791. dbt.mustExec("CREATE TABLE test (value BOOL)")
  1792. // Set the session to read-only. We didn't set the `rejectReadOnly`
  1793. // option, so any writes after this should fail.
  1794. _, err := dbt.db.Exec("SET SESSION TRANSACTION READ ONLY")
  1795. // Error 1193: Unknown system variable 'TRANSACTION' => skip test,
  1796. // MySQL server version is too old
  1797. maybeSkip(t, err, 1193)
  1798. if _, err := dbt.db.Exec("DROP TABLE test"); err == nil {
  1799. t.Fatalf("writing to DB in read-only session without " +
  1800. "rejectReadOnly did not error")
  1801. }
  1802. // Set the session back to read-write so runTests() can properly clean
  1803. // up the table `test`.
  1804. dbt.mustExec("SET SESSION TRANSACTION READ WRITE")
  1805. })
  1806. // Enable the `rejectReadOnly` option.
  1807. runTests(t, dsn+"&rejectReadOnly=true", func(dbt *DBTest) {
  1808. // Create Table
  1809. dbt.mustExec("CREATE TABLE test (value BOOL)")
  1810. // Set the session to read only. Any writes after this should error on
  1811. // a driver.ErrBadConn, and cause `database/sql` to initiate a new
  1812. // connection.
  1813. dbt.mustExec("SET SESSION TRANSACTION READ ONLY")
  1814. // This would error, but `database/sql` should automatically retry on a
  1815. // new connection which is not read-only, and eventually succeed.
  1816. dbt.mustExec("DROP TABLE test")
  1817. })
  1818. }
  1819. func TestPing(t *testing.T) {
  1820. runTests(t, dsn, func(dbt *DBTest) {
  1821. if err := dbt.db.Ping(); err != nil {
  1822. dbt.fail("Ping", "Ping", err)
  1823. }
  1824. })
  1825. }
  1826. // See Issue #799
  1827. func TestEmptyPassword(t *testing.T) {
  1828. if !available {
  1829. t.Skipf("MySQL server not running on %s", netAddr)
  1830. }
  1831. dsn := fmt.Sprintf("%s:%s@%s/%s?timeout=30s", user, "", netAddr, dbname)
  1832. db, err := sql.Open("mysql", dsn)
  1833. if err == nil {
  1834. defer db.Close()
  1835. err = db.Ping()
  1836. }
  1837. if pass == "" {
  1838. if err != nil {
  1839. t.Fatal(err.Error())
  1840. }
  1841. } else {
  1842. if err == nil {
  1843. t.Fatal("expected authentication error")
  1844. }
  1845. if !strings.HasPrefix(err.Error(), "Error 1045") {
  1846. t.Fatal(err.Error())
  1847. }
  1848. }
  1849. }
  1850. // static interface implementation checks of mysqlConn
  1851. var (
  1852. _ driver.ConnBeginTx = &mysqlConn{}
  1853. _ driver.ConnPrepareContext = &mysqlConn{}
  1854. _ driver.ExecerContext = &mysqlConn{}
  1855. _ driver.Pinger = &mysqlConn{}
  1856. _ driver.QueryerContext = &mysqlConn{}
  1857. )
  1858. // static interface implementation checks of mysqlStmt
  1859. var (
  1860. _ driver.StmtExecContext = &mysqlStmt{}
  1861. _ driver.StmtQueryContext = &mysqlStmt{}
  1862. )
  1863. // Ensure that all the driver interfaces are implemented
  1864. var (
  1865. // _ driver.RowsColumnTypeLength = &binaryRows{}
  1866. // _ driver.RowsColumnTypeLength = &textRows{}
  1867. _ driver.RowsColumnTypeDatabaseTypeName = &binaryRows{}
  1868. _ driver.RowsColumnTypeDatabaseTypeName = &textRows{}
  1869. _ driver.RowsColumnTypeNullable = &binaryRows{}
  1870. _ driver.RowsColumnTypeNullable = &textRows{}
  1871. _ driver.RowsColumnTypePrecisionScale = &binaryRows{}
  1872. _ driver.RowsColumnTypePrecisionScale = &textRows{}
  1873. _ driver.RowsColumnTypeScanType = &binaryRows{}
  1874. _ driver.RowsColumnTypeScanType = &textRows{}
  1875. _ driver.RowsNextResultSet = &binaryRows{}
  1876. _ driver.RowsNextResultSet = &textRows{}
  1877. )
  1878. func TestMultiResultSet(t *testing.T) {
  1879. type result struct {
  1880. values [][]int
  1881. columns []string
  1882. }
  1883. // checkRows is a helper test function to validate rows containing 3 result
  1884. // sets with specific values and columns. The basic query would look like this:
  1885. //
  1886. // SELECT 1 AS col1, 2 AS col2 UNION SELECT 3, 4;
  1887. // SELECT 0 UNION SELECT 1;
  1888. // SELECT 1 AS col1, 2 AS col2, 3 AS col3 UNION SELECT 4, 5, 6;
  1889. //
  1890. // to distinguish test cases the first string argument is put in front of
  1891. // every error or fatal message.
  1892. checkRows := func(desc string, rows *sql.Rows, dbt *DBTest) {
  1893. expected := []result{
  1894. {
  1895. values: [][]int{{1, 2}, {3, 4}},
  1896. columns: []string{"col1", "col2"},
  1897. },
  1898. {
  1899. values: [][]int{{1, 2, 3}, {4, 5, 6}},
  1900. columns: []string{"col1", "col2", "col3"},
  1901. },
  1902. }
  1903. var res1 result
  1904. for rows.Next() {
  1905. var res [2]int
  1906. if err := rows.Scan(&res[0], &res[1]); err != nil {
  1907. dbt.Fatal(err)
  1908. }
  1909. res1.values = append(res1.values, res[:])
  1910. }
  1911. cols, err := rows.Columns()
  1912. if err != nil {
  1913. dbt.Fatal(desc, err)
  1914. }
  1915. res1.columns = cols
  1916. if !reflect.DeepEqual(expected[0], res1) {
  1917. dbt.Error(desc, "want =", expected[0], "got =", res1)
  1918. }
  1919. if !rows.NextResultSet() {
  1920. dbt.Fatal(desc, "expected next result set")
  1921. }
  1922. // ignoring one result set
  1923. if !rows.NextResultSet() {
  1924. dbt.Fatal(desc, "expected next result set")
  1925. }
  1926. var res2 result
  1927. cols, err = rows.Columns()
  1928. if err != nil {
  1929. dbt.Fatal(desc, err)
  1930. }
  1931. res2.columns = cols
  1932. for rows.Next() {
  1933. var res [3]int
  1934. if err := rows.Scan(&res[0], &res[1], &res[2]); err != nil {
  1935. dbt.Fatal(desc, err)
  1936. }
  1937. res2.values = append(res2.values, res[:])
  1938. }
  1939. if !reflect.DeepEqual(expected[1], res2) {
  1940. dbt.Error(desc, "want =", expected[1], "got =", res2)
  1941. }
  1942. if rows.NextResultSet() {
  1943. dbt.Error(desc, "unexpected next result set")
  1944. }
  1945. if err := rows.Err(); err != nil {
  1946. dbt.Error(desc, err)
  1947. }
  1948. }
  1949. runTestsWithMultiStatement(t, dsn, func(dbt *DBTest) {
  1950. rows := dbt.mustQuery(`DO 1;
  1951. SELECT 1 AS col1, 2 AS col2 UNION SELECT 3, 4;
  1952. DO 1;
  1953. SELECT 0 UNION SELECT 1;
  1954. SELECT 1 AS col1, 2 AS col2, 3 AS col3 UNION SELECT 4, 5, 6;`)
  1955. defer rows.Close()
  1956. checkRows("query: ", rows, dbt)
  1957. })
  1958. runTestsWithMultiStatement(t, dsn, func(dbt *DBTest) {
  1959. queries := []string{
  1960. `
  1961. DROP PROCEDURE IF EXISTS test_mrss;
  1962. CREATE PROCEDURE test_mrss()
  1963. BEGIN
  1964. DO 1;
  1965. SELECT 1 AS col1, 2 AS col2 UNION SELECT 3, 4;
  1966. DO 1;
  1967. SELECT 0 UNION SELECT 1;
  1968. SELECT 1 AS col1, 2 AS col2, 3 AS col3 UNION SELECT 4, 5, 6;
  1969. END
  1970. `,
  1971. `
  1972. DROP PROCEDURE IF EXISTS test_mrss;
  1973. CREATE PROCEDURE test_mrss()
  1974. BEGIN
  1975. SELECT 1 AS col1, 2 AS col2 UNION SELECT 3, 4;
  1976. SELECT 0 UNION SELECT 1;
  1977. SELECT 1 AS col1, 2 AS col2, 3 AS col3 UNION SELECT 4, 5, 6;
  1978. END
  1979. `,
  1980. }
  1981. defer dbt.mustExec("DROP PROCEDURE IF EXISTS test_mrss")
  1982. for i, query := range queries {
  1983. dbt.mustExec(query)
  1984. stmt, err := dbt.db.Prepare("CALL test_mrss()")
  1985. if err != nil {
  1986. dbt.Fatalf("%v (i=%d)", err, i)
  1987. }
  1988. defer stmt.Close()
  1989. for j := 0; j < 2; j++ {
  1990. rows, err := stmt.Query()
  1991. if err != nil {
  1992. dbt.Fatalf("%v (i=%d) (j=%d)", err, i, j)
  1993. }
  1994. checkRows(fmt.Sprintf("prepared stmt query (i=%d) (j=%d): ", i, j), rows, dbt)
  1995. }
  1996. }
  1997. })
  1998. }
  1999. func TestMultiResultSetNoSelect(t *testing.T) {
  2000. runTestsWithMultiStatement(t, dsn, func(dbt *DBTest) {
  2001. rows := dbt.mustQuery("DO 1; DO 2;")
  2002. defer rows.Close()
  2003. if rows.Next() {
  2004. dbt.Error("unexpected row")
  2005. }
  2006. if rows.NextResultSet() {
  2007. dbt.Error("unexpected next result set")
  2008. }
  2009. if err := rows.Err(); err != nil {
  2010. dbt.Error("expected nil; got ", err)
  2011. }
  2012. })
  2013. }
  2014. // tests if rows are set in a proper state if some results were ignored before
  2015. // calling rows.NextResultSet.
  2016. func TestSkipResults(t *testing.T) {
  2017. runTests(t, dsn, func(dbt *DBTest) {
  2018. rows := dbt.mustQuery("SELECT 1, 2")
  2019. defer rows.Close()
  2020. if !rows.Next() {
  2021. dbt.Error("expected row")
  2022. }
  2023. if rows.NextResultSet() {
  2024. dbt.Error("unexpected next result set")
  2025. }
  2026. if err := rows.Err(); err != nil {
  2027. dbt.Error("expected nil; got ", err)
  2028. }
  2029. })
  2030. }
  2031. func TestPingContext(t *testing.T) {
  2032. runTests(t, dsn, func(dbt *DBTest) {
  2033. ctx, cancel := context.WithCancel(context.Background())
  2034. cancel()
  2035. if err := dbt.db.PingContext(ctx); err != context.Canceled {
  2036. dbt.Errorf("expected context.Canceled, got %v", err)
  2037. }
  2038. })
  2039. }
  2040. func TestContextCancelExec(t *testing.T) {
  2041. runTests(t, dsn, func(dbt *DBTest) {
  2042. dbt.mustExec("CREATE TABLE test (v INTEGER)")
  2043. ctx, cancel := context.WithCancel(context.Background())
  2044. // Delay execution for just a bit until db.ExecContext has begun.
  2045. defer time.AfterFunc(250*time.Millisecond, cancel).Stop()
  2046. // This query will be canceled.
  2047. startTime := time.Now()
  2048. if _, err := dbt.db.ExecContext(ctx, "INSERT INTO test VALUES (SLEEP(1))"); err != context.Canceled {
  2049. dbt.Errorf("expected context.Canceled, got %v", err)
  2050. }
  2051. if d := time.Since(startTime); d > 500*time.Millisecond {
  2052. dbt.Errorf("too long execution time: %s", d)
  2053. }
  2054. // Wait for the INSERT query to be done.
  2055. time.Sleep(time.Second)
  2056. // Check how many times the query is executed.
  2057. var v int
  2058. if err := dbt.db.QueryRow("SELECT COUNT(*) FROM test").Scan(&v); err != nil {
  2059. dbt.Fatalf("%s", err.Error())
  2060. }
  2061. if v != 1 { // TODO: need to kill the query, and v should be 0.
  2062. dbt.Skipf("[WARN] expected val to be 1, got %d", v)
  2063. }
  2064. // Context is already canceled, so error should come before execution.
  2065. if _, err := dbt.db.ExecContext(ctx, "INSERT INTO test VALUES (1)"); err == nil {
  2066. dbt.Error("expected error")
  2067. } else if err.Error() != "context canceled" {
  2068. dbt.Fatalf("unexpected error: %s", err)
  2069. }
  2070. // The second insert query will fail, so the table has no changes.
  2071. if err := dbt.db.QueryRow("SELECT COUNT(*) FROM test").Scan(&v); err != nil {
  2072. dbt.Fatalf("%s", err.Error())
  2073. }
  2074. if v != 1 {
  2075. dbt.Skipf("[WARN] expected val to be 1, got %d", v)
  2076. }
  2077. })
  2078. }
  2079. func TestContextCancelQuery(t *testing.T) {
  2080. runTests(t, dsn, func(dbt *DBTest) {
  2081. dbt.mustExec("CREATE TABLE test (v INTEGER)")
  2082. ctx, cancel := context.WithCancel(context.Background())
  2083. // Delay execution for just a bit until db.ExecContext has begun.
  2084. defer time.AfterFunc(250*time.Millisecond, cancel).Stop()
  2085. // This query will be canceled.
  2086. startTime := time.Now()
  2087. if _, err := dbt.db.QueryContext(ctx, "INSERT INTO test VALUES (SLEEP(1))"); err != context.Canceled {
  2088. dbt.Errorf("expected context.Canceled, got %v", err)
  2089. }
  2090. if d := time.Since(startTime); d > 500*time.Millisecond {
  2091. dbt.Errorf("too long execution time: %s", d)
  2092. }
  2093. // Wait for the INSERT query to be done.
  2094. time.Sleep(time.Second)
  2095. // Check how many times the query is executed.
  2096. var v int
  2097. if err := dbt.db.QueryRow("SELECT COUNT(*) FROM test").Scan(&v); err != nil {
  2098. dbt.Fatalf("%s", err.Error())
  2099. }
  2100. if v != 1 { // TODO: need to kill the query, and v should be 0.
  2101. dbt.Skipf("[WARN] expected val to be 1, got %d", v)
  2102. }
  2103. // Context is already canceled, so error should come before execution.
  2104. if _, err := dbt.db.QueryContext(ctx, "INSERT INTO test VALUES (1)"); err != context.Canceled {
  2105. dbt.Errorf("expected context.Canceled, got %v", err)
  2106. }
  2107. // The second insert query will fail, so the table has no changes.
  2108. if err := dbt.db.QueryRow("SELECT COUNT(*) FROM test").Scan(&v); err != nil {
  2109. dbt.Fatalf("%s", err.Error())
  2110. }
  2111. if v != 1 {
  2112. dbt.Skipf("[WARN] expected val to be 1, got %d", v)
  2113. }
  2114. })
  2115. }
  2116. func TestContextCancelQueryRow(t *testing.T) {
  2117. runTests(t, dsn, func(dbt *DBTest) {
  2118. dbt.mustExec("CREATE TABLE test (v INTEGER)")
  2119. dbt.mustExec("INSERT INTO test VALUES (1), (2), (3)")
  2120. ctx, cancel := context.WithCancel(context.Background())
  2121. rows, err := dbt.db.QueryContext(ctx, "SELECT v FROM test")
  2122. if err != nil {
  2123. dbt.Fatalf("%s", err.Error())
  2124. }
  2125. // the first row will be succeed.
  2126. var v int
  2127. if !rows.Next() {
  2128. dbt.Fatalf("unexpected end")
  2129. }
  2130. if err := rows.Scan(&v); err != nil {
  2131. dbt.Fatalf("%s", err.Error())
  2132. }
  2133. cancel()
  2134. // make sure the driver receives the cancel request.
  2135. time.Sleep(100 * time.Millisecond)
  2136. if rows.Next() {
  2137. dbt.Errorf("expected end, but not")
  2138. }
  2139. if err := rows.Err(); err != context.Canceled {
  2140. dbt.Errorf("expected context.Canceled, got %v", err)
  2141. }
  2142. })
  2143. }
  2144. func TestContextCancelPrepare(t *testing.T) {
  2145. runTests(t, dsn, func(dbt *DBTest) {
  2146. ctx, cancel := context.WithCancel(context.Background())
  2147. cancel()
  2148. if _, err := dbt.db.PrepareContext(ctx, "SELECT 1"); err != context.Canceled {
  2149. dbt.Errorf("expected context.Canceled, got %v", err)
  2150. }
  2151. })
  2152. }
  2153. func TestContextCancelStmtExec(t *testing.T) {
  2154. runTests(t, dsn, func(dbt *DBTest) {
  2155. dbt.mustExec("CREATE TABLE test (v INTEGER)")
  2156. ctx, cancel := context.WithCancel(context.Background())
  2157. stmt, err := dbt.db.PrepareContext(ctx, "INSERT INTO test VALUES (SLEEP(1))")
  2158. if err != nil {
  2159. dbt.Fatalf("unexpected error: %v", err)
  2160. }
  2161. // Delay execution for just a bit until db.ExecContext has begun.
  2162. defer time.AfterFunc(250*time.Millisecond, cancel).Stop()
  2163. // This query will be canceled.
  2164. startTime := time.Now()
  2165. if _, err := stmt.ExecContext(ctx); err != context.Canceled {
  2166. dbt.Errorf("expected context.Canceled, got %v", err)
  2167. }
  2168. if d := time.Since(startTime); d > 500*time.Millisecond {
  2169. dbt.Errorf("too long execution time: %s", d)
  2170. }
  2171. // Wait for the INSERT query to be done.
  2172. time.Sleep(time.Second)
  2173. // Check how many times the query is executed.
  2174. var v int
  2175. if err := dbt.db.QueryRow("SELECT COUNT(*) FROM test").Scan(&v); err != nil {
  2176. dbt.Fatalf("%s", err.Error())
  2177. }
  2178. if v != 1 { // TODO: need to kill the query, and v should be 0.
  2179. dbt.Skipf("[WARN] expected val to be 1, got %d", v)
  2180. }
  2181. })
  2182. }
  2183. func TestContextCancelStmtQuery(t *testing.T) {
  2184. runTests(t, dsn, func(dbt *DBTest) {
  2185. dbt.mustExec("CREATE TABLE test (v INTEGER)")
  2186. ctx, cancel := context.WithCancel(context.Background())
  2187. stmt, err := dbt.db.PrepareContext(ctx, "INSERT INTO test VALUES (SLEEP(1))")
  2188. if err != nil {
  2189. dbt.Fatalf("unexpected error: %v", err)
  2190. }
  2191. // Delay execution for just a bit until db.ExecContext has begun.
  2192. defer time.AfterFunc(250*time.Millisecond, cancel).Stop()
  2193. // This query will be canceled.
  2194. startTime := time.Now()
  2195. if _, err := stmt.QueryContext(ctx); err != context.Canceled {
  2196. dbt.Errorf("expected context.Canceled, got %v", err)
  2197. }
  2198. if d := time.Since(startTime); d > 500*time.Millisecond {
  2199. dbt.Errorf("too long execution time: %s", d)
  2200. }
  2201. // Wait for the INSERT query has done.
  2202. time.Sleep(time.Second)
  2203. // Check how many times the query is executed.
  2204. var v int
  2205. if err := dbt.db.QueryRow("SELECT COUNT(*) FROM test").Scan(&v); err != nil {
  2206. dbt.Fatalf("%s", err.Error())
  2207. }
  2208. if v != 1 { // TODO: need to kill the query, and v should be 0.
  2209. dbt.Skipf("[WARN] expected val to be 1, got %d", v)
  2210. }
  2211. })
  2212. }
  2213. func TestContextCancelBegin(t *testing.T) {
  2214. runTests(t, dsn, func(dbt *DBTest) {
  2215. dbt.mustExec("CREATE TABLE test (v INTEGER)")
  2216. ctx, cancel := context.WithCancel(context.Background())
  2217. tx, err := dbt.db.BeginTx(ctx, nil)
  2218. if err != nil {
  2219. dbt.Fatal(err)
  2220. }
  2221. // Delay execution for just a bit until db.ExecContext has begun.
  2222. defer time.AfterFunc(100*time.Millisecond, cancel).Stop()
  2223. // This query will be canceled.
  2224. startTime := time.Now()
  2225. if _, err := tx.ExecContext(ctx, "INSERT INTO test VALUES (SLEEP(1))"); err != context.Canceled {
  2226. dbt.Errorf("expected context.Canceled, got %v", err)
  2227. }
  2228. if d := time.Since(startTime); d > 500*time.Millisecond {
  2229. dbt.Errorf("too long execution time: %s", d)
  2230. }
  2231. // Transaction is canceled, so expect an error.
  2232. switch err := tx.Commit(); err {
  2233. case sql.ErrTxDone:
  2234. // because the transaction has already been rollbacked.
  2235. // the database/sql package watches ctx
  2236. // and rollbacks when ctx is canceled.
  2237. case context.Canceled:
  2238. // the database/sql package rollbacks on another goroutine,
  2239. // so the transaction may not be rollbacked depending on goroutine scheduling.
  2240. default:
  2241. dbt.Errorf("expected sql.ErrTxDone or context.Canceled, got %v", err)
  2242. }
  2243. // Context is canceled, so cannot begin a transaction.
  2244. if _, err := dbt.db.BeginTx(ctx, nil); err != context.Canceled {
  2245. dbt.Errorf("expected context.Canceled, got %v", err)
  2246. }
  2247. })
  2248. }
  2249. func TestContextBeginIsolationLevel(t *testing.T) {
  2250. runTests(t, dsn, func(dbt *DBTest) {
  2251. dbt.mustExec("CREATE TABLE test (v INTEGER)")
  2252. ctx, cancel := context.WithCancel(context.Background())
  2253. defer cancel()
  2254. tx1, err := dbt.db.BeginTx(ctx, &sql.TxOptions{
  2255. Isolation: sql.LevelRepeatableRead,
  2256. })
  2257. if err != nil {
  2258. dbt.Fatal(err)
  2259. }
  2260. tx2, err := dbt.db.BeginTx(ctx, &sql.TxOptions{
  2261. Isolation: sql.LevelReadCommitted,
  2262. })
  2263. if err != nil {
  2264. dbt.Fatal(err)
  2265. }
  2266. _, err = tx1.ExecContext(ctx, "INSERT INTO test VALUES (1)")
  2267. if err != nil {
  2268. dbt.Fatal(err)
  2269. }
  2270. var v int
  2271. row := tx2.QueryRowContext(ctx, "SELECT COUNT(*) FROM test")
  2272. if err := row.Scan(&v); err != nil {
  2273. dbt.Fatal(err)
  2274. }
  2275. // Because writer transaction wasn't commited yet, it should be available
  2276. if v != 0 {
  2277. dbt.Errorf("expected val to be 0, got %d", v)
  2278. }
  2279. err = tx1.Commit()
  2280. if err != nil {
  2281. dbt.Fatal(err)
  2282. }
  2283. row = tx2.QueryRowContext(ctx, "SELECT COUNT(*) FROM test")
  2284. if err := row.Scan(&v); err != nil {
  2285. dbt.Fatal(err)
  2286. }
  2287. // Data written by writer transaction is already commited, it should be selectable
  2288. if v != 1 {
  2289. dbt.Errorf("expected val to be 1, got %d", v)
  2290. }
  2291. tx2.Commit()
  2292. })
  2293. }
  2294. func TestContextBeginReadOnly(t *testing.T) {
  2295. runTests(t, dsn, func(dbt *DBTest) {
  2296. dbt.mustExec("CREATE TABLE test (v INTEGER)")
  2297. ctx, cancel := context.WithCancel(context.Background())
  2298. defer cancel()
  2299. tx, err := dbt.db.BeginTx(ctx, &sql.TxOptions{
  2300. ReadOnly: true,
  2301. })
  2302. if _, ok := err.(*MySQLError); ok {
  2303. dbt.Skip("It seems that your MySQL does not support READ ONLY transactions")
  2304. return
  2305. } else if err != nil {
  2306. dbt.Fatal(err)
  2307. }
  2308. // INSERT queries fail in a READ ONLY transaction.
  2309. _, err = tx.ExecContext(ctx, "INSERT INTO test VALUES (1)")
  2310. if _, ok := err.(*MySQLError); !ok {
  2311. dbt.Errorf("expected MySQLError, got %v", err)
  2312. }
  2313. // SELECT queries can be executed.
  2314. var v int
  2315. row := tx.QueryRowContext(ctx, "SELECT COUNT(*) FROM test")
  2316. if err := row.Scan(&v); err != nil {
  2317. dbt.Fatal(err)
  2318. }
  2319. if v != 0 {
  2320. dbt.Errorf("expected val to be 0, got %d", v)
  2321. }
  2322. if err := tx.Commit(); err != nil {
  2323. dbt.Fatal(err)
  2324. }
  2325. })
  2326. }
  2327. func TestRowsColumnTypes(t *testing.T) {
  2328. niNULL := sql.NullInt64{Int64: 0, Valid: false}
  2329. ni0 := sql.NullInt64{Int64: 0, Valid: true}
  2330. ni1 := sql.NullInt64{Int64: 1, Valid: true}
  2331. ni42 := sql.NullInt64{Int64: 42, Valid: true}
  2332. nfNULL := sql.NullFloat64{Float64: 0.0, Valid: false}
  2333. nf0 := sql.NullFloat64{Float64: 0.0, Valid: true}
  2334. nf1337 := sql.NullFloat64{Float64: 13.37, Valid: true}
  2335. nt0 := NullTime{Time: time.Date(2006, 01, 02, 15, 04, 05, 0, time.UTC), Valid: true}
  2336. nt1 := NullTime{Time: time.Date(2006, 01, 02, 15, 04, 05, 100000000, time.UTC), Valid: true}
  2337. nt2 := NullTime{Time: time.Date(2006, 01, 02, 15, 04, 05, 110000000, time.UTC), Valid: true}
  2338. nt6 := NullTime{Time: time.Date(2006, 01, 02, 15, 04, 05, 111111000, time.UTC), Valid: true}
  2339. nd1 := NullTime{Time: time.Date(2006, 01, 02, 0, 0, 0, 0, time.UTC), Valid: true}
  2340. nd2 := NullTime{Time: time.Date(2006, 03, 04, 0, 0, 0, 0, time.UTC), Valid: true}
  2341. ndNULL := NullTime{Time: time.Time{}, Valid: false}
  2342. rbNULL := sql.RawBytes(nil)
  2343. rb0 := sql.RawBytes("0")
  2344. rb42 := sql.RawBytes("42")
  2345. rbTest := sql.RawBytes("Test")
  2346. rb0pad4 := sql.RawBytes("0\x00\x00\x00") // BINARY right-pads values with 0x00
  2347. rbx0 := sql.RawBytes("\x00")
  2348. rbx42 := sql.RawBytes("\x42")
  2349. var columns = []struct {
  2350. name string
  2351. fieldType string // type used when creating table schema
  2352. databaseTypeName string // actual type used by MySQL
  2353. scanType reflect.Type
  2354. nullable bool
  2355. precision int64 // 0 if not ok
  2356. scale int64
  2357. valuesIn [3]string
  2358. valuesOut [3]interface{}
  2359. }{
  2360. {"bit8null", "BIT(8)", "BIT", scanTypeRawBytes, true, 0, 0, [3]string{"0x0", "NULL", "0x42"}, [3]interface{}{rbx0, rbNULL, rbx42}},
  2361. {"boolnull", "BOOL", "TINYINT", scanTypeNullInt, true, 0, 0, [3]string{"NULL", "true", "0"}, [3]interface{}{niNULL, ni1, ni0}},
  2362. {"bool", "BOOL NOT NULL", "TINYINT", scanTypeInt8, false, 0, 0, [3]string{"1", "0", "FALSE"}, [3]interface{}{int8(1), int8(0), int8(0)}},
  2363. {"intnull", "INTEGER", "INT", scanTypeNullInt, true, 0, 0, [3]string{"0", "NULL", "42"}, [3]interface{}{ni0, niNULL, ni42}},
  2364. {"smallint", "SMALLINT NOT NULL", "SMALLINT", scanTypeInt16, false, 0, 0, [3]string{"0", "-32768", "32767"}, [3]interface{}{int16(0), int16(-32768), int16(32767)}},
  2365. {"smallintnull", "SMALLINT", "SMALLINT", scanTypeNullInt, true, 0, 0, [3]string{"0", "NULL", "42"}, [3]interface{}{ni0, niNULL, ni42}},
  2366. {"int3null", "INT(3)", "INT", scanTypeNullInt, true, 0, 0, [3]string{"0", "NULL", "42"}, [3]interface{}{ni0, niNULL, ni42}},
  2367. {"int7", "INT(7) NOT NULL", "INT", scanTypeInt32, false, 0, 0, [3]string{"0", "-1337", "42"}, [3]interface{}{int32(0), int32(-1337), int32(42)}},
  2368. {"mediumintnull", "MEDIUMINT", "MEDIUMINT", scanTypeNullInt, true, 0, 0, [3]string{"0", "42", "NULL"}, [3]interface{}{ni0, ni42, niNULL}},
  2369. {"bigint", "BIGINT NOT NULL", "BIGINT", scanTypeInt64, false, 0, 0, [3]string{"0", "65535", "-42"}, [3]interface{}{int64(0), int64(65535), int64(-42)}},
  2370. {"bigintnull", "BIGINT", "BIGINT", scanTypeNullInt, true, 0, 0, [3]string{"NULL", "1", "42"}, [3]interface{}{niNULL, ni1, ni42}},
  2371. {"tinyuint", "TINYINT UNSIGNED NOT NULL", "TINYINT", scanTypeUint8, false, 0, 0, [3]string{"0", "255", "42"}, [3]interface{}{uint8(0), uint8(255), uint8(42)}},
  2372. {"smalluint", "SMALLINT UNSIGNED NOT NULL", "SMALLINT", scanTypeUint16, false, 0, 0, [3]string{"0", "65535", "42"}, [3]interface{}{uint16(0), uint16(65535), uint16(42)}},
  2373. {"biguint", "BIGINT UNSIGNED NOT NULL", "BIGINT", scanTypeUint64, false, 0, 0, [3]string{"0", "65535", "42"}, [3]interface{}{uint64(0), uint64(65535), uint64(42)}},
  2374. {"uint13", "INT(13) UNSIGNED NOT NULL", "INT", scanTypeUint32, false, 0, 0, [3]string{"0", "1337", "42"}, [3]interface{}{uint32(0), uint32(1337), uint32(42)}},
  2375. {"float", "FLOAT NOT NULL", "FLOAT", scanTypeFloat32, false, math.MaxInt64, math.MaxInt64, [3]string{"0", "42", "13.37"}, [3]interface{}{float32(0), float32(42), float32(13.37)}},
  2376. {"floatnull", "FLOAT", "FLOAT", scanTypeNullFloat, true, math.MaxInt64, math.MaxInt64, [3]string{"0", "NULL", "13.37"}, [3]interface{}{nf0, nfNULL, nf1337}},
  2377. {"float74null", "FLOAT(7,4)", "FLOAT", scanTypeNullFloat, true, math.MaxInt64, 4, [3]string{"0", "NULL", "13.37"}, [3]interface{}{nf0, nfNULL, nf1337}},
  2378. {"double", "DOUBLE NOT NULL", "DOUBLE", scanTypeFloat64, false, math.MaxInt64, math.MaxInt64, [3]string{"0", "42", "13.37"}, [3]interface{}{float64(0), float64(42), float64(13.37)}},
  2379. {"doublenull", "DOUBLE", "DOUBLE", scanTypeNullFloat, true, math.MaxInt64, math.MaxInt64, [3]string{"0", "NULL", "13.37"}, [3]interface{}{nf0, nfNULL, nf1337}},
  2380. {"decimal1", "DECIMAL(10,6) NOT NULL", "DECIMAL", scanTypeRawBytes, false, 10, 6, [3]string{"0", "13.37", "1234.123456"}, [3]interface{}{sql.RawBytes("0.000000"), sql.RawBytes("13.370000"), sql.RawBytes("1234.123456")}},
  2381. {"decimal1null", "DECIMAL(10,6)", "DECIMAL", scanTypeRawBytes, true, 10, 6, [3]string{"0", "NULL", "1234.123456"}, [3]interface{}{sql.RawBytes("0.000000"), rbNULL, sql.RawBytes("1234.123456")}},
  2382. {"decimal2", "DECIMAL(8,4) NOT NULL", "DECIMAL", scanTypeRawBytes, false, 8, 4, [3]string{"0", "13.37", "1234.123456"}, [3]interface{}{sql.RawBytes("0.0000"), sql.RawBytes("13.3700"), sql.RawBytes("1234.1235")}},
  2383. {"decimal2null", "DECIMAL(8,4)", "DECIMAL", scanTypeRawBytes, true, 8, 4, [3]string{"0", "NULL", "1234.123456"}, [3]interface{}{sql.RawBytes("0.0000"), rbNULL, sql.RawBytes("1234.1235")}},
  2384. {"decimal3", "DECIMAL(5,0) NOT NULL", "DECIMAL", scanTypeRawBytes, false, 5, 0, [3]string{"0", "13.37", "-12345.123456"}, [3]interface{}{rb0, sql.RawBytes("13"), sql.RawBytes("-12345")}},
  2385. {"decimal3null", "DECIMAL(5,0)", "DECIMAL", scanTypeRawBytes, true, 5, 0, [3]string{"0", "NULL", "-12345.123456"}, [3]interface{}{rb0, rbNULL, sql.RawBytes("-12345")}},
  2386. {"char25null", "CHAR(25)", "CHAR", scanTypeRawBytes, true, 0, 0, [3]string{"0", "NULL", "'Test'"}, [3]interface{}{rb0, rbNULL, rbTest}},
  2387. {"varchar42", "VARCHAR(42) NOT NULL", "VARCHAR", scanTypeRawBytes, false, 0, 0, [3]string{"0", "'Test'", "42"}, [3]interface{}{rb0, rbTest, rb42}},
  2388. {"binary4null", "BINARY(4)", "BINARY", scanTypeRawBytes, true, 0, 0, [3]string{"0", "NULL", "'Test'"}, [3]interface{}{rb0pad4, rbNULL, rbTest}},
  2389. {"varbinary42", "VARBINARY(42) NOT NULL", "VARBINARY", scanTypeRawBytes, false, 0, 0, [3]string{"0", "'Test'", "42"}, [3]interface{}{rb0, rbTest, rb42}},
  2390. {"tinyblobnull", "TINYBLOB", "BLOB", scanTypeRawBytes, true, 0, 0, [3]string{"0", "NULL", "'Test'"}, [3]interface{}{rb0, rbNULL, rbTest}},
  2391. {"tinytextnull", "TINYTEXT", "TEXT", scanTypeRawBytes, true, 0, 0, [3]string{"0", "NULL", "'Test'"}, [3]interface{}{rb0, rbNULL, rbTest}},
  2392. {"blobnull", "BLOB", "BLOB", scanTypeRawBytes, true, 0, 0, [3]string{"0", "NULL", "'Test'"}, [3]interface{}{rb0, rbNULL, rbTest}},
  2393. {"textnull", "TEXT", "TEXT", scanTypeRawBytes, true, 0, 0, [3]string{"0", "NULL", "'Test'"}, [3]interface{}{rb0, rbNULL, rbTest}},
  2394. {"mediumblob", "MEDIUMBLOB NOT NULL", "BLOB", scanTypeRawBytes, false, 0, 0, [3]string{"0", "'Test'", "42"}, [3]interface{}{rb0, rbTest, rb42}},
  2395. {"mediumtext", "MEDIUMTEXT NOT NULL", "TEXT", scanTypeRawBytes, false, 0, 0, [3]string{"0", "'Test'", "42"}, [3]interface{}{rb0, rbTest, rb42}},
  2396. {"longblob", "LONGBLOB NOT NULL", "BLOB", scanTypeRawBytes, false, 0, 0, [3]string{"0", "'Test'", "42"}, [3]interface{}{rb0, rbTest, rb42}},
  2397. {"longtext", "LONGTEXT NOT NULL", "TEXT", scanTypeRawBytes, false, 0, 0, [3]string{"0", "'Test'", "42"}, [3]interface{}{rb0, rbTest, rb42}},
  2398. {"datetime", "DATETIME", "DATETIME", scanTypeNullTime, true, 0, 0, [3]string{"'2006-01-02 15:04:05'", "'2006-01-02 15:04:05.1'", "'2006-01-02 15:04:05.111111'"}, [3]interface{}{nt0, nt0, nt0}},
  2399. {"datetime2", "DATETIME(2)", "DATETIME", scanTypeNullTime, true, 2, 2, [3]string{"'2006-01-02 15:04:05'", "'2006-01-02 15:04:05.1'", "'2006-01-02 15:04:05.111111'"}, [3]interface{}{nt0, nt1, nt2}},
  2400. {"datetime6", "DATETIME(6)", "DATETIME", scanTypeNullTime, true, 6, 6, [3]string{"'2006-01-02 15:04:05'", "'2006-01-02 15:04:05.1'", "'2006-01-02 15:04:05.111111'"}, [3]interface{}{nt0, nt1, nt6}},
  2401. {"date", "DATE", "DATE", scanTypeNullTime, true, 0, 0, [3]string{"'2006-01-02'", "NULL", "'2006-03-04'"}, [3]interface{}{nd1, ndNULL, nd2}},
  2402. {"year", "YEAR NOT NULL", "YEAR", scanTypeUint16, false, 0, 0, [3]string{"2006", "2000", "1994"}, [3]interface{}{uint16(2006), uint16(2000), uint16(1994)}},
  2403. }
  2404. schema := ""
  2405. values1 := ""
  2406. values2 := ""
  2407. values3 := ""
  2408. for _, column := range columns {
  2409. schema += fmt.Sprintf("`%s` %s, ", column.name, column.fieldType)
  2410. values1 += column.valuesIn[0] + ", "
  2411. values2 += column.valuesIn[1] + ", "
  2412. values3 += column.valuesIn[2] + ", "
  2413. }
  2414. schema = schema[:len(schema)-2]
  2415. values1 = values1[:len(values1)-2]
  2416. values2 = values2[:len(values2)-2]
  2417. values3 = values3[:len(values3)-2]
  2418. dsns := []string{
  2419. dsn + "&parseTime=true",
  2420. dsn + "&parseTime=false",
  2421. }
  2422. for _, testdsn := range dsns {
  2423. runTests(t, testdsn, func(dbt *DBTest) {
  2424. dbt.mustExec("CREATE TABLE test (" + schema + ")")
  2425. dbt.mustExec("INSERT INTO test VALUES (" + values1 + "), (" + values2 + "), (" + values3 + ")")
  2426. rows, err := dbt.db.Query("SELECT * FROM test")
  2427. if err != nil {
  2428. t.Fatalf("Query: %v", err)
  2429. }
  2430. tt, err := rows.ColumnTypes()
  2431. if err != nil {
  2432. t.Fatalf("ColumnTypes: %v", err)
  2433. }
  2434. if len(tt) != len(columns) {
  2435. t.Fatalf("unexpected number of columns: expected %d, got %d", len(columns), len(tt))
  2436. }
  2437. types := make([]reflect.Type, len(tt))
  2438. for i, tp := range tt {
  2439. column := columns[i]
  2440. // Name
  2441. name := tp.Name()
  2442. if name != column.name {
  2443. t.Errorf("column name mismatch %s != %s", name, column.name)
  2444. continue
  2445. }
  2446. // DatabaseTypeName
  2447. databaseTypeName := tp.DatabaseTypeName()
  2448. if databaseTypeName != column.databaseTypeName {
  2449. t.Errorf("databasetypename name mismatch for column %q: %s != %s", name, databaseTypeName, column.databaseTypeName)
  2450. continue
  2451. }
  2452. // ScanType
  2453. scanType := tp.ScanType()
  2454. if scanType != column.scanType {
  2455. if scanType == nil {
  2456. t.Errorf("scantype is null for column %q", name)
  2457. } else {
  2458. t.Errorf("scantype mismatch for column %q: %s != %s", name, scanType.Name(), column.scanType.Name())
  2459. }
  2460. continue
  2461. }
  2462. types[i] = scanType
  2463. // Nullable
  2464. nullable, ok := tp.Nullable()
  2465. if !ok {
  2466. t.Errorf("nullable not ok %q", name)
  2467. continue
  2468. }
  2469. if nullable != column.nullable {
  2470. t.Errorf("nullable mismatch for column %q: %t != %t", name, nullable, column.nullable)
  2471. }
  2472. // Length
  2473. // length, ok := tp.Length()
  2474. // if length != column.length {
  2475. // if !ok {
  2476. // t.Errorf("length not ok for column %q", name)
  2477. // } else {
  2478. // t.Errorf("length mismatch for column %q: %d != %d", name, length, column.length)
  2479. // }
  2480. // continue
  2481. // }
  2482. // Precision and Scale
  2483. precision, scale, ok := tp.DecimalSize()
  2484. if precision != column.precision {
  2485. if !ok {
  2486. t.Errorf("precision not ok for column %q", name)
  2487. } else {
  2488. t.Errorf("precision mismatch for column %q: %d != %d", name, precision, column.precision)
  2489. }
  2490. continue
  2491. }
  2492. if scale != column.scale {
  2493. if !ok {
  2494. t.Errorf("scale not ok for column %q", name)
  2495. } else {
  2496. t.Errorf("scale mismatch for column %q: %d != %d", name, scale, column.scale)
  2497. }
  2498. continue
  2499. }
  2500. }
  2501. values := make([]interface{}, len(tt))
  2502. for i := range values {
  2503. values[i] = reflect.New(types[i]).Interface()
  2504. }
  2505. i := 0
  2506. for rows.Next() {
  2507. err = rows.Scan(values...)
  2508. if err != nil {
  2509. t.Fatalf("failed to scan values in %v", err)
  2510. }
  2511. for j := range values {
  2512. value := reflect.ValueOf(values[j]).Elem().Interface()
  2513. if !reflect.DeepEqual(value, columns[j].valuesOut[i]) {
  2514. if columns[j].scanType == scanTypeRawBytes {
  2515. t.Errorf("row %d, column %d: %v != %v", i, j, string(value.(sql.RawBytes)), string(columns[j].valuesOut[i].(sql.RawBytes)))
  2516. } else {
  2517. t.Errorf("row %d, column %d: %v != %v", i, j, value, columns[j].valuesOut[i])
  2518. }
  2519. }
  2520. }
  2521. i++
  2522. }
  2523. if i != 3 {
  2524. t.Errorf("expected 3 rows, got %d", i)
  2525. }
  2526. if err := rows.Close(); err != nil {
  2527. t.Errorf("error closing rows: %s", err)
  2528. }
  2529. })
  2530. }
  2531. }
  2532. func TestValuerWithValueReceiverGivenNilValue(t *testing.T) {
  2533. runTests(t, dsn, func(dbt *DBTest) {
  2534. dbt.mustExec("CREATE TABLE test (value VARCHAR(255))")
  2535. dbt.db.Exec("INSERT INTO test VALUES (?)", (*testValuer)(nil))
  2536. // This test will panic on the INSERT if ConvertValue() does not check for typed nil before calling Value()
  2537. })
  2538. }