ConnectionTest.php 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804
  1. <?php
  2. /**
  3. * CakePHP(tm) : Rapid Development Framework (http://cakephp.org)
  4. * Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
  5. *
  6. * Licensed under The MIT License
  7. * For full copyright and license information, please see the LICENSE.txt
  8. * Redistributions of files must retain the above copyright notice.
  9. *
  10. * @copyright Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
  11. * @link http://cakephp.org CakePHP(tm) Project
  12. * @since 3.0.0
  13. * @license MIT License (http://www.opensource.org/licenses/mit-license.php)
  14. */
  15. namespace Cake\Test\TestCase\Database;
  16. use Cake\Core\Configure;
  17. use Cake\Database\Connection;
  18. use Cake\Datasource\ConnectionManager;
  19. use Cake\TestSuite\TestCase;
  20. /**
  21. * Tests Connection class
  22. */
  23. class ConnectionTest extends TestCase {
  24. public $fixtures = ['core.thing'];
  25. public function setUp() {
  26. parent::setUp();
  27. $this->connection = ConnectionManager::get('test');
  28. }
  29. public function tearDown() {
  30. $this->connection->useSavePoints(false);
  31. unset($this->connection);
  32. parent::tearDown();
  33. }
  34. /**
  35. * Auxiliary method to build a mock for a driver so it can be injected into
  36. * the connection object
  37. *
  38. * @return \Cake\Database\Driver
  39. */
  40. public function getMockFormDriver() {
  41. $driver = $this->getMock('Cake\Database\Driver');
  42. $driver->expects($this->once())
  43. ->method('enabled')
  44. ->will($this->returnValue(true));
  45. return $driver;
  46. }
  47. /**
  48. * Tests connecting to database
  49. *
  50. * @return void
  51. */
  52. public function testConnect() {
  53. $this->assertTrue($this->connection->connect());
  54. $this->assertTrue($this->connection->isConnected());
  55. }
  56. /**
  57. * Tests creating a connection using no driver throws an exception
  58. *
  59. * @expectedException \Cake\Database\Exception\MissingDriverException
  60. * @expectedExceptionMessage Database driver "" could not be found.
  61. * @return void
  62. */
  63. public function testNoDriver() {
  64. $connection = new Connection([]);
  65. }
  66. /**
  67. * Tests creating a connection using an invalid driver throws an exception
  68. *
  69. * @expectedException \Cake\Database\Exception\MissingDriverException
  70. * @expectedExceptionMessage Database driver "" could not be found.
  71. * @return void
  72. */
  73. public function testEmptyDriver() {
  74. $connection = new Connection(['driver' => false]);
  75. }
  76. /**
  77. * Tests creating a connection using an invalid driver throws an exception
  78. *
  79. * @expectedException \Cake\Database\Exception\MissingDriverException
  80. * @expectedExceptionMessage Database driver "\Foo\InvalidDriver" could not be found.
  81. * @return void
  82. */
  83. public function testMissingDriver() {
  84. $connection = new Connection(['driver' => '\Foo\InvalidDriver']);
  85. }
  86. /**
  87. * Tests trying to use a disabled driver throws an exception
  88. *
  89. * @expectedException \Cake\Database\Exception\MissingExtensionException
  90. * @expectedExceptionMessage Database driver DriverMock cannot be used due to a missing PHP extension or unmet dependency
  91. * @return void
  92. */
  93. public function testDisabledDriver() {
  94. $mock = $this->getMock('\Cake\Database\Connection\Driver', ['enabled'], [], 'DriverMock');
  95. $connection = new Connection(['driver' => $mock]);
  96. }
  97. /**
  98. * Tests that connecting with invalid credentials or database name throws an exception
  99. *
  100. * @expectedException \Cake\Database\Exception\MissingConnectionException
  101. * @return void
  102. **/
  103. public function testWrongCredentials() {
  104. $config = ConnectionManager::config('test');
  105. $this->skipIf(isset($config['dsn']), 'Datasource has dsn, skipping.');
  106. $connection = new Connection(['database' => '_probably_not_there_'] + ConnectionManager::config('test'));
  107. $connection->connect();
  108. }
  109. /**
  110. * Tests creation of prepared statements
  111. *
  112. * @return void
  113. **/
  114. public function testPrepare() {
  115. $sql = 'SELECT 1 + 1';
  116. $result = $this->connection->prepare($sql);
  117. $this->assertInstanceOf('Cake\Database\StatementInterface', $result);
  118. $this->assertEquals($sql, $result->queryString);
  119. $query = $this->connection->newQuery()->select('1 + 1');
  120. $result = $this->connection->prepare($query);
  121. $this->assertInstanceOf('Cake\Database\StatementInterface', $result);
  122. $sql = '#SELECT [`"\[]?1 \+ 1[`"\]]?#';
  123. $this->assertRegExp($sql, $result->queryString);
  124. }
  125. /**
  126. * Tests executing a simple query using bound values
  127. *
  128. * @return void
  129. **/
  130. public function testExecuteWithArguments() {
  131. $sql = 'SELECT 1 + ?';
  132. $statement = $this->connection->execute($sql, [1], array('integer'));
  133. $this->assertCount(1, $statement);
  134. $result = $statement->fetch();
  135. $this->assertEquals([2], $result);
  136. $statement->closeCursor();
  137. $sql = 'SELECT 1 + ? + ? AS total';
  138. $statement = $this->connection->execute($sql, [2, 3], array('integer', 'integer'));
  139. $this->assertCount(1, $statement);
  140. $result = $statement->fetch('assoc');
  141. $this->assertEquals(['total' => 6], $result);
  142. $statement->closeCursor();
  143. $sql = 'SELECT 1 + :one + :two AS total';
  144. $statement = $this->connection->execute($sql, ['one' => 2, 'two' => 3], array('one' => 'integer', 'two' => 'integer'));
  145. $this->assertCount(1, $statement);
  146. $result = $statement->fetch('assoc');
  147. $statement->closeCursor();
  148. $this->assertEquals(['total' => 6], $result);
  149. }
  150. /**
  151. * Tests executing a query with params and associated types
  152. *
  153. * @return void
  154. **/
  155. public function testExecuteWithArgumentsAndTypes() {
  156. $sql = "SELECT ? = '2012-01-01'";
  157. $statement = $this->connection->execute($sql, [new \DateTime('2012-01-01')], ['date']);
  158. $result = $statement->fetch();
  159. $statement->closeCursor();
  160. $this->assertTrue((bool)$result[0]);
  161. $sql = "SELECT ? = '2012-01-01', ? = '2000-01-01 10:10:10', ? = 2";
  162. $params = [new \DateTime('2012-01-01 10:10:10'), '2000-01-01 10:10:10', 2.1];
  163. $statement = $this->connection->execute($sql, $params, ['date', 'string', 'integer']);
  164. $result = $statement->fetch();
  165. $statement->closeCursor();
  166. $this->assertEquals($result, array_filter($result));
  167. }
  168. /**
  169. * Tests that passing a unknown value to a query throws an exception
  170. *
  171. * @expectedException \InvalidArgumentException
  172. * @return void
  173. **/
  174. public function testExecuteWithMissingType() {
  175. $sql = 'SELECT ?';
  176. $statement = $this->connection->execute($sql, [new \DateTime('2012-01-01')], ['bar']);
  177. }
  178. /**
  179. * Tests executing a query with no params also works
  180. *
  181. * @return void
  182. **/
  183. public function testExecuteWithNoParams() {
  184. $sql = 'SELECT 1';
  185. $statement = $this->connection->execute($sql);
  186. $result = $statement->fetch();
  187. $this->assertCount(1, $result);
  188. $this->assertEquals([1], $result);
  189. $statement->closeCursor();
  190. }
  191. /**
  192. * Tests it is possible to insert data into a table using matching types by key name
  193. *
  194. * @return void
  195. **/
  196. public function testInsertWithMatchingTypes() {
  197. $data = ['id' => '3', 'title' => 'a title', 'body' => 'a body'];
  198. $result = $this->connection->insert(
  199. 'things',
  200. $data,
  201. ['id' => 'integer', 'title' => 'string', 'body' => 'string']
  202. );
  203. $this->assertInstanceOf('Cake\Database\StatementInterface', $result);
  204. $result->closeCursor();
  205. $result = $this->connection->execute('SELECT * from things where id = 3');
  206. $this->assertCount(1, $result);
  207. $row = $result->fetch('assoc');
  208. $result->closeCursor();
  209. $this->assertEquals($data, $row);
  210. }
  211. /**
  212. * Tests it is possible to insert data into a table using matching types by array position
  213. *
  214. * @return void
  215. **/
  216. public function _testInsertWithPositionalTypes() {
  217. $data = ['id' => '3', 'title' => 'a title', 'body' => 'a body'];
  218. $result = $this->connection->insert(
  219. 'things',
  220. $data,
  221. ['integer', 'string', 'string']
  222. );
  223. $result->closeCursor();
  224. $this->assertInstanceOf('Cake\Database\StatementInterface', $result);
  225. $result = $this->connection->execute('SELECT * from things where id = 3');
  226. $this->assertCount(1, $result);
  227. $row = $result->fetch('assoc');
  228. $result->closeCursor();
  229. $this->assertEquals($data, $row);
  230. }
  231. /**
  232. * Tests an statement class can be reused for multiple executions
  233. *
  234. * @return void
  235. **/
  236. public function testStatementReusing() {
  237. $total = $this->connection->execute('SELECT COUNT(*) AS total FROM things');
  238. $result = $total->fetch('assoc');
  239. $this->assertEquals(2, $result['total']);
  240. $total->closeCursor();
  241. $result = $this->connection->execute('SELECT title, body FROM things');
  242. $row = $result->fetch('assoc');
  243. $this->assertEquals('a title', $row['title']);
  244. $this->assertEquals('a body', $row['body']);
  245. $row = $result->fetch('assoc');
  246. $result->closeCursor();
  247. $this->assertEquals('another title', $row['title']);
  248. $this->assertEquals('another body', $row['body']);
  249. }
  250. /**
  251. * Tests rows can be updated without specifying any conditions nor types
  252. *
  253. * @return void
  254. **/
  255. public function testUpdateWithoutConditionsNorTypes() {
  256. $title = 'changed the title!';
  257. $body = 'changed the body!';
  258. $this->connection->update('things', ['title' => $title, 'body' => $body]);
  259. $result = $this->connection->execute('SELECT * FROM things WHERE title = ? AND body = ?', [$title, $body]);
  260. $this->assertCount(2, $result);
  261. $result->closeCursor();
  262. }
  263. /**
  264. * Tests it is possible to use key => value conditions for update
  265. *
  266. * @return void
  267. **/
  268. public function testUpdateWithConditionsNoTypes() {
  269. $title = 'changed the title!';
  270. $body = 'changed the body!';
  271. $this->connection->update('things', ['title' => $title, 'body' => $body], ['id' => 2]);
  272. $result = $this->connection->execute('SELECT * FROM things WHERE title = ? AND body = ?', [$title, $body]);
  273. $this->assertCount(1, $result);
  274. $result->closeCursor();
  275. }
  276. /**
  277. * Tests it is possible to use key => value and string conditions for update
  278. *
  279. * @return void
  280. **/
  281. public function testUpdateWithConditionsCombinedNoTypes() {
  282. $title = 'changed the title!';
  283. $body = 'changed the body!';
  284. $this->connection->update('things', ['title' => $title, 'body' => $body], ['id' => 2, 'body is not null']);
  285. $result = $this->connection->execute('SELECT * FROM things WHERE title = ? AND body = ?', [$title, $body]);
  286. $this->assertCount(1, $result);
  287. $result->closeCursor();
  288. }
  289. /**
  290. * Tests you can bind types to update values
  291. *
  292. * @return void
  293. **/
  294. public function testUpdateWithTypes() {
  295. $title = 'changed the title!';
  296. $body = new \DateTime('2012-01-01');
  297. $values = compact('title', 'body');
  298. $this->connection->update('things', $values, [], ['body' => 'date']);
  299. $result = $this->connection->execute('SELECT * FROM things WHERE title = :title AND body = :body', $values, ['body' => 'date']);
  300. $this->assertCount(2, $result);
  301. $row = $result->fetch('assoc');
  302. $this->assertEquals('2012-01-01', $row['body']);
  303. $row = $result->fetch('assoc');
  304. $this->assertEquals('2012-01-01', $row['body']);
  305. $result->closeCursor();
  306. }
  307. /**
  308. * Tests you can bind types to update values
  309. *
  310. * @return void
  311. **/
  312. public function testUpdateWithConditionsAndTypes() {
  313. $title = 'changed the title!';
  314. $body = new \DateTime('2012-01-01');
  315. $values = compact('title', 'body');
  316. $this->connection->update('things', $values, ['id' => '1-string-parsed-as-int'], ['body' => 'date', 'id' => 'integer']);
  317. $result = $this->connection->execute('SELECT * FROM things WHERE title = :title AND body = :body', $values, ['body' => 'date']);
  318. $this->assertCount(1, $result);
  319. $row = $result->fetch('assoc');
  320. $this->assertEquals('2012-01-01', $row['body']);
  321. $result->closeCursor();
  322. }
  323. /**
  324. * Tests delete from table with no conditions
  325. *
  326. * @return void
  327. **/
  328. public function testDeleteNoConditions() {
  329. $this->connection->delete('things');
  330. $result = $this->connection->execute('SELECT * FROM things');
  331. $this->assertCount(0, $result);
  332. }
  333. /**
  334. * Tests delete from table with conditions
  335. * @return void
  336. **/
  337. public function testDeleteWithConditions() {
  338. $this->connection->delete('things', ['id' => '1-rest-is-ommited'], ['id' => 'integer']);
  339. $result = $this->connection->execute('SELECT * FROM things');
  340. $this->assertCount(1, $result);
  341. $this->connection->delete('things', ['id' => '1-rest-is-ommited'], ['id' => 'integer']);
  342. $result = $this->connection->execute('SELECT * FROM things');
  343. $this->assertCount(1, $result);
  344. $this->connection->delete('things', ['id' => '2-rest-is-ommited'], ['id' => 'integer']);
  345. $result = $this->connection->execute('SELECT * FROM things');
  346. $this->assertCount(0, $result);
  347. }
  348. /**
  349. * Tests that it is possible to use simple database transactions
  350. *
  351. * @return void
  352. **/
  353. public function testSimpleTransactions() {
  354. $this->connection->begin();
  355. $this->connection->delete('things', ['id' => 1]);
  356. $this->connection->rollback();
  357. $result = $this->connection->execute('SELECT * FROM things');
  358. $this->assertCount(2, $result);
  359. $this->connection->begin();
  360. $this->connection->delete('things', ['id' => 1]);
  361. $this->connection->commit();
  362. $result = $this->connection->execute('SELECT * FROM things');
  363. $this->assertCount(1, $result);
  364. }
  365. /**
  366. * Tests that it is possible to use virtualized nested transaction
  367. * with early rollback algorithm
  368. *
  369. * @return void
  370. **/
  371. public function testVirtualNestedTrasanction() {
  372. //starting 3 virtual transaction
  373. $this->connection->begin();
  374. $this->connection->begin();
  375. $this->connection->begin();
  376. $this->connection->delete('things', ['id' => 1]);
  377. $result = $this->connection->execute('SELECT * FROM things');
  378. $this->assertCount(1, $result);
  379. $this->connection->commit();
  380. $this->connection->rollback();
  381. $result = $this->connection->execute('SELECT * FROM things');
  382. $this->assertCount(2, $result);
  383. }
  384. /**
  385. * Tests that it is possible to use virtualized nested transaction
  386. * with early rollback algorithm
  387. *
  388. * @return void
  389. **/
  390. public function testVirtualNestedTrasanction2() {
  391. //starting 3 virtual transaction
  392. $this->connection->begin();
  393. $this->connection->begin();
  394. $this->connection->begin();
  395. $this->connection->delete('things', ['id' => 1]);
  396. $result = $this->connection->execute('SELECT * FROM things');
  397. $this->assertCount(1, $result);
  398. $this->connection->rollback();
  399. $result = $this->connection->execute('SELECT * FROM things');
  400. $this->assertCount(2, $result);
  401. }
  402. /**
  403. * Tests that it is possible to use virtualized nested transaction
  404. * with early rollback algorithm
  405. *
  406. * @return void
  407. **/
  408. public function testVirtualNestedTrasanction3() {
  409. //starting 3 virtual transaction
  410. $this->connection->begin();
  411. $this->connection->begin();
  412. $this->connection->begin();
  413. $this->connection->delete('things', ['id' => 1]);
  414. $result = $this->connection->execute('SELECT * FROM things');
  415. $this->assertCount(1, $result);
  416. $this->connection->commit();
  417. $this->connection->commit();
  418. $this->connection->commit();
  419. $result = $this->connection->execute('SELECT * FROM things');
  420. $this->assertCount(1, $result);
  421. }
  422. /**
  423. * Tests that it is possible to real use nested transactions
  424. *
  425. * @return void
  426. **/
  427. public function testSavePoints() {
  428. $this->skipIf(!$this->connection->useSavePoints(true));
  429. $this->connection->begin();
  430. $this->connection->delete('things', ['id' => 1]);
  431. $result = $this->connection->execute('SELECT * FROM things');
  432. $this->assertCount(1, $result);
  433. $this->connection->begin();
  434. $this->connection->delete('things', ['id' => 2]);
  435. $result = $this->connection->execute('SELECT * FROM things');
  436. $this->assertCount(0, $result);
  437. $this->connection->rollback();
  438. $result = $this->connection->execute('SELECT * FROM things');
  439. $this->assertCount(1, $result);
  440. $this->connection->rollback();
  441. $result = $this->connection->execute('SELECT * FROM things');
  442. $this->assertCount(2, $result);
  443. }
  444. /**
  445. * Tests that it is possible to real use nested transactions
  446. *
  447. * @return void
  448. **/
  449. public function testSavePoints2() {
  450. $this->skipIf(!$this->connection->useSavePoints(true));
  451. $this->connection->begin();
  452. $this->connection->delete('things', ['id' => 1]);
  453. $result = $this->connection->execute('SELECT * FROM things');
  454. $this->assertCount(1, $result);
  455. $this->connection->begin();
  456. $this->connection->delete('things', ['id' => 2]);
  457. $result = $this->connection->execute('SELECT * FROM things');
  458. $this->assertCount(0, $result);
  459. $this->connection->rollback();
  460. $result = $this->connection->execute('SELECT * FROM things');
  461. $this->assertCount(1, $result);
  462. $this->connection->commit();
  463. $result = $this->connection->execute('SELECT * FROM things');
  464. $this->assertCount(1, $result);
  465. }
  466. /**
  467. * Tests connection can quote values to be safely used in query strings
  468. *
  469. * @return void
  470. **/
  471. public function testQuote() {
  472. $this->skipIf(!$this->connection->supportsQuoting());
  473. $expected = "'2012-01-01'";
  474. $result = $this->connection->quote(new \DateTime('2012-01-01'), 'date');
  475. $this->assertEquals($expected, $result);
  476. $expected = "'1'";
  477. $result = $this->connection->quote(1, 'string');
  478. $this->assertEquals($expected, $result);
  479. $expected = "'hello'";
  480. $result = $this->connection->quote('hello', 'string');
  481. $this->assertEquals($expected, $result);
  482. }
  483. /**
  484. * Tests identifier quoting
  485. *
  486. * @return void
  487. */
  488. public function testQuoteIdentifier() {
  489. $driver = $this->getMock('Cake\Database\Driver\Sqlite', ['enabled']);
  490. $driver->expects($this->once())
  491. ->method('enabled')
  492. ->will($this->returnValue(true));
  493. $connection = new Connection(['driver' => $driver]);
  494. $result = $connection->quoteIdentifier('name');
  495. $expected = '"name"';
  496. $this->assertEquals($expected, $result);
  497. $result = $connection->quoteIdentifier('Model.*');
  498. $expected = '"Model".*';
  499. $this->assertEquals($expected, $result);
  500. $result = $connection->quoteIdentifier('MTD()');
  501. $expected = 'MTD()';
  502. $this->assertEquals($expected, $result);
  503. $result = $connection->quoteIdentifier('(sm)');
  504. $expected = '(sm)';
  505. $this->assertEquals($expected, $result);
  506. $result = $connection->quoteIdentifier('name AS x');
  507. $expected = '"name" AS "x"';
  508. $this->assertEquals($expected, $result);
  509. $result = $connection->quoteIdentifier('Model.name AS x');
  510. $expected = '"Model"."name" AS "x"';
  511. $this->assertEquals($expected, $result);
  512. $result = $connection->quoteIdentifier('Function(Something.foo)');
  513. $expected = 'Function("Something"."foo")';
  514. $this->assertEquals($expected, $result);
  515. $result = $connection->quoteIdentifier('Function(SubFunction(Something.foo))');
  516. $expected = 'Function(SubFunction("Something"."foo"))';
  517. $this->assertEquals($expected, $result);
  518. $result = $connection->quoteIdentifier('Function(Something.foo) AS x');
  519. $expected = 'Function("Something"."foo") AS "x"';
  520. $this->assertEquals($expected, $result);
  521. $result = $connection->quoteIdentifier('name-with-minus');
  522. $expected = '"name-with-minus"';
  523. $this->assertEquals($expected, $result);
  524. $result = $connection->quoteIdentifier('my-name');
  525. $expected = '"my-name"';
  526. $this->assertEquals($expected, $result);
  527. $result = $connection->quoteIdentifier('Foo-Model.*');
  528. $expected = '"Foo-Model".*';
  529. $this->assertEquals($expected, $result);
  530. $result = $connection->quoteIdentifier('Team.P%');
  531. $expected = '"Team"."P%"';
  532. $this->assertEquals($expected, $result);
  533. $result = $connection->quoteIdentifier('Team.G/G');
  534. $expected = '"Team"."G/G"';
  535. $result = $connection->quoteIdentifier('Model.name as y');
  536. $expected = '"Model"."name" AS "y"';
  537. $this->assertEquals($expected, $result);
  538. }
  539. /**
  540. * Tests default return vale for logger() function
  541. *
  542. * @return void
  543. */
  544. public function testLoggerDefault() {
  545. $logger = $this->connection->logger();
  546. $this->assertInstanceOf('\Cake\Database\Log\QueryLogger', $logger);
  547. $this->assertSame($logger, $this->connection->logger());
  548. }
  549. /**
  550. * Tests that a custom logger object can be set
  551. *
  552. * @return void
  553. */
  554. public function testSetLogger() {
  555. $logger = new \Cake\Database\Log\QueryLogger;
  556. $this->connection->logger($logger);
  557. $this->assertSame($logger, $this->connection->logger());
  558. }
  559. /**
  560. * Tests that statements are decorated with a logger when logQueries is set to true
  561. *
  562. * @return void
  563. */
  564. public function testLoggerDecorator() {
  565. $logger = new \Cake\Database\Log\QueryLogger;
  566. $this->connection->logQueries(true);
  567. $this->connection->logger($logger);
  568. $st = $this->connection->prepare('SELECT 1');
  569. $this->assertInstanceOf('\Cake\Database\Log\LoggingStatement', $st);
  570. $this->assertSame($logger, $st->logger());
  571. $this->connection->logQueries(false);
  572. $st = $this->connection->prepare('SELECT 1');
  573. $this->assertNotInstanceOf('\Cake\Database\Log\LoggingStatement', $st);
  574. }
  575. /**
  576. * Tests that log() function logs to the configured query logger
  577. *
  578. * @return void
  579. */
  580. public function testLogFunction() {
  581. $logger = $this->getMock('\Cake\Database\Log\QueryLogger');
  582. $this->connection->logger($logger);
  583. $logger->expects($this->once())->method('log')
  584. ->with($this->logicalAnd(
  585. $this->isInstanceOf('\Cake\Database\Log\LoggedQuery'),
  586. $this->attributeEqualTo('query', 'SELECT 1')
  587. ));
  588. $this->connection->log('SELECT 1');
  589. }
  590. /**
  591. * Tests that begin and rollback are also logged
  592. *
  593. * @return void
  594. */
  595. public function testLogBeginRollbackTransaction() {
  596. $connection = $this
  597. ->getMockBuilder('\Cake\Database\Connection')
  598. ->setMethods(['connect'])
  599. ->disableOriginalConstructor()
  600. ->getMock();
  601. $connection->logQueries(true);
  602. $driver = $this->getMockFormDriver();
  603. $connection->driver($driver);
  604. $logger = $this->getMock('\Cake\Database\Log\QueryLogger');
  605. $connection->logger($logger);
  606. $logger->expects($this->at(0))->method('log')
  607. ->with($this->logicalAnd(
  608. $this->isInstanceOf('\Cake\Database\Log\LoggedQuery'),
  609. $this->attributeEqualTo('query', 'BEGIN')
  610. ));
  611. $logger->expects($this->at(1))->method('log')
  612. ->with($this->logicalAnd(
  613. $this->isInstanceOf('\Cake\Database\Log\LoggedQuery'),
  614. $this->attributeEqualTo('query', 'ROLLBACK')
  615. ));
  616. $connection->begin();
  617. $connection->begin(); //This one will not be logged
  618. $connection->rollback();
  619. }
  620. /**
  621. * Tests that commits are logged
  622. *
  623. * @return void
  624. */
  625. public function testLogCommitTransaction() {
  626. $driver = $this->getMockFormDriver();
  627. $connection = $this->getMock(
  628. '\Cake\Database\Connection',
  629. ['connect'],
  630. [['driver' => $driver]]
  631. );
  632. $logger = $this->getMock('\Cake\Database\Log\QueryLogger');
  633. $connection->logger($logger);
  634. $logger->expects($this->at(1))->method('log')
  635. ->with($this->logicalAnd(
  636. $this->isInstanceOf('\Cake\Database\Log\LoggedQuery'),
  637. $this->attributeEqualTo('query', 'COMMIT')
  638. ));
  639. $connection->logQueries(true);
  640. $connection->begin();
  641. $connection->commit();
  642. }
  643. /**
  644. * Tests that the transactional method will start and commit a transaction
  645. * around some arbitrary function passed as argument
  646. *
  647. * @return void
  648. */
  649. public function testTransactionalSuccess() {
  650. $driver = $this->getMockFormDriver();
  651. $connection = $this->getMock(
  652. '\Cake\Database\Connection',
  653. ['connect', 'commit', 'begin'],
  654. [['driver' => $driver]]
  655. );
  656. $connection->expects($this->at(0))->method('begin');
  657. $connection->expects($this->at(1))->method('commit');
  658. $result = $connection->transactional(function($conn) use ($connection) {
  659. $this->assertSame($connection, $conn);
  660. return 'thing';
  661. });
  662. $this->assertEquals('thing', $result);
  663. }
  664. /**
  665. * Tests that the transactional method will rollback the transaction if false
  666. * is returned from the callback
  667. *
  668. * @return void
  669. */
  670. public function testTransactionalFail() {
  671. $driver = $this->getMockFormDriver();
  672. $connection = $this->getMock(
  673. '\Cake\Database\Connection',
  674. ['connect', 'commit', 'begin', 'rollback'],
  675. [['driver' => $driver]]
  676. );
  677. $connection->expects($this->at(0))->method('begin');
  678. $connection->expects($this->at(1))->method('rollback');
  679. $connection->expects($this->never())->method('commit');
  680. $result = $connection->transactional(function($conn) use ($connection) {
  681. $this->assertSame($connection, $conn);
  682. return false;
  683. });
  684. $this->assertFalse($result);
  685. }
  686. /**
  687. * Tests that the transactional method will rollback the transaction
  688. * and throw the same exception if the callback raises one
  689. *
  690. * @expectedException \InvalidArgumentException
  691. * @return void
  692. * @throws \InvalidArgumentException
  693. */
  694. public function testTransactionalWithException() {
  695. $driver = $this->getMockFormDriver();
  696. $connection = $this->getMock(
  697. '\Cake\Database\Connection',
  698. ['connect', 'commit', 'begin', 'rollback'],
  699. [['driver' => $driver]]
  700. );
  701. $connection->expects($this->at(0))->method('begin');
  702. $connection->expects($this->at(1))->method('rollback');
  703. $connection->expects($this->never())->method('commit');
  704. $connection->transactional(function($conn) use ($connection) {
  705. $this->assertSame($connection, $conn);
  706. throw new \InvalidArgumentException;
  707. });
  708. }
  709. }