mysql_tests.erl 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515
  1. %% MySQL/OTP – MySQL client library for Erlang/OTP
  2. %% Copyright (C) 2014 Viktor Söderqvist
  3. %%
  4. %% This file is part of MySQL/OTP.
  5. %%
  6. %% MySQL/OTP is free software: you can redistribute it and/or modify it under
  7. %% the terms of the GNU Lesser General Public License as published by the Free
  8. %% Software Foundation, either version 3 of the License, or (at your option)
  9. %% any later version.
  10. %%
  11. %% This program is distributed in the hope that it will be useful, but WITHOUT
  12. %% ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13. %% FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  14. %% more details.
  15. %%
  16. %% You should have received a copy of the GNU Lesser General Public License
  17. %% along with this program. If not, see <https://www.gnu.org/licenses/>.
  18. %% @doc This module performs test to an actual database.
  19. -module(mysql_tests).
  20. -include_lib("eunit/include/eunit.hrl").
  21. -define(user, "otptest").
  22. -define(password, "otptest").
  23. -define(create_table_t, <<"CREATE TABLE t ("
  24. " id INT NOT NULL PRIMARY KEY AUTO_INCREMENT,"
  25. " bl BLOB,"
  26. " tx TEXT NOT NULL," %% No default value
  27. " f FLOAT,"
  28. " d DOUBLE,"
  29. " dc DECIMAL(5,3),"
  30. " y YEAR,"
  31. " ti TIME,"
  32. " ts TIMESTAMP,"
  33. " da DATE,"
  34. " c CHAR(2)"
  35. ") ENGINE=InnoDB">>).
  36. connect_test() ->
  37. %% A connection with a registered name
  38. Options = [{name, {local, tardis}}, {user, ?user}, {password, ?password}],
  39. {ok, Pid} = mysql:start_link(Options),
  40. %% Test some gen_server callbacks not tested elsewhere
  41. State = get_state(Pid),
  42. ?assertMatch({ok, State}, mysql:code_change("0.1.0", State, [])),
  43. ?assertMatch({error, _}, mysql:code_change("2.0.0", unknown_state, [])),
  44. exit(whereis(tardis), normal).
  45. keep_alive_test() ->
  46. %% Let the connection send a few pings.
  47. process_flag(trap_exit, true),
  48. Options = [{user, ?user}, {password, ?password}, {keep_alive, 20}],
  49. {ok, Pid} = mysql:start_link(Options),
  50. receive after 70 -> ok end,
  51. State = get_state(Pid),
  52. [state, _Version, _ConnectionId, Socket | _] = tuple_to_list(State),
  53. {ok, ExitMessage, LoggedErrors} = error_logger_acc:capture(fun () ->
  54. gen_tcp:close(Socket),
  55. receive
  56. Message -> Message
  57. after 1000 ->
  58. ping_didnt_crash_connection
  59. end
  60. end),
  61. process_flag(trap_exit, false),
  62. %% Check that we got the expected crash report in the error log.
  63. ?assertMatch({'EXIT', Pid, _Reason}, ExitMessage),
  64. [{error, LoggedMsg}, {error_report, LoggedReport}] = LoggedErrors,
  65. ExpectedPrefix = io_lib:format("** Generic server ~p terminating", [Pid]),
  66. ?assert(lists:prefix(lists:flatten(ExpectedPrefix), LoggedMsg)),
  67. ?assertMatch({crash_report, _}, LoggedReport),
  68. exit(Pid, normal).
  69. %% For R16B where sys:get_state/1 is not available.
  70. get_state(Process) ->
  71. {status,_,_,[_,_,_,_,Misc]} = sys:get_status(Process),
  72. hd([State || {data,[{"State", State}]} <- Misc]).
  73. query_test_() ->
  74. {setup,
  75. fun () ->
  76. {ok, Pid} = mysql:start_link([{user, ?user}, {password, ?password},
  77. {log_warnings, false},
  78. {keep_alive, true}]),
  79. ok = mysql:query(Pid, <<"DROP DATABASE IF EXISTS otptest">>),
  80. ok = mysql:query(Pid, <<"CREATE DATABASE otptest">>),
  81. ok = mysql:query(Pid, <<"USE otptest">>),
  82. ok = mysql:query(Pid, <<"SET autocommit = 1">>),
  83. Pid
  84. end,
  85. fun (Pid) ->
  86. ok = mysql:query(Pid, <<"DROP DATABASE otptest">>),
  87. exit(Pid, normal)
  88. end,
  89. fun (Pid) ->
  90. [{"Select db on connect", fun () -> connect_with_db(Pid) end},
  91. {"Autocommit", fun () -> autocommit(Pid) end},
  92. {"Basic queries", fun () -> basic_queries(Pid) end},
  93. {"Text protocol", fun () -> text_protocol(Pid) end},
  94. {"Binary protocol", fun () -> binary_protocol(Pid) end},
  95. {"FLOAT rounding", fun () -> float_rounding(Pid) end},
  96. {"DECIMAL", fun () -> decimal(Pid) end},
  97. {"INT", fun () -> int(Pid) end},
  98. {"BIT(N)", fun () -> bit(Pid) end},
  99. {"DATE", fun () -> date(Pid) end},
  100. {"TIME", fun () -> time(Pid) end},
  101. {"DATETIME", fun () -> datetime(Pid) end},
  102. {"Microseconds", fun () -> microseconds(Pid) end}]
  103. end}.
  104. connect_with_db(_Pid) ->
  105. %% Make another connection and set the db in the handshake phase
  106. {ok, Pid} = mysql:start_link([{user, ?user}, {password, ?password},
  107. {database, "otptest"}]),
  108. ?assertMatch({ok, _, [[<<"otptest">>]]},
  109. mysql:query(Pid, "SELECT DATABASE()")),
  110. %% Capture error log to check that we get a warning logged
  111. {ok, ok, LoggedErrors} = error_logger_acc:capture(fun () ->
  112. mysql:query(Pid, <<"DROP TABLE IF EXISTS dummy">>)
  113. end),
  114. ?assertMatch([{_, "Note 1051: Unknown table 'dummy'\n"
  115. " in DROP TABLE IF EXISTS dummy\n"}],
  116. LoggedErrors),
  117. exit(Pid, normal).
  118. autocommit(Pid) ->
  119. ?assert(mysql:autocommit(Pid)),
  120. ok = mysql:query(Pid, <<"SET autocommit = 0">>),
  121. ?assertNot(mysql:autocommit(Pid)),
  122. ok = mysql:query(Pid, <<"SET autocommit = 1">>),
  123. ?assert(mysql:autocommit(Pid)).
  124. basic_queries(Pid) ->
  125. %% warning count
  126. ?assertEqual(ok, mysql:query(Pid, <<"DROP TABLE IF EXISTS foo">>)),
  127. ?assertEqual(1, mysql:warning_count(Pid)),
  128. %% SQL parse error
  129. ?assertMatch({error, {1064, <<"42000">>, <<"You have an erro", _/binary>>}},
  130. mysql:query(Pid, <<"FOO">>)),
  131. %% Simple resultset with various types
  132. ?assertEqual({ok, [<<"i">>, <<"s">>], [[42, <<"foo">>]]},
  133. mysql:query(Pid, <<"SELECT 42 AS i, 'foo' AS s;">>)),
  134. ok.
  135. text_protocol(Pid) ->
  136. ok = mysql:query(Pid, ?create_table_t),
  137. ok = mysql:query(Pid, <<"INSERT INTO t (bl, f, d, dc, y, ti, ts, da, c)"
  138. " VALUES ('blob', 3.14, 3.14, 3.14, 2014,"
  139. "'00:22:11', '2014-11-03 00:22:24', '2014-11-03',"
  140. " NULL)">>),
  141. ?assertEqual(1, mysql:warning_count(Pid)), %% tx has no default value
  142. ?assertEqual(1, mysql:insert_id(Pid)), %% auto_increment starts from 1
  143. ?assertEqual(1, mysql:affected_rows(Pid)),
  144. %% select
  145. {ok, Columns, Rows} = mysql:query(Pid, <<"SELECT * FROM t">>),
  146. ?assertEqual([<<"id">>, <<"bl">>, <<"tx">>, <<"f">>, <<"d">>, <<"dc">>,
  147. <<"y">>, <<"ti">>, <<"ts">>, <<"da">>, <<"c">>], Columns),
  148. ?assertEqual([[1, <<"blob">>, <<>>, 3.14, 3.14, 3.14,
  149. 2014, {0, {0, 22, 11}},
  150. {{2014, 11, 03}, {00, 22, 24}}, {2014, 11, 03}, null]],
  151. Rows),
  152. ok = mysql:query(Pid, <<"DROP TABLE t">>).
  153. binary_protocol(Pid) ->
  154. ok = mysql:query(Pid, ?create_table_t),
  155. %% The same queries as in the text protocol. Expect the same results.
  156. {ok, Ins} = mysql:prepare(Pid, <<"INSERT INTO t (bl, f, d, dc, y, ti,"
  157. " ts, da, c)"
  158. " VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)">>),
  159. ok = mysql:execute(Pid, Ins, [<<"blob">>, 3.14, 3.14, 3.14,
  160. 2014, {0, {0, 22, 11}},
  161. {{2014, 11, 03}, {0, 22, 24}},
  162. {2014, 11, 03}, null]),
  163. {ok, Stmt} = mysql:prepare(Pid, <<"SELECT * FROM t WHERE id=?">>),
  164. {ok, Columns, Rows} = mysql:execute(Pid, Stmt, [1]),
  165. ?assertEqual([<<"id">>, <<"bl">>, <<"tx">>, <<"f">>, <<"d">>, <<"dc">>,
  166. <<"y">>, <<"ti">>,
  167. <<"ts">>, <<"da">>, <<"c">>], Columns),
  168. ?assertEqual([[1, <<"blob">>, <<>>, 3.14, 3.14, 3.14,
  169. 2014, {0, {0, 22, 11}},
  170. {{2014, 11, 03}, {00, 22, 24}}, {2014, 11, 03}, null]],
  171. Rows),
  172. ok = mysql:query(Pid, <<"DROP TABLE t">>).
  173. float_rounding(Pid) ->
  174. %% This is to make sure we get the same values for 32-bit FLOATs in the text
  175. %% and binary protocols for ordinary queries and prepared statements
  176. %% respectively.
  177. %%
  178. %% MySQL rounds to 6 significant digits when "printing" floats over the
  179. %% text protocol. When we receive a float on the binary protocol, we round
  180. %% it in the same way to match what MySQL does on the text protocol. This
  181. %% way we should to get the same values regardless of which protocol is
  182. %% used.
  183. %% Table for testing floats
  184. ok = mysql:query(Pid, "CREATE TABLE f (f FLOAT)"),
  185. %% Prepared statements
  186. {ok, Insert} = mysql:prepare(Pid, "INSERT INTO f (f) VALUES (?)"),
  187. {ok, Select} = mysql:prepare(Pid, "SELECT f FROM f"),
  188. %% [{Input, Expected}]
  189. TestData = [{1.0, 1.0}, {3.14, 3.14}, {0.2, 0.2},
  190. {0.20082111, 0.200821}, {0.000123456789, 0.000123457},
  191. {33.3333333, 33.3333}, {-33.2233443322, -33.2233},
  192. {400.0123, 400.012}, {1000.1234, 1000.12},
  193. {999.00009, 999.0},
  194. {1234.5678, 1234.57}, {68888.8888, 68888.9},
  195. {123456.789, 123457.0}, {7654321.0, 7654320.0},
  196. {80001111.1, 80001100.0}, {987654321.0, 987654000.0},
  197. {-123456789.0, -123457000.0},
  198. {2.12345111e-23, 2.12345e-23}, {-2.12345111e-23, -2.12345e-23},
  199. {2.12345111e23, 2.12345e23}, {-2.12345111e23, -2.12345e23}],
  200. lists:foreach(fun ({Input, Expected}) ->
  201. %% Insert using binary protocol (sending it as a double)
  202. ok = mysql:execute(Pid, Insert, [Input]),
  203. %% Text (plain query)
  204. {ok, _, [[Value]]} = mysql:query(Pid, "SELECT f FROM f"),
  205. ?assertEqual(Expected, Value),
  206. %% Binary (prepared statement)
  207. {ok, _, [[BinValue]]} = mysql:execute(Pid, Select, []),
  208. ?assertEqual(Expected, BinValue),
  209. %% cleanup before the next test
  210. ok = mysql:query(Pid, "DELETE FROM f")
  211. end,
  212. TestData),
  213. ok = mysql:query(Pid, "DROP TABLE f").
  214. decimal(Pid) ->
  215. %% As integer when S == 0
  216. ok = mysql:query(Pid, "CREATE TABLE dec0 (d DECIMAL(50, 0))"),
  217. write_read_text_binary(
  218. Pid, 14159265358979323846264338327950288419716939937510,
  219. <<"14159265358979323846264338327950288419716939937510">>,
  220. <<"dec0">>, <<"d">>
  221. ),
  222. write_read_text_binary(
  223. Pid, -14159265358979323846264338327950288419716939937510,
  224. <<"-14159265358979323846264338327950288419716939937510">>,
  225. <<"dec0">>, <<"d">>
  226. ),
  227. ok = mysql:query(Pid, "DROP TABLE dec0"),
  228. %% As float when P =< 15, S > 0
  229. ok = mysql:query(Pid, "CREATE TABLE dec15 (d DECIMAL(15, 14))"),
  230. write_read_text_binary(Pid, 3.14159265358979, <<"3.14159265358979">>,
  231. <<"dec15">>, <<"d">>),
  232. write_read_text_binary(Pid, -3.14159265358979, <<"-3.14159265358979">>,
  233. <<"dec15">>, <<"d">>),
  234. write_read_text_binary(Pid, 3.0, <<"3">>, <<"dec15">>, <<"d">>),
  235. ok = mysql:query(Pid, "DROP TABLE dec15"),
  236. %% As binary when P >= 16, S > 0
  237. ok = mysql:query(Pid, "CREATE TABLE dec16 (d DECIMAL(16, 15))"),
  238. write_read_text_binary(Pid, <<"3.141592653589793">>,
  239. <<"3.141592653589793">>, <<"dec16">>, <<"d">>),
  240. write_read_text_binary(Pid, <<"-3.141592653589793">>,
  241. <<"-3.141592653589793">>, <<"dec16">>, <<"d">>),
  242. write_read_text_binary(Pid, <<"3.000000000000000">>, <<"3">>,
  243. <<"dec16">>, <<"d">>),
  244. ok = mysql:query(Pid, "DROP TABLE dec16").
  245. int(Pid) ->
  246. ok = mysql:query(Pid, "CREATE TABLE ints (i INT)"),
  247. write_read_text_binary(Pid, 42, <<"42">>, <<"ints">>, <<"i">>),
  248. write_read_text_binary(Pid, -42, <<"-42">>, <<"ints">>, <<"i">>),
  249. write_read_text_binary(Pid, 987654321, <<"987654321">>, <<"ints">>,
  250. <<"i">>),
  251. write_read_text_binary(Pid, -987654321, <<"-987654321">>,
  252. <<"ints">>, <<"i">>),
  253. ok = mysql:query(Pid, "DROP TABLE ints"),
  254. %% Overflow with TINYINT
  255. ok = mysql:query(Pid, "CREATE TABLE tint (i TINYINT)"),
  256. write_read_text_binary(Pid, 127, <<"1000">>, <<"tint">>, <<"i">>),
  257. write_read_text_binary(Pid, -128, <<"-1000">>, <<"tint">>, <<"i">>),
  258. ok = mysql:query(Pid, "DROP TABLE tint"),
  259. %% SMALLINT
  260. ok = mysql:query(Pid, "CREATE TABLE sint (i SMALLINT)"),
  261. write_read_text_binary(Pid, 32000, <<"32000">>, <<"sint">>, <<"i">>),
  262. write_read_text_binary(Pid, -32000, <<"-32000">>, <<"sint">>, <<"i">>),
  263. ok = mysql:query(Pid, "DROP TABLE sint"),
  264. %% BIGINT
  265. ok = mysql:query(Pid, "CREATE TABLE bint (i BIGINT)"),
  266. write_read_text_binary(Pid, 123456789012, <<"123456789012">>,
  267. <<"bint">>, <<"i">>),
  268. write_read_text_binary(Pid, -123456789012, <<"-123456789012">>,
  269. <<"bint">>, <<"i">>),
  270. ok = mysql:query(Pid, "DROP TABLE bint").
  271. %% The BIT(N) datatype in MySQL 5.0.3 and later: the equivallent to bitstring()
  272. bit(Pid) ->
  273. ok = mysql:query(Pid, "CREATE TABLE bits (b BIT(11))"),
  274. write_read_text_binary(Pid, <<16#ff, 0:3>>, <<"b'11111111000'">>,
  275. <<"bits">>, <<"b">>),
  276. write_read_text_binary(Pid, <<16#7f, 6:3>>, <<"b'01111111110'">>,
  277. <<"bits">>, <<"b">>),
  278. ok = mysql:query(Pid, "DROP TABLE bits").
  279. date(Pid) ->
  280. ok = mysql:query(Pid, "CREATE TABLE d (d DATE)"),
  281. lists:foreach(
  282. fun ({Value, SqlLiteral}) ->
  283. write_read_text_binary(Pid, Value, SqlLiteral, <<"d">>, <<"d">>)
  284. end,
  285. [{{2014, 11, 03}, <<"'2014-11-03'">>},
  286. {{0, 0, 0}, <<"'0000-00-00'">>}]
  287. ),
  288. ok = mysql:query(Pid, "DROP TABLE d").
  289. %% Test TIME value representation. There are a few things to check.
  290. time(Pid) ->
  291. ok = mysql:query(Pid, "CREATE TABLE tm (tm TIME)"),
  292. lists:foreach(
  293. fun ({Value, SqlLiteral}) ->
  294. write_read_text_binary(Pid, Value, SqlLiteral, <<"tm">>, <<"tm">>)
  295. end,
  296. [{{0, {10, 11, 12}}, <<"'10:11:12'">>},
  297. {{5, {0, 0, 1}}, <<"'120:00:01'">>},
  298. {{-1, {23, 59, 59}}, <<"'-00:00:01'">>},
  299. {{-1, {23, 59, 0}}, <<"'-00:01:00'">>},
  300. {{-1, {23, 0, 0}}, <<"'-01:00:00'">>},
  301. {{-1, {0, 0, 0}}, <<"'-24:00:00'">>},
  302. {{-5, {10, 0, 0}}, <<"'-110:00:00'">>},
  303. {{0, {0, 0, 0}}, <<"'00:00:00'">>}]
  304. ),
  305. %% Zero seconds as a float.
  306. ok = mysql:query(Pid, "INSERT INTO tm (tm) VALUES (?)",
  307. [{-1, {1, 2, 0.0}}]),
  308. ?assertEqual({ok, [<<"tm">>], [[{-1, {1, 2, 0}}]]},
  309. mysql:query(Pid, "SELECT tm FROM tm")),
  310. ok = mysql:query(Pid, "DROP TABLE tm").
  311. datetime(Pid) ->
  312. ok = mysql:query(Pid, "CREATE TABLE dt (dt DATETIME)"),
  313. lists:foreach(
  314. fun ({Value, SqlLiteral}) ->
  315. write_read_text_binary(Pid, Value, SqlLiteral, <<"dt">>, <<"dt">>)
  316. end,
  317. [{{{2014, 12, 14}, {19, 39, 20}}, <<"'2014-12-14 19:39:20'">>},
  318. {{{2014, 12, 14}, {0, 0, 0}}, <<"'2014-12-14 00:00:00'">>},
  319. {{{0, 0, 0}, {0, 0, 0}}, <<"'0000-00-00 00:00:00'">>}]
  320. ),
  321. ok = mysql:query(Pid, "DROP TABLE dt").
  322. microseconds(Pid) ->
  323. %% Check whether we have the required version for this testcase.
  324. {ok, _, [[Version]]} = mysql:query(Pid, <<"SELECT @@version">>),
  325. try
  326. %% Remove stuff after dash for e.g. "5.5.40-0ubuntu0.12.04.1-log"
  327. [Version1 | _] = binary:split(Version, <<"-">>),
  328. Version2 = lists:map(fun binary_to_integer/1,
  329. binary:split(Version1, <<".">>, [global])),
  330. Version2 >= [5, 6, 4] orelse throw(nope)
  331. of _ ->
  332. test_time_microseconds(Pid),
  333. test_datetime_microseconds(Pid)
  334. catch _:_ ->
  335. error_logger:info_msg("Skipping microseconds test. Current MySQL"
  336. " version is ~s. Required version is >= 5.6.4.~n",
  337. [Version])
  338. end.
  339. test_time_microseconds(Pid) ->
  340. ok = mysql:query(Pid, "CREATE TABLE m (t TIME(6))"),
  341. %% Positive time
  342. write_read_text_binary(Pid, {0, {23, 59, 57.654321}},
  343. <<"'23:59:57.654321'">>, <<"m">>, <<"t">>),
  344. %% Negative time
  345. write_read_text_binary(Pid, {-1, {23, 59, 57.654321}},
  346. <<"'-00:00:02.345679'">>, <<"m">>, <<"t">>),
  347. ok = mysql:query(Pid, "DROP TABLE m").
  348. test_datetime_microseconds(Pid) ->
  349. ok = mysql:query(Pid, "CREATE TABLE dt (dt DATETIME(6))"),
  350. write_read_text_binary(Pid, {{2014, 11, 23}, {23, 59, 57.654321}},
  351. <<"'2014-11-23 23:59:57.654321'">>, <<"dt">>,
  352. <<"dt">>),
  353. ok = mysql:query(Pid, "DROP TABLE dt").
  354. %% @doc Tests write and read in text and the binary protocol, all combinations.
  355. %% This helper function assumes an empty table with a single column.
  356. write_read_text_binary(Conn, Term, SqlLiteral, Table, Column) ->
  357. SelectQuery = <<"SELECT ", Column/binary, " FROM ", Table/binary>>,
  358. {ok, SelectStmt} = mysql:prepare(Conn, SelectQuery),
  359. %% Insert as text, read text and binary, delete
  360. InsertQuery = <<"INSERT INTO ", Table/binary, " (", Column/binary, ")"
  361. " VALUES (", SqlLiteral/binary, ")">>,
  362. ok = mysql:query(Conn, InsertQuery),
  363. ?assertEqual({ok, [Column], [[Term]]}, mysql:query(Conn, SelectQuery)),
  364. ?assertEqual({ok, [Column], [[Term]]}, mysql:execute(Conn, SelectStmt, [])),
  365. mysql:query(Conn, <<"DELETE FROM ", Table/binary>>),
  366. %% Insert as binary, read text and binary, delete
  367. InsertQ = <<"INSERT INTO ", Table/binary, " (", Column/binary, ")",
  368. " VALUES (?)">>,
  369. {ok, InsertStmt} = mysql:prepare(Conn, InsertQ),
  370. ok = mysql:execute(Conn, InsertStmt, [Term]),
  371. ok = mysql:unprepare(Conn, InsertStmt),
  372. ?assertEqual({ok, [Column], [[Term]]}, mysql:query(Conn, SelectQuery)),
  373. ?assertEqual({ok, [Column], [[Term]]}, mysql:execute(Conn, SelectStmt, [])),
  374. mysql:query(Conn, <<"DELETE FROM ", Table/binary>>),
  375. %% Cleanup
  376. ok = mysql:unprepare(Conn, SelectStmt).
  377. %% --------------------------------------------------------------------------
  378. timeout_test_() ->
  379. {setup,
  380. fun () ->
  381. {ok, Pid} = mysql:start_link([{user, ?user}, {password, ?password},
  382. {log_warnings, false}]),
  383. Pid
  384. end,
  385. fun (Pid) ->
  386. exit(Pid, normal)
  387. end,
  388. {with, [fun (Pid) ->
  389. %% SLEEP was added in MySQL 5.0.12
  390. ?assertEqual({ok, [<<"SLEEP(5)">>], [[1]]},
  391. mysql:query(Pid, <<"SELECT SLEEP(5)">>, 40)),
  392. %% A query after an interrupted query shouldn't get a timeout.
  393. ?assertMatch({ok,[<<"42">>], [[42]]},
  394. mysql:query(Pid, <<"SELECT 42">>)),
  395. %% Parametrized query
  396. ?assertEqual({ok, [<<"SLEEP(?)">>], [[1]]},
  397. mysql:query(Pid, <<"SELECT SLEEP(?)">>, [5], 40)),
  398. %% Prepared statement
  399. {ok, Stmt} = mysql:prepare(Pid, <<"SELECT SLEEP(?)">>),
  400. ?assertEqual({ok, [<<"SLEEP(?)">>], [[1]]},
  401. mysql:execute(Pid, Stmt, [5], 40)),
  402. ok = mysql:unprepare(Pid, Stmt)
  403. end]}}.
  404. %% --------------------------------------------------------------------------
  405. %% Prepared statements
  406. with_table_foo_test_() ->
  407. {setup,
  408. fun () ->
  409. {ok, Pid} = mysql:start_link([{user, ?user}, {password, ?password},
  410. {query_cache_time, 50},
  411. {log_warnings, false}]),
  412. ok = mysql:query(Pid, <<"DROP DATABASE IF EXISTS otptest">>),
  413. ok = mysql:query(Pid, <<"CREATE DATABASE otptest">>),
  414. ok = mysql:query(Pid, <<"USE otptest">>),
  415. ok = mysql:query(Pid, <<"CREATE TABLE foo (bar INT) engine=InnoDB">>),
  416. Pid
  417. end,
  418. fun (Pid) ->
  419. ok = mysql:query(Pid, <<"DROP DATABASE otptest">>),
  420. exit(Pid, normal)
  421. end,
  422. fun (Pid) ->
  423. [{"Prepared statements", fun () -> prepared_statements(Pid) end},
  424. {"Parametrized queries", fun () -> parameterized_query(Pid) end}]
  425. end}.
  426. prepared_statements(Pid) ->
  427. %% Unnamed
  428. ?assertEqual({error,{1146, <<"42S02">>,
  429. <<"Table 'otptest.tab' doesn't exist">>}},
  430. mysql:prepare(Pid, "SELECT * FROM tab WHERE id = ?")),
  431. {ok, StmtId} = mysql:prepare(Pid, "SELECT * FROM foo WHERE bar = ?"),
  432. ?assert(is_integer(StmtId)),
  433. ?assertEqual(ok, mysql:unprepare(Pid, StmtId)),
  434. ?assertEqual({error, not_prepared}, mysql:unprepare(Pid, StmtId)),
  435. %% Named
  436. ?assertEqual({error,{1146, <<"42S02">>,
  437. <<"Table 'otptest.tab' doesn't exist">>}},
  438. mysql:prepare(Pid, tab, "SELECT * FROM tab WHERE id = ?")),
  439. ?assertEqual({ok, foo},
  440. mysql:prepare(Pid, foo, "SELECT * FROM foo WHERE bar = ?")),
  441. %% Prepare again unprepares the old stmt associated with this name.
  442. ?assertEqual({ok, foo},
  443. mysql:prepare(Pid, foo, "SELECT bar FROM foo WHERE bar = ?")),
  444. ?assertEqual(ok, mysql:unprepare(Pid, foo)),
  445. ?assertEqual({error, not_prepared}, mysql:unprepare(Pid, foo)),
  446. %% Execute when not prepared
  447. ?assertEqual({error, not_prepared}, mysql:execute(Pid, not_a_stmt, [])),
  448. ok.
  449. parameterized_query(Conn) ->
  450. %% To see that cache eviction works as expected, look at the code coverage.
  451. {ok, _, []} = mysql:query(Conn, "SELECT * FROM foo WHERE bar = ?", [1]),
  452. {ok, _, []} = mysql:query(Conn, "SELECT * FROM foo WHERE bar = ?", [2]),
  453. receive after 150 -> ok end, %% Now the query cache should emptied
  454. {ok, _, []} = mysql:query(Conn, "SELECT * FROM foo WHERE bar = ?", [3]).
  455. %% --- simple gen_server callbacks ---
  456. gen_server_coverage_test() ->
  457. {noreply, state} = mysql:handle_cast(foo, state),
  458. {noreply, state} = mysql:handle_info(foo, state),
  459. ok = mysql:terminate(kill, state).