driver_test.go 79 KB

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