ConnectionTest.php 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809
  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. $this->connection = ConnectionManager::get('test');
  27. parent::setUp();
  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. $result->closeCursor();
  333. }
  334. /**
  335. * Tests delete from table with conditions
  336. * @return void
  337. **/
  338. public function testDeleteWithConditions() {
  339. $this->connection->delete('things', ['id' => '1-rest-is-ommited'], ['id' => 'integer']);
  340. $result = $this->connection->execute('SELECT * FROM things');
  341. $this->assertCount(1, $result);
  342. $result->closeCursor();
  343. $this->connection->delete('things', ['id' => '1-rest-is-ommited'], ['id' => 'integer']);
  344. $result = $this->connection->execute('SELECT * FROM things');
  345. $this->assertCount(1, $result);
  346. $result->closeCursor();
  347. $this->connection->delete('things', ['id' => '2-rest-is-ommited'], ['id' => 'integer']);
  348. $result = $this->connection->execute('SELECT * FROM things');
  349. $this->assertCount(0, $result);
  350. $result->closeCursor();
  351. }
  352. /**
  353. * Tests that it is possible to use simple database transactions
  354. *
  355. * @return void
  356. **/
  357. public function testSimpleTransactions() {
  358. $this->connection->begin();
  359. $this->connection->delete('things', ['id' => 1]);
  360. $this->connection->rollback();
  361. $result = $this->connection->execute('SELECT * FROM things');
  362. $this->assertCount(2, $result);
  363. $result->closeCursor();
  364. $this->connection->begin();
  365. $this->connection->delete('things', ['id' => 1]);
  366. $this->connection->commit();
  367. $result = $this->connection->execute('SELECT * FROM things');
  368. $this->assertCount(1, $result);
  369. }
  370. /**
  371. * Tests that it is possible to use virtualized nested transaction
  372. * with early rollback algorithm
  373. *
  374. * @return void
  375. **/
  376. public function testVirtualNestedTrasanction() {
  377. //starting 3 virtual transaction
  378. $this->connection->begin();
  379. $this->connection->begin();
  380. $this->connection->begin();
  381. $this->connection->delete('things', ['id' => 1]);
  382. $result = $this->connection->execute('SELECT * FROM things');
  383. $this->assertCount(1, $result);
  384. $this->connection->commit();
  385. $this->connection->rollback();
  386. $result = $this->connection->execute('SELECT * FROM things');
  387. $this->assertCount(2, $result);
  388. }
  389. /**
  390. * Tests that it is possible to use virtualized nested transaction
  391. * with early rollback algorithm
  392. *
  393. * @return void
  394. **/
  395. public function testVirtualNestedTrasanction2() {
  396. //starting 3 virtual transaction
  397. $this->connection->begin();
  398. $this->connection->begin();
  399. $this->connection->begin();
  400. $this->connection->delete('things', ['id' => 1]);
  401. $result = $this->connection->execute('SELECT * FROM things');
  402. $this->assertCount(1, $result);
  403. $this->connection->rollback();
  404. $result = $this->connection->execute('SELECT * FROM things');
  405. $this->assertCount(2, $result);
  406. }
  407. /**
  408. * Tests that it is possible to use virtualized nested transaction
  409. * with early rollback algorithm
  410. *
  411. * @return void
  412. **/
  413. public function testVirtualNestedTrasanction3() {
  414. //starting 3 virtual transaction
  415. $this->connection->begin();
  416. $this->connection->begin();
  417. $this->connection->begin();
  418. $this->connection->delete('things', ['id' => 1]);
  419. $result = $this->connection->execute('SELECT * FROM things');
  420. $this->assertCount(1, $result);
  421. $this->connection->commit();
  422. $this->connection->commit();
  423. $this->connection->commit();
  424. $result = $this->connection->execute('SELECT * FROM things');
  425. $this->assertCount(1, $result);
  426. }
  427. /**
  428. * Tests that it is possible to real use nested transactions
  429. *
  430. * @return void
  431. **/
  432. public function testSavePoints() {
  433. $this->skipIf(!$this->connection->useSavePoints(true));
  434. $this->connection->begin();
  435. $this->connection->delete('things', ['id' => 1]);
  436. $result = $this->connection->execute('SELECT * FROM things');
  437. $this->assertCount(1, $result);
  438. $this->connection->begin();
  439. $this->connection->delete('things', ['id' => 2]);
  440. $result = $this->connection->execute('SELECT * FROM things');
  441. $this->assertCount(0, $result);
  442. $this->connection->rollback();
  443. $result = $this->connection->execute('SELECT * FROM things');
  444. $this->assertCount(1, $result);
  445. $this->connection->rollback();
  446. $result = $this->connection->execute('SELECT * FROM things');
  447. $this->assertCount(2, $result);
  448. }
  449. /**
  450. * Tests that it is possible to real use nested transactions
  451. *
  452. * @return void
  453. **/
  454. public function testSavePoints2() {
  455. $this->skipIf(!$this->connection->useSavePoints(true));
  456. $this->connection->begin();
  457. $this->connection->delete('things', ['id' => 1]);
  458. $result = $this->connection->execute('SELECT * FROM things');
  459. $this->assertCount(1, $result);
  460. $this->connection->begin();
  461. $this->connection->delete('things', ['id' => 2]);
  462. $result = $this->connection->execute('SELECT * FROM things');
  463. $this->assertCount(0, $result);
  464. $this->connection->rollback();
  465. $result = $this->connection->execute('SELECT * FROM things');
  466. $this->assertCount(1, $result);
  467. $this->connection->commit();
  468. $result = $this->connection->execute('SELECT * FROM things');
  469. $this->assertCount(1, $result);
  470. }
  471. /**
  472. * Tests connection can quote values to be safely used in query strings
  473. *
  474. * @return void
  475. **/
  476. public function testQuote() {
  477. $this->skipIf(!$this->connection->supportsQuoting());
  478. $expected = "'2012-01-01'";
  479. $result = $this->connection->quote(new \DateTime('2012-01-01'), 'date');
  480. $this->assertEquals($expected, $result);
  481. $expected = "'1'";
  482. $result = $this->connection->quote(1, 'string');
  483. $this->assertEquals($expected, $result);
  484. $expected = "'hello'";
  485. $result = $this->connection->quote('hello', 'string');
  486. $this->assertEquals($expected, $result);
  487. }
  488. /**
  489. * Tests identifier quoting
  490. *
  491. * @return void
  492. */
  493. public function testQuoteIdentifier() {
  494. $driver = $this->getMock('Cake\Database\Driver\Sqlite', ['enabled']);
  495. $driver->expects($this->once())
  496. ->method('enabled')
  497. ->will($this->returnValue(true));
  498. $connection = new Connection(['driver' => $driver]);
  499. $result = $connection->quoteIdentifier('name');
  500. $expected = '"name"';
  501. $this->assertEquals($expected, $result);
  502. $result = $connection->quoteIdentifier('Model.*');
  503. $expected = '"Model".*';
  504. $this->assertEquals($expected, $result);
  505. $result = $connection->quoteIdentifier('MTD()');
  506. $expected = 'MTD()';
  507. $this->assertEquals($expected, $result);
  508. $result = $connection->quoteIdentifier('(sm)');
  509. $expected = '(sm)';
  510. $this->assertEquals($expected, $result);
  511. $result = $connection->quoteIdentifier('name AS x');
  512. $expected = '"name" AS "x"';
  513. $this->assertEquals($expected, $result);
  514. $result = $connection->quoteIdentifier('Model.name AS x');
  515. $expected = '"Model"."name" AS "x"';
  516. $this->assertEquals($expected, $result);
  517. $result = $connection->quoteIdentifier('Function(Something.foo)');
  518. $expected = 'Function("Something"."foo")';
  519. $this->assertEquals($expected, $result);
  520. $result = $connection->quoteIdentifier('Function(SubFunction(Something.foo))');
  521. $expected = 'Function(SubFunction("Something"."foo"))';
  522. $this->assertEquals($expected, $result);
  523. $result = $connection->quoteIdentifier('Function(Something.foo) AS x');
  524. $expected = 'Function("Something"."foo") AS "x"';
  525. $this->assertEquals($expected, $result);
  526. $result = $connection->quoteIdentifier('name-with-minus');
  527. $expected = '"name-with-minus"';
  528. $this->assertEquals($expected, $result);
  529. $result = $connection->quoteIdentifier('my-name');
  530. $expected = '"my-name"';
  531. $this->assertEquals($expected, $result);
  532. $result = $connection->quoteIdentifier('Foo-Model.*');
  533. $expected = '"Foo-Model".*';
  534. $this->assertEquals($expected, $result);
  535. $result = $connection->quoteIdentifier('Team.P%');
  536. $expected = '"Team"."P%"';
  537. $this->assertEquals($expected, $result);
  538. $result = $connection->quoteIdentifier('Team.G/G');
  539. $expected = '"Team"."G/G"';
  540. $result = $connection->quoteIdentifier('Model.name as y');
  541. $expected = '"Model"."name" AS "y"';
  542. $this->assertEquals($expected, $result);
  543. }
  544. /**
  545. * Tests default return vale for logger() function
  546. *
  547. * @return void
  548. */
  549. public function testLoggerDefault() {
  550. $logger = $this->connection->logger();
  551. $this->assertInstanceOf('\Cake\Database\Log\QueryLogger', $logger);
  552. $this->assertSame($logger, $this->connection->logger());
  553. }
  554. /**
  555. * Tests that a custom logger object can be set
  556. *
  557. * @return void
  558. */
  559. public function testSetLogger() {
  560. $logger = new \Cake\Database\Log\QueryLogger;
  561. $this->connection->logger($logger);
  562. $this->assertSame($logger, $this->connection->logger());
  563. }
  564. /**
  565. * Tests that statements are decorated with a logger when logQueries is set to true
  566. *
  567. * @return void
  568. */
  569. public function testLoggerDecorator() {
  570. $logger = new \Cake\Database\Log\QueryLogger;
  571. $this->connection->logQueries(true);
  572. $this->connection->logger($logger);
  573. $st = $this->connection->prepare('SELECT 1');
  574. $this->assertInstanceOf('\Cake\Database\Log\LoggingStatement', $st);
  575. $this->assertSame($logger, $st->logger());
  576. $this->connection->logQueries(false);
  577. $st = $this->connection->prepare('SELECT 1');
  578. $this->assertNotInstanceOf('\Cake\Database\Log\LoggingStatement', $st);
  579. }
  580. /**
  581. * Tests that log() function logs to the configured query logger
  582. *
  583. * @return void
  584. */
  585. public function testLogFunction() {
  586. $logger = $this->getMock('\Cake\Database\Log\QueryLogger');
  587. $this->connection->logger($logger);
  588. $logger->expects($this->once())->method('log')
  589. ->with($this->logicalAnd(
  590. $this->isInstanceOf('\Cake\Database\Log\LoggedQuery'),
  591. $this->attributeEqualTo('query', 'SELECT 1')
  592. ));
  593. $this->connection->log('SELECT 1');
  594. }
  595. /**
  596. * Tests that begin and rollback are also logged
  597. *
  598. * @return void
  599. */
  600. public function testLogBeginRollbackTransaction() {
  601. $connection = $this
  602. ->getMockBuilder('\Cake\Database\Connection')
  603. ->setMethods(['connect'])
  604. ->disableOriginalConstructor()
  605. ->getMock();
  606. $connection->logQueries(true);
  607. $driver = $this->getMockFormDriver();
  608. $connection->driver($driver);
  609. $logger = $this->getMock('\Cake\Database\Log\QueryLogger');
  610. $connection->logger($logger);
  611. $logger->expects($this->at(0))->method('log')
  612. ->with($this->logicalAnd(
  613. $this->isInstanceOf('\Cake\Database\Log\LoggedQuery'),
  614. $this->attributeEqualTo('query', 'BEGIN')
  615. ));
  616. $logger->expects($this->at(1))->method('log')
  617. ->with($this->logicalAnd(
  618. $this->isInstanceOf('\Cake\Database\Log\LoggedQuery'),
  619. $this->attributeEqualTo('query', 'ROLLBACK')
  620. ));
  621. $connection->begin();
  622. $connection->begin(); //This one will not be logged
  623. $connection->rollback();
  624. }
  625. /**
  626. * Tests that commits are logged
  627. *
  628. * @return void
  629. */
  630. public function testLogCommitTransaction() {
  631. $driver = $this->getMockFormDriver();
  632. $connection = $this->getMock(
  633. '\Cake\Database\Connection',
  634. ['connect'],
  635. [['driver' => $driver]]
  636. );
  637. $logger = $this->getMock('\Cake\Database\Log\QueryLogger');
  638. $connection->logger($logger);
  639. $logger->expects($this->at(1))->method('log')
  640. ->with($this->logicalAnd(
  641. $this->isInstanceOf('\Cake\Database\Log\LoggedQuery'),
  642. $this->attributeEqualTo('query', 'COMMIT')
  643. ));
  644. $connection->logQueries(true);
  645. $connection->begin();
  646. $connection->commit();
  647. }
  648. /**
  649. * Tests that the transactional method will start and commit a transaction
  650. * around some arbitrary function passed as argument
  651. *
  652. * @return void
  653. */
  654. public function testTransactionalSuccess() {
  655. $driver = $this->getMockFormDriver();
  656. $connection = $this->getMock(
  657. '\Cake\Database\Connection',
  658. ['connect', 'commit', 'begin'],
  659. [['driver' => $driver]]
  660. );
  661. $connection->expects($this->at(0))->method('begin');
  662. $connection->expects($this->at(1))->method('commit');
  663. $result = $connection->transactional(function($conn) use ($connection) {
  664. $this->assertSame($connection, $conn);
  665. return 'thing';
  666. });
  667. $this->assertEquals('thing', $result);
  668. }
  669. /**
  670. * Tests that the transactional method will rollback the transaction if false
  671. * is returned from the callback
  672. *
  673. * @return void
  674. */
  675. public function testTransactionalFail() {
  676. $driver = $this->getMockFormDriver();
  677. $connection = $this->getMock(
  678. '\Cake\Database\Connection',
  679. ['connect', 'commit', 'begin', 'rollback'],
  680. [['driver' => $driver]]
  681. );
  682. $connection->expects($this->at(0))->method('begin');
  683. $connection->expects($this->at(1))->method('rollback');
  684. $connection->expects($this->never())->method('commit');
  685. $result = $connection->transactional(function($conn) use ($connection) {
  686. $this->assertSame($connection, $conn);
  687. return false;
  688. });
  689. $this->assertFalse($result);
  690. }
  691. /**
  692. * Tests that the transactional method will rollback the transaction
  693. * and throw the same exception if the callback raises one
  694. *
  695. * @expectedException \InvalidArgumentException
  696. * @return void
  697. * @throws \InvalidArgumentException
  698. */
  699. public function testTransactionalWithException() {
  700. $driver = $this->getMockFormDriver();
  701. $connection = $this->getMock(
  702. '\Cake\Database\Connection',
  703. ['connect', 'commit', 'begin', 'rollback'],
  704. [['driver' => $driver]]
  705. );
  706. $connection->expects($this->at(0))->method('begin');
  707. $connection->expects($this->at(1))->method('rollback');
  708. $connection->expects($this->never())->method('commit');
  709. $connection->transactional(function($conn) use ($connection) {
  710. $this->assertSame($connection, $conn);
  711. throw new \InvalidArgumentException;
  712. });
  713. }
  714. }