TranslationQueryWalkerTest.php 28KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717
  1. <?php
  2. namespace Gedmo\Translatable;
  3. use Doctrine\Common\EventManager;
  4. use Tool\BaseTestCaseORM;
  5. use Doctrine\ORM\Query;
  6. use Gedmo\Translatable\Query\TreeWalker\TranslationWalker;
  7. use Translatable\Fixture\Article;
  8. use Translatable\Fixture\Comment;
  9. /**
  10. * These are tests for translation query walker
  11. *
  12. * @author Gediminas Morkevicius <gediminas.morkevicius@gmail.com>
  13. * @package Gedmo.Translatable
  14. * @link http://www.gediminasm.org
  15. * @license MIT License (http://www.opensource.org/licenses/mit-license.php)
  16. */
  17. class TranslationQueryWalkerTest extends BaseTestCaseORM
  18. {
  19. const ARTICLE = 'Translatable\\Fixture\\Article';
  20. const COMMENT = 'Translatable\\Fixture\\Comment';
  21. const TRANSLATION = 'Gedmo\\Translatable\\Entity\\Translation';
  22. const TREE_WALKER_TRANSLATION = 'Gedmo\\Translatable\\Query\\TreeWalker\\TranslationWalker';
  23. private $translatableListener;
  24. protected function setUp()
  25. {
  26. parent::setUp();
  27. $evm = new EventManager;
  28. $this->translatableListener = new TranslatableListener();
  29. $this->translatableListener->setTranslatableLocale('en_us');
  30. $this->translatableListener->setDefaultLocale('en_us');
  31. $evm->addEventSubscriber($this->translatableListener);
  32. $this->getMockSqliteEntityManager($evm);
  33. $this->populate();
  34. }
  35. /**
  36. * @test
  37. */
  38. function shouldHandleQueryCache()
  39. {
  40. $cache = new \Doctrine\Common\Cache\ArrayCache();
  41. $this->em
  42. ->getConfiguration()
  43. ->expects($this->any())
  44. ->method('getQueryCacheImpl')
  45. ->will($this->returnValue($cache))
  46. ;
  47. $dql = 'SELECT a FROM ' . self::ARTICLE . ' a';
  48. $q = $this->em->createQuery($dql);
  49. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  50. // array hydration
  51. $this->translatableListener->setTranslatableLocale('en_us');
  52. $result = $q->getArrayResult();
  53. $this->assertCount(1, $result);
  54. $q2 = clone $q;
  55. $q2->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  56. $result = $q->getArrayResult();
  57. $this->assertCount(1, $result);
  58. }
  59. public function testSubselectByTranslatedField()
  60. {
  61. $this->populateMore();
  62. $dql = 'SELECT a FROM ' . self::ARTICLE . ' a';
  63. $subSelect = 'SELECT a2.title FROM ' . self::ARTICLE . ' a2';
  64. $subSelect .= " WHERE a2.title LIKE '%ab%'";
  65. $dql .= " WHERE a.title IN ({$subSelect})";
  66. $dql .= ' ORDER BY a.title';
  67. $q = $this->em->createQuery($dql);
  68. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  69. // array hydration
  70. $this->translatableListener->setTranslatableLocale('en_us');
  71. $result = $q->getArrayResult();
  72. $this->assertCount(2, $result);
  73. $this->assertEquals('Alfabet', $result[0]['title']);
  74. $this->assertEquals('Cabbages', $result[1]['title']);
  75. }
  76. public function testSubselectStatements()
  77. {
  78. $this->populateMore();
  79. $dql = 'SELECT a FROM ' . self::ARTICLE . ' a';
  80. $subSelect = 'SELECT a2.id FROM ' . self::ARTICLE . ' a2';
  81. $subSelect .= " WHERE a2.title LIKE '%ab%'";
  82. $dql .= " WHERE a.id IN ({$subSelect})";
  83. $dql .= ' ORDER BY a.title';
  84. $q = $this->em->createQuery($dql);
  85. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  86. // array hydration
  87. $this->translatableListener->setTranslatableLocale('en_us');
  88. $result = $q->getArrayResult();
  89. $this->assertCount(2, $result);
  90. $this->assertEquals('Alfabet', $result[0]['title']);
  91. $this->assertEquals('Cabbages', $result[1]['title']);
  92. }
  93. public function testJoinedWithStatements()
  94. {
  95. $this->populateMore();
  96. $dql = 'SELECT a, c FROM ' . self::ARTICLE . ' a';
  97. $dql .= ' LEFT JOIN a.comments c WITH c.subject LIKE :lookup';
  98. $dql .= ' WHERE a.title LIKE :filter';
  99. $dql .= ' ORDER BY a.title';
  100. $q = $this->em->createQuery($dql);
  101. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  102. // array hydration
  103. $this->translatableListener->setTranslatableLocale('en_us');
  104. $q->setParameter('lookup', '%goo%');
  105. $q->setParameter('filter', 'Foo%');
  106. $result = $q->getArrayResult();
  107. $this->assertCount(1, $result);
  108. $this->assertEquals('Food', $result[0]['title']);
  109. $comments = $result[0]['comments'];
  110. $this->assertCount(1, $comments);
  111. $this->assertEquals('good', $comments[0]['subject']);
  112. }
  113. public function testSelectWithTranslationFallbackOnSimpleObjectHydration()
  114. {
  115. $this->em
  116. ->getConfiguration()
  117. ->expects($this->any())
  118. ->method('getCustomHydrationMode')
  119. ->with(TranslationWalker::HYDRATE_SIMPLE_OBJECT_TRANSLATION)
  120. ->will($this->returnValue('Gedmo\\Translatable\\Hydrator\\ORM\\SimpleObjectHydrator'));
  121. $dql = 'SELECT a FROM ' . self::ARTICLE . ' a';
  122. $q = $this->em->createQuery($dql);
  123. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  124. $this->translatableListener->setTranslatableLocale('ru_ru');
  125. $this->translatableListener->setTranslationFallback(false);
  126. // simple object hydration
  127. $this->startQueryLog();
  128. $result = $q->getResult(Query::HYDRATE_SIMPLEOBJECT);
  129. $this->assertEquals(1, $this->queryAnalyzer->getNumExecutedQueries());
  130. $this->assertEquals('', $result[0]->getTitle());
  131. $this->assertEquals('', $result[0]->getContent());
  132. $this->translatableListener->setTranslationFallback(true);
  133. $this->queryAnalyzer->cleanUp();
  134. $result = $q->getResult(Query::HYDRATE_SIMPLEOBJECT);
  135. $this->assertEquals(1, $this->queryAnalyzer->getNumExecutedQueries());
  136. //Default translation is en_us, so we expect the results in that locale
  137. $this->assertEquals('Food', $result[0]->getTitle());
  138. $this->assertEquals('about food', $result[0]->getContent());
  139. }
  140. public function testSelectWithTranslationFallbackOnArrayHydration()
  141. {
  142. $dql = 'SELECT a, c FROM ' . self::ARTICLE . ' a';
  143. $dql .= ' LEFT JOIN a.comments c';
  144. $q = $this->em->createQuery($dql);
  145. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  146. $this->translatableListener->setTranslatableLocale('ru_ru');
  147. $this->translatableListener->setTranslationFallback(false);
  148. // array hydration
  149. $this->startQueryLog();
  150. $result = $q->getArrayResult();
  151. $this->assertEquals(1, $this->queryAnalyzer->getNumExecutedQueries());
  152. $this->assertEquals('', $result[0]['title']);
  153. $this->assertEquals('', $result[0]['content']);
  154. $this->translatableListener->setTranslationFallback(true);
  155. $this->queryAnalyzer->cleanUp();
  156. $result = $q->getArrayResult();
  157. $this->assertEquals(1, $this->queryAnalyzer->getNumExecutedQueries());
  158. //Default translation is en_us, so we expect the results in that locale
  159. $this->assertEquals('Food', $result[0]['title']);
  160. $this->assertEquals('about food', $result[0]['content']);
  161. }
  162. public function testSelectWithOptionalFallbackOnSimpleObjectHydration()
  163. {
  164. $this->em
  165. ->getConfiguration()
  166. ->expects($this->any())
  167. ->method('getCustomHydrationMode')
  168. ->with(TranslationWalker::HYDRATE_SIMPLE_OBJECT_TRANSLATION)
  169. ->will($this->returnValue('Gedmo\\Translatable\\Hydrator\\ORM\\SimpleObjectHydrator'));
  170. $dql = 'SELECT a FROM ' . self::ARTICLE . ' a';
  171. $q = $this->em->createQuery($dql);
  172. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  173. $this->translatableListener->setTranslatableLocale('ru_ru');
  174. $this->translatableListener->setTranslationFallback(false);
  175. // simple object hydration
  176. $this->startQueryLog();
  177. $result = $q->getResult(Query::HYDRATE_SIMPLEOBJECT);
  178. $this->assertEquals(1, $this->queryAnalyzer->getNumExecutedQueries());
  179. $this->assertEquals('', $result[0]->getTitle());
  180. $this->assertEquals('John Doe', $result[0]->getAuthor()); // optional fallback is true, force fallback
  181. $this->assertEquals(0, $result[0]->getViews());
  182. $this->translatableListener->setTranslationFallback(true);
  183. $this->queryAnalyzer->cleanUp();
  184. $result = $q->getResult(Query::HYDRATE_SIMPLEOBJECT);
  185. $this->assertEquals(1, $this->queryAnalyzer->getNumExecutedQueries());
  186. //Default translation is en_us, so we expect the results in that locale
  187. $this->assertEquals('Food', $result[0]->getTitle());
  188. $this->assertEquals('John Doe', $result[0]->getAuthor());
  189. $this->assertEquals(0, $result[0]->getViews()); // optional fallback is false, thus no translation required
  190. }
  191. /**
  192. * @test
  193. */
  194. public function shouldBeAbleToUseInnerJoinStrategyForTranslations()
  195. {
  196. $dql = 'SELECT a FROM ' . self::ARTICLE . ' a';
  197. $q = $this->em->createQuery($dql);
  198. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  199. $q->setHint(TranslatableListener::HINT_INNER_JOIN, true);
  200. $this->translatableListener->setTranslatableLocale('ru_ru');
  201. $this->translatableListener->setTranslationFallback(false);
  202. // array hydration
  203. $result = $q->getArrayResult();
  204. $this->assertCount(0, $result);
  205. }
  206. /**
  207. * @test
  208. */
  209. public function shouldBeAbleToOverrideTranslatableLocale()
  210. {
  211. $dql = 'SELECT a FROM ' . self::ARTICLE . ' a';
  212. $q = $this->em->createQuery($dql);
  213. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  214. $q->setHint(TranslatableListener::HINT_TRANSLATABLE_LOCALE, 'lt_lt');
  215. $this->translatableListener->setTranslatableLocale('ru_ru');
  216. $this->translatableListener->setTranslationFallback(false);
  217. // array hydration
  218. $result = $q->getArrayResult();
  219. $this->assertCount(1, $result);
  220. $this->assertEquals('Maistas', $result[0]['title']);
  221. }
  222. public function testSelectWithTranslationFallbackOnObjectHydration()
  223. {
  224. $this->em
  225. ->getConfiguration()
  226. ->expects($this->any())
  227. ->method('getCustomHydrationMode')
  228. ->with(TranslationWalker::HYDRATE_OBJECT_TRANSLATION)
  229. ->will($this->returnValue('Gedmo\\Translatable\\Hydrator\\ORM\\ObjectHydrator'));
  230. $dql = 'SELECT a FROM ' . self::ARTICLE . ' a';
  231. $q = $this->em->createQuery($dql);
  232. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  233. $this->translatableListener->setTranslatableLocale('ru_ru');
  234. $this->translatableListener->setTranslationFallback(false);
  235. // object hydration
  236. $this->startQueryLog();
  237. $result = $q->getResult();
  238. $this->assertEquals(1, $this->queryAnalyzer->getNumExecutedQueries());
  239. $this->assertEquals('', $result[0]->getTitle());
  240. $this->assertEquals('', $result[0]->getContent());
  241. $this->translatableListener->setTranslationFallback(true);
  242. $this->queryAnalyzer->cleanUp();
  243. $result = $q->getResult();
  244. $this->assertEquals(1, $this->queryAnalyzer->getNumExecutedQueries());
  245. //Default translation is en_us, so we expect the results in that locale
  246. $this->assertEquals('Food', $result[0]->getTitle());
  247. $this->assertEquals('about food', $result[0]->getContent());
  248. // test fallback hint
  249. $this->translatableListener->setTranslationFallback(false);
  250. $q->setHint(TranslatableListener::HINT_FALLBACK, 1);
  251. $result = $q->getResult();
  252. //Default translation is en_us, so we expect the results in that locale
  253. $this->assertEquals('Food', $result[0]->getTitle());
  254. $this->assertEquals('about food', $result[0]->getContent());
  255. // test fallback hint
  256. $this->translatableListener->setTranslationFallback(true);
  257. $q->setHint(TranslatableListener::HINT_FALLBACK, 0);
  258. $result = $q->getResult();
  259. //Default translation is en_us, so we expect the results in that locale
  260. $this->assertEquals('', $result[0]->getTitle());
  261. $this->assertEquals('', $result[0]->getContent());
  262. }
  263. public function testSelectCountStatement()
  264. {
  265. $dql = 'SELECT COUNT(a) FROM ' . self::ARTICLE . ' a';
  266. $dql .= ' WHERE a.title LIKE :title';
  267. $q = $this->em->createQuery($dql);
  268. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  269. $this->translatableListener->setTranslatableLocale('en_us');
  270. $q->setParameter('title', 'Foo%');
  271. $result = $q->getSingleScalarResult();
  272. $this->assertEquals(1, $result);
  273. $this->translatableListener->setTranslatableLocale('lt_lt');
  274. $q->setParameter('title', 'Mai%');
  275. $result = $q->getSingleScalarResult();
  276. $this->assertEquals(1, $result);
  277. $this->translatableListener->setTranslatableLocale('en_us');
  278. $q->setParameter('title', 'Mai%');
  279. $result = $q->getSingleScalarResult();
  280. $this->assertEquals(0, $result);
  281. }
  282. public function testSelectOrderedJoinedComponentTranslation()
  283. {
  284. $this->em
  285. ->getConfiguration()
  286. ->expects($this->any())
  287. ->method('getCustomHydrationMode')
  288. ->with(TranslationWalker::HYDRATE_OBJECT_TRANSLATION)
  289. ->will($this->returnValue('Gedmo\\Translatable\\Hydrator\\ORM\\ObjectHydrator'));
  290. $this->populateMore();
  291. $dql = 'SELECT a, c FROM ' . self::ARTICLE . ' a';
  292. $dql .= ' LEFT JOIN a.comments c';
  293. $dql .= ' ORDER BY a.title';
  294. $q = $this->em->createQuery($dql);
  295. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  296. // array hydration
  297. $this->translatableListener->setTranslatableLocale('en_us');
  298. $result = $q->getArrayResult();
  299. $this->assertCount(4, $result);
  300. $this->assertEquals('Alfabet', $result[0]['title']);
  301. $this->assertEquals('Cabbages', $result[1]['title']);
  302. $this->assertEquals('Food', $result[2]['title']);
  303. $this->assertEquals('Woman', $result[3]['title']);
  304. $this->translatableListener->setTranslatableLocale('lt_lt');
  305. $result = $q->getArrayResult();
  306. $this->assertCount(4, $result);
  307. $this->assertEquals('Alfabetas', $result[0]['title']);
  308. $this->assertEquals('Kopustai', $result[1]['title']);
  309. $this->assertEquals('Maistas', $result[2]['title']);
  310. $this->assertEquals('Moteris', $result[3]['title']);
  311. // object hydration
  312. $this->translatableListener->setTranslatableLocale('en_us');
  313. $result = $q->getResult();
  314. $this->assertCount(4, $result);
  315. $this->assertEquals('Alfabet', $result[0]->getTitle());
  316. $this->assertEquals('Cabbages', $result[1]->getTitle());
  317. $this->assertEquals('Food', $result[2]->getTitle());
  318. $this->assertEquals('Woman', $result[3]->getTitle());
  319. $this->translatableListener->setTranslatableLocale('lt_lt');
  320. $result = $q->getResult();
  321. $this->assertCount(4, $result);
  322. $this->assertEquals('Alfabetas', $result[0]->getTitle());
  323. $this->assertEquals('Kopustai', $result[1]->getTitle());
  324. $this->assertEquals('Maistas', $result[2]->getTitle());
  325. $this->assertEquals('Moteris', $result[3]->getTitle());
  326. }
  327. public function testSelectOrderedByTranslatableInteger()
  328. {
  329. // Given
  330. $this->populateMore();
  331. $dql = 'SELECT a.title, a.views FROM ' . self::ARTICLE . ' a';
  332. $dql .= ' ORDER BY a.views';
  333. $q = $this->em->createQuery($dql);
  334. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  335. // Test original
  336. $this->translatableListener->setTranslatableLocale('en_us');
  337. $result = $q->getArrayResult();
  338. array_walk($result, function($value, $key) use(&$result) {
  339. // Make each record be a "Title - Views" string
  340. $result[$key] = implode(" - ", $value);
  341. });
  342. $this->assertEquals(
  343. array("Alfabet - 1", "Food - 99", "Cabbages - 2222", "Woman - 3333"), $result,
  344. "Original of localizible integers should be sorted numerically"
  345. );
  346. $this->translatableListener->setTranslatableLocale('lt_lt');
  347. $result = $q->getArrayResult();
  348. array_walk($result, function($value, $key) use(&$result) {
  349. // Make each record be a "Title - Views" string
  350. $result[$key] = implode(" - ", $value);
  351. });
  352. $this->assertEquals(
  353. array("Moteris - 33", "Alfabetas - 111", "Maistas - 999", "Kopustai - 22222"), $result,
  354. "Localized integers should be sorted numerically"
  355. );
  356. }
  357. public function testSelectSecondJoinedComponentTranslation()
  358. {
  359. $this->em
  360. ->getConfiguration()
  361. ->expects($this->any())
  362. ->method('getCustomHydrationMode')
  363. ->with(TranslationWalker::HYDRATE_OBJECT_TRANSLATION)
  364. ->will($this->returnValue('Gedmo\\Translatable\\Hydrator\\ORM\\ObjectHydrator'));
  365. $dql = 'SELECT a, c FROM ' . self::ARTICLE . ' a';
  366. $dql .= ' LEFT JOIN a.comments c';
  367. $q = $this->em->createQuery($dql);
  368. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  369. // array hydration
  370. $this->translatableListener->setTranslatableLocale('en_us');
  371. $result = $q->getArrayResult();
  372. $this->assertCount(1, $result);
  373. $food = $result[0];
  374. $this->assertCount(6, $food);
  375. $this->assertEquals('Food', $food['title']);
  376. $this->assertEquals('about food', $food['content']);
  377. $comments = $food['comments'];
  378. $this->assertCount(2, $comments);
  379. $good = $comments[0];
  380. $this->assertCount(3, $good);
  381. $this->assertEquals('good', $good['subject']);
  382. $this->assertEquals('food is good', $good['message']);
  383. $bad = $comments[1];
  384. $this->assertCount(3, $bad);
  385. $this->assertEquals('bad', $bad['subject']);
  386. $this->assertEquals('food is bad', $bad['message']);
  387. $this->translatableListener->setTranslatableLocale('lt_lt');
  388. $result = $q->getArrayResult();
  389. $this->assertCount(1, $result);
  390. $food = $result[0];
  391. $this->assertCount(6, $food);
  392. $this->assertEquals('Maistas', $food['title']);
  393. $this->assertEquals('apie maista', $food['content']);
  394. $comments = $food['comments'];
  395. $this->assertCount(2, $comments);
  396. $good = $comments[0];
  397. $this->assertCount(3, $good);
  398. $this->assertEquals('geras', $good['subject']);
  399. $this->assertEquals('maistas yra geras', $good['message']);
  400. $bad = $comments[1];
  401. $this->assertCount(3, $bad);
  402. $this->assertEquals('blogas', $bad['subject']);
  403. $this->assertEquals('maistas yra blogas', $bad['message']);
  404. // object hydration
  405. $this->translatableListener->setTranslatableLocale('en_us');
  406. $result = $q->getResult();
  407. $this->assertCount(1, $result);
  408. $food = $result[0];
  409. $this->assertEquals('Food', $food->getTitle());
  410. $this->assertEquals('about food', $food->getContent());
  411. $comments = $food->getComments();
  412. $this->assertCount(2, $comments);
  413. $good = $comments[0];
  414. $this->assertEquals('good', $good->getSubject());
  415. $this->assertEquals('food is good', $good->getMessage());
  416. $bad = $comments[1];
  417. $this->assertEquals('bad', $bad->getSubject());
  418. $this->assertEquals('food is bad', $bad->getMessage());
  419. $this->translatableListener->setTranslatableLocale('lt_lt');
  420. $result = $q->getResult();
  421. $this->assertCount(1, $result);
  422. $food = $result[0];
  423. $this->assertEquals('Maistas', $food->getTitle());
  424. $this->assertEquals('apie maista', $food->getContent());
  425. $comments = $food->getComments();
  426. $this->assertCount(2, $comments);
  427. $good = $comments[0];
  428. $this->assertInstanceOf(self::COMMENT, $good);
  429. $this->assertEquals('geras', $good->getSubject());
  430. $this->assertEquals('maistas yra geras', $good->getMessage());
  431. $bad = $comments[1];
  432. $this->assertEquals('blogas', $bad->getSubject());
  433. $this->assertEquals('maistas yra blogas', $bad->getMessage());
  434. }
  435. public function testSelectSinglePartializedComponentTranslation()
  436. {
  437. $this->em
  438. ->getConfiguration()
  439. ->expects($this->any())
  440. ->method('getCustomHydrationMode')
  441. ->with(TranslationWalker::HYDRATE_OBJECT_TRANSLATION)
  442. ->will($this->returnValue('Gedmo\\Translatable\\Hydrator\\ORM\\ObjectHydrator'));
  443. $dql = 'SELECT a.title FROM ' . self::ARTICLE . ' a';
  444. $q = $this->em->createQuery($dql);
  445. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  446. // array hydration
  447. $this->translatableListener->setTranslatableLocale('en_us');
  448. $result = $q->getArrayResult();
  449. $this->assertCount(1, $result);
  450. $food = $result[0];
  451. $this->assertCount(1, $food);
  452. $this->assertEquals('Food', $food['title']);
  453. $this->translatableListener->setTranslatableLocale('lt_lt');
  454. $result = $q->getArrayResult();
  455. $this->assertCount(1, $result);
  456. $food = $result[0];
  457. $this->assertCount(1, $food);
  458. $this->assertEquals('Maistas', $food['title']);
  459. // object hydration
  460. $this->translatableListener->setTranslatableLocale('en_us');
  461. $result = $q->getResult();
  462. $this->assertCount(1, $result);
  463. $food = $result[0];
  464. $this->assertCount(1, $food);
  465. $this->assertEquals('Food', $food['title']);
  466. $this->translatableListener->setTranslatableLocale('lt_lt');
  467. $result = $q->getResult();
  468. $this->assertCount(1, $result);
  469. $food = $result[0];
  470. $this->assertCount(1, $food);
  471. $this->assertEquals('Maistas', $food['title']);
  472. }
  473. public function testSelectSingleComponentTranslation()
  474. {
  475. $this->em
  476. ->getConfiguration()
  477. ->expects($this->any())
  478. ->method('getCustomHydrationMode')
  479. ->with(TranslationWalker::HYDRATE_OBJECT_TRANSLATION)
  480. ->will($this->returnValue('Gedmo\\Translatable\\Hydrator\\ORM\\ObjectHydrator'));
  481. $dql = 'SELECT a FROM ' . self::ARTICLE . ' a';
  482. $q = $this->em->createQuery($dql);
  483. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  484. // array hydration
  485. $this->translatableListener->setTranslatableLocale('en_us');
  486. $result = $q->getArrayResult();
  487. $this->assertCount(1, $result);
  488. $food = $result[0];
  489. $this->assertCount(5, $food);
  490. $this->assertEquals('Food', $food['title']);
  491. $this->assertEquals('about food', $food['content']);
  492. $this->translatableListener->setTranslatableLocale('lt_lt');
  493. $result = $q->getArrayResult();
  494. $this->assertCount(1, $result);
  495. $food = $result[0];
  496. $this->assertCount(5, $food);
  497. $this->assertEquals('Maistas', $food['title']);
  498. $this->assertEquals('apie maista', $food['content']);
  499. // object hydration
  500. $this->translatableListener->setTranslatableLocale('en_us');
  501. $result = $q->getResult();
  502. $this->assertCount(1, $result);
  503. $food = $result[0];
  504. $this->assertInstanceOf(self::ARTICLE, $food);
  505. $this->assertEquals('Food', $food->getTitle());
  506. $this->assertEquals('about food', $food->getContent());
  507. $this->translatableListener->setTranslatableLocale('lt_lt');
  508. $result = $q->getResult();
  509. $this->assertCount(1, $result);
  510. $food = $result[0];
  511. $this->assertEquals('Maistas', $food->getTitle());
  512. $this->assertEquals('apie maista', $food->getContent());
  513. }
  514. /**
  515. * @group testSelectWithUnmappedField
  516. */
  517. public function testSelectWithUnmappedField()
  518. {
  519. $dql = 'SELECT a.title, count(a.id) AS num FROM ' . self::ARTICLE . ' a';
  520. $dql .= ' ORDER BY a.title';
  521. $q = $this->em->createQuery($dql);
  522. $q->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, self::TREE_WALKER_TRANSLATION);
  523. // array hydration
  524. $this->translatableListener->setTranslatableLocale('en_us');
  525. $result = $q->getArrayResult();
  526. $this->assertCount(1, $result);
  527. $this->assertEquals('Food', $result[0]['title']);
  528. $this->assertEquals(1, $result[0]['num']);
  529. }
  530. protected function getUsedEntityFixtures()
  531. {
  532. return array(
  533. self::ARTICLE,
  534. self::TRANSLATION,
  535. self::COMMENT
  536. );
  537. }
  538. private function populateMore()
  539. {
  540. $repo = $this->em->getRepository(self::ARTICLE);
  541. $commentRepo = $this->em->getRepository(self::COMMENT);
  542. $this->translatableListener->setTranslatableLocale('en_us');
  543. $alfabet = new Article;
  544. $alfabet->setTitle('Alfabet');
  545. $alfabet->setContent('hey wtf!');
  546. $alfabet->setViews(1);
  547. $woman = new Article;
  548. $woman->setTitle('Woman');
  549. $woman->setContent('i like them');
  550. $woman->setViews(3333);
  551. $cabbages = new Article;
  552. $cabbages->setTitle('Cabbages');
  553. $cabbages->setContent('where went the woman?');
  554. $cabbages->setViews(2222);
  555. $this->em->persist($alfabet);
  556. $this->em->persist($woman);
  557. $this->em->persist($cabbages);
  558. $this->em->flush();
  559. $this->em->clear();
  560. $this->translatableListener->setTranslatableLocale('lt_lt');
  561. $alfabet = $repo->find(2);
  562. $alfabet->setTitle('Alfabetas');
  563. $alfabet->setContent('ei wtf!');
  564. $alfabet->setViews(111);
  565. $woman = $repo->find(3);
  566. $woman->setTitle('Moteris');
  567. $woman->setContent('as megstu jas');
  568. $woman->setViews(33);
  569. $cabbages = $repo->find(4);
  570. $cabbages->setTitle('Kopustai');
  571. $cabbages->setContent('kur dingo moteris?');
  572. $cabbages->setViews(22222);
  573. $this->em->persist($alfabet);
  574. $this->em->persist($woman);
  575. $this->em->persist($cabbages);
  576. $this->em->flush();
  577. $this->em->clear();
  578. }
  579. private function populate()
  580. {
  581. $repo = $this->em->getRepository(self::ARTICLE);
  582. $commentRepo = $this->em->getRepository(self::COMMENT);
  583. $food = new Article;
  584. $food->setTitle('Food');
  585. $food->setContent('about food');
  586. $food->setAuthor('John Doe');
  587. $food->setViews(99);
  588. $goodFood = new Comment;
  589. $goodFood->setArticle($food);
  590. $goodFood->setMessage('food is good');
  591. $goodFood->setSubject('good');
  592. $badFood = new Comment;
  593. $badFood->setArticle($food);
  594. $badFood->setMessage('food is bad');
  595. $badFood->setSubject('bad');
  596. $this->em->persist($food);
  597. $this->em->persist($goodFood);
  598. $this->em->persist($badFood);
  599. $this->em->flush();
  600. $this->em->clear();
  601. $this->translatableListener->setTranslatableLocale('lt_lt');
  602. $food = $repo->find(1);
  603. $food->setTitle('Maistas');
  604. $food->setContent('apie maista');
  605. $food->setViews(999);
  606. $goodFood = $commentRepo->find(1);
  607. $goodFood->setArticle($food);
  608. $goodFood->setMessage('maistas yra geras');
  609. $goodFood->setSubject('geras');
  610. $badFood = $commentRepo->find(2);
  611. $badFood->setArticle($food);
  612. $badFood->setMessage('maistas yra blogas');
  613. $badFood->setSubject('blogas');
  614. $this->em->persist($food);
  615. $this->em->persist($goodFood);
  616. $this->em->persist($badFood);
  617. $this->em->flush();
  618. $this->em->clear();
  619. }
  620. }