TableStorageTest.php 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786
  1. <?php
  2. /**
  3. * Zend Framework
  4. *
  5. * LICENSE
  6. *
  7. * This source file is subject to the new BSD license that is bundled
  8. * with this package in the file LICENSE.txt.
  9. * It is also available through the world-wide-web at this URL:
  10. * http://framework.zend.com/license/new-bsd
  11. * If you did not receive a copy of the license and are unable to
  12. * obtain it through the world-wide-web, please send an email
  13. * to license@zend.com so we can send you a copy immediately.
  14. *
  15. * @category Zend
  16. * @package Zend_Service_WindowsAzure
  17. * @subpackage UnitTests
  18. * @version $Id: BlobStorageTest.php 14561 2009-05-07 08:05:12Z unknown $
  19. * @copyright Copyright (c) 2005-2009 Zend Technologies USA Inc. (http://www.zend.com)
  20. * @license http://framework.zend.com/license/new-bsd New BSD License
  21. */
  22. require_once dirname(dirname(dirname(dirname(__FILE__)))) . '/TestHelper.php';
  23. if (!defined('PHPUnit_MAIN_METHOD')) {
  24. define('PHPUnit_MAIN_METHOD', 'Zend_Service_WindowsAzure_TableStorageTest::main');
  25. }
  26. require_once 'Zend/Service/WindowsAzure/Storage/Table.php';
  27. /**
  28. * @category Zend
  29. * @package Zend_Service_WindowsAzure
  30. * @subpackage UnitTests
  31. * @group Zend_Service
  32. * @group Zend_Service_WindowsAzure
  33. * @version $Id: BlobStorageTest.php 14561 2009-05-07 08:05:12Z unknown $
  34. * @copyright Copyright (c) 2005-2009 Zend Technologies USA Inc. (http://www.zend.com)
  35. * @license http://framework.zend.com/license/new-bsd New BSD License
  36. */
  37. class Zend_Service_WindowsAzure_TableStorageTest extends PHPUnit_Framework_TestCase
  38. {
  39. public function __construct()
  40. {
  41. }
  42. public static function main()
  43. {
  44. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  45. $suite = new PHPUnit_Framework_TestSuite("Zend_Service_WindowsAzure_TableStorageTest");
  46. $result = PHPUnit_TextUI_TestRunner::run($suite);
  47. }
  48. }
  49. /**
  50. * Test setup
  51. */
  52. protected function setUp()
  53. {
  54. }
  55. /**
  56. * Test teardown
  57. */
  58. protected function tearDown()
  59. {
  60. $storageClient = $this->createStorageInstance();
  61. for ($i = 1; $i <= self::$uniqId; $i++)
  62. {
  63. try { $storageClient->deleteTable(TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_TABLENAME_PREFIX . $i); } catch (Exception $e) { }
  64. }
  65. }
  66. protected function createStorageInstance()
  67. {
  68. $storageClient = null;
  69. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNONPROD) {
  70. $storageClient = new Zend_Service_WindowsAzure_Storage_Table(TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_HOST_PROD, TESTS_ZEND_SERVICE_WINDOWSAZURE_STORAGE_ACCOUNT_PROD, TESTS_ZEND_SERVICE_WINDOWSAZURE_STORAGE_KEY_PROD, false, Zend_Service_WindowsAzure_RetryPolicy_RetryPolicyAbstract::retryN(10, 250));
  71. } else {
  72. $storageClient = new Zend_Service_WindowsAzure_Storage_Table(TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_HOST_DEV, TESTS_ZEND_SERVICE_WINDOWSAZURE_STORAGE_ACCOUNT_DEV, TESTS_ZEND_SERVICE_WINDOWSAZURE_STORAGE_KEY_DEV, true, Zend_Service_WindowsAzure_RetryPolicy_RetryPolicyAbstract::retryN(10, 250));
  73. }
  74. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_STORAGE_USEPROXY) {
  75. $storageClient->setProxy(TESTS_ZEND_SERVICE_WINDOWSAZURE_STORAGE_USEPROXY, TESTS_ZEND_SERVICE_WINDOWSAZURE_STORAGE_PROXY, TESTS_ZEND_SERVICE_WINDOWSAZURE_STORAGE_PROXY_PORT, TESTS_ZEND_SERVICE_WINDOWSAZURE_STORAGE_PROXY_CREDENTIALS);
  76. }
  77. return $storageClient;
  78. }
  79. protected static $uniqId = 0;
  80. protected function generateName()
  81. {
  82. self::$uniqId++;
  83. return TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_TABLENAME_PREFIX . self::$uniqId;
  84. }
  85. /**
  86. * Test create table
  87. */
  88. public function testCreateTable()
  89. {
  90. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  91. $tableName = $this->generateName();
  92. $storageClient = $this->createStorageInstance();
  93. $result = $storageClient->createTable($tableName);
  94. $this->assertEquals($tableName, $result->Name);
  95. $result = $storageClient->listTables();
  96. $this->assertEquals(1, count($result));
  97. $this->assertEquals($tableName, $result[0]->Name);
  98. }
  99. }
  100. /**
  101. * Test table exists
  102. */
  103. public function testTableExists()
  104. {
  105. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  106. $tableName1 = $this->generateName();
  107. $tableName2 = $this->generateName();
  108. $storageClient = $this->createStorageInstance();
  109. $storageClient->createTable($tableName1);
  110. $storageClient->createTable($tableName2);
  111. $result = $storageClient->tableExists($tableName2);
  112. $this->assertTrue($result);
  113. $result = $storageClient->tableExists(md5(time()));
  114. $this->assertFalse($result);
  115. }
  116. }
  117. /**
  118. * Test list tables
  119. */
  120. public function testListTables()
  121. {
  122. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  123. $tableName1 = $this->generateName();
  124. $tableName2 = $this->generateName();
  125. $storageClient = $this->createStorageInstance();
  126. $storageClient->createTable($tableName1);
  127. $storageClient->createTable($tableName2);
  128. $result = $storageClient->listTables();
  129. $this->assertEquals(2, count($result));
  130. $this->assertEquals($tableName1, $result[0]->Name);
  131. $this->assertEquals($tableName2, $result[1]->Name);
  132. }
  133. }
  134. /**
  135. * Test delete table
  136. */
  137. public function testDeleteTable()
  138. {
  139. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  140. $tableName = $this->generateName();
  141. $storageClient = $this->createStorageInstance();
  142. $storageClient->createTable($tableName);
  143. $storageClient->deleteTable($tableName);
  144. $result = $storageClient->listTables();
  145. $this->assertEquals(0, count($result));
  146. }
  147. }
  148. /**
  149. * Test insert entity
  150. */
  151. public function testInsertEntity()
  152. {
  153. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  154. $tableName = $this->generateName();
  155. $storageClient = $this->createStorageInstance();
  156. $storageClient->createTable($tableName);
  157. $entities = $this->_generateEntities(1);
  158. $entity = $entities[0];
  159. $result = $storageClient->insertEntity($tableName, $entity);
  160. $this->assertNotEquals('0001-01-01T00:00:00', $result->getTimestamp());
  161. $this->assertNotEquals('', $result->getEtag());
  162. $this->assertEquals($entity, $result);
  163. }
  164. }
  165. /**
  166. * Test delete entity, not taking etag into account
  167. */
  168. public function testDeleteEntity_NoEtag()
  169. {
  170. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  171. $tableName = $this->generateName();
  172. $storageClient = $this->createStorageInstance();
  173. $storageClient->createTable($tableName);
  174. $entities = $this->_generateEntities(1);
  175. $entity = $entities[0];
  176. $result = $storageClient->insertEntity($tableName, $entity);
  177. $this->assertEquals($entity, $result);
  178. $storageClient->deleteEntity($tableName, $entity);
  179. }
  180. }
  181. /**
  182. * Test delete entity, taking etag into account
  183. */
  184. public function testDeleteEntity_Etag()
  185. {
  186. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  187. $tableName = $this->generateName();
  188. $storageClient = $this->createStorageInstance();
  189. $storageClient->createTable($tableName);
  190. $entities = $this->_generateEntities(1);
  191. $entity = $entities[0];
  192. $result = $storageClient->insertEntity($tableName, $entity);
  193. $this->assertEquals($entity, $result);
  194. // Set "old" etag
  195. $entity->setEtag('W/"datetime\'2009-05-27T12%3A15%3A15.3321531Z\'"');
  196. $exceptionThrown = false;
  197. try {
  198. $storageClient->deleteEntity($tableName, $entity, true);
  199. } catch (Exception $ex) {
  200. $exceptionThrown = true;
  201. }
  202. $this->assertTrue($exceptionThrown);
  203. }
  204. }
  205. /**
  206. * Test retrieve entity by id
  207. */
  208. public function testRetrieveEntityById()
  209. {
  210. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  211. $tableName = $this->generateName();
  212. $storageClient = $this->createStorageInstance();
  213. $storageClient->createTable($tableName);
  214. $entities = $this->_generateEntities(1);
  215. $entity = $entities[0];
  216. $storageClient->insertEntity($tableName, $entity);
  217. $result = $storageClient->retrieveEntityById($tableName, $entity->getPartitionKey(), $entity->getRowKey(), 'TSTest_TestEntity');
  218. $this->assertEquals($entity, $result);
  219. }
  220. }
  221. /**
  222. * Test retrieve entity by id (> 256 key characters)
  223. */
  224. public function testRetrieveEntityById_Large()
  225. {
  226. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  227. $tableName = $this->generateName();
  228. $storageClient = $this->createStorageInstance();
  229. $storageClient->createTable($tableName);
  230. $entities = $this->_generateEntities(1);
  231. $entity = $entities[0];
  232. $entity->setPartitionKey(str_repeat('a', 200));
  233. $entity->setRowKey(str_repeat('a', 200));
  234. $storageClient->insertEntity($tableName, $entity);
  235. $result = $storageClient->retrieveEntityById($tableName, $entity->getPartitionKey(), $entity->getRowKey(), 'TSTest_TestEntity');
  236. $this->assertEquals($entity, $result);
  237. }
  238. }
  239. /**
  240. * Test retrieve entity by id, DynamicTableEntity
  241. */
  242. public function testRetrieveEntityById_DynamicTableEntity()
  243. {
  244. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  245. $tableName = $this->generateName();
  246. $storageClient = $this->createStorageInstance();
  247. $storageClient->createTable($tableName);
  248. $entities = $this->_generateEntities(1);
  249. $entity = $entities[0];
  250. $storageClient->insertEntity($tableName, $entity);
  251. $result = $storageClient->retrieveEntityById($tableName, $entity->getPartitionKey(), $entity->getRowKey());
  252. $this->assertEquals($entity->FullName, $result->Name);
  253. $this->assertType('Zend_Service_WindowsAzure_Storage_DynamicTableEntity', $result);
  254. }
  255. }
  256. /**
  257. * Test update entity, not taking etag into account
  258. */
  259. public function testUpdateEntity_NoEtag()
  260. {
  261. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  262. $tableName = $this->generateName();
  263. $storageClient = $this->createStorageInstance();
  264. $storageClient->createTable($tableName);
  265. $entities = $this->_generateEntities(1);
  266. $entity = $entities[0];
  267. $storageClient->insertEntity($tableName, $entity);
  268. $entity->Age = 0;
  269. $result = $storageClient->updateEntity($tableName, $entity);
  270. $this->assertNotEquals('0001-01-01T00:00:00', $result->getTimestamp());
  271. $this->assertNotEquals('', $result->getEtag());
  272. $this->assertEquals(0, $result->Age);
  273. $this->assertEquals($entity, $result);
  274. }
  275. }
  276. /**
  277. * Test update entity, taking etag into account
  278. */
  279. public function testUpdateEntity_Etag()
  280. {
  281. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  282. $tableName = $this->generateName();
  283. $storageClient = $this->createStorageInstance();
  284. $storageClient->createTable($tableName);
  285. $entities = $this->_generateEntities(1);
  286. $entity = $entities[0];
  287. $storageClient->insertEntity($tableName, $entity);
  288. $entity->Age = 0;
  289. // Set "old" etag
  290. $entity->setEtag('W/"datetime\'2009-05-27T12%3A15%3A15.3321531Z\'"');
  291. $exceptionThrown = false;
  292. try {
  293. $storageClient->updateEntity($tableName, $entity, true);
  294. } catch (Exception $ex) {
  295. $exceptionThrown = true;
  296. }
  297. $this->assertTrue($exceptionThrown);
  298. }
  299. }
  300. /**
  301. * Test merge entity, not taking etag into account
  302. */
  303. public function testMergeEntity_NoEtag()
  304. {
  305. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  306. $tableName = $this->generateName();
  307. $storageClient = $this->createStorageInstance();
  308. $storageClient->createTable($tableName);
  309. $entities = $this->_generateEntities(1);
  310. $entity = $entities[0];
  311. $storageClient->insertEntity($tableName, $entity);
  312. $dynamicEntity = new Zend_Service_WindowsAzure_Storage_DynamicTableEntity($entity->getPartitionKey(), $entity->getRowKey());
  313. $dynamicEntity->Myproperty = 10;
  314. $dynamicEntity->Otherproperty = "Test";
  315. $dynamicEntity->Age = 0;
  316. $storageClient->mergeEntity($tableName, $dynamicEntity, false, array('Myproperty', 'Otherproperty')); // only update 'Myproperty' and 'Otherproperty'
  317. $result = $storageClient->retrieveEntityById($tableName, $entity->getPartitionKey(), $entity->getRowKey());
  318. $this->assertNotEquals('0001-01-01T00:00:00', $result->getTimestamp());
  319. $this->assertNotEquals('', $result->getEtag());
  320. $this->assertNotEquals(0, $result->Age);
  321. $this->assertEquals($entity->FullName, $result->Name);
  322. $this->assertEquals($dynamicEntity->Myproperty, $result->Myproperty);
  323. $this->assertEquals($dynamicEntity->Otherproperty, $result->Otherproperty);
  324. }
  325. }
  326. /**
  327. * Test merge entity, taking etag into account
  328. */
  329. public function testMergeEntity_Etag()
  330. {
  331. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  332. $tableName = $this->generateName();
  333. $storageClient = $this->createStorageInstance();
  334. $storageClient->createTable($tableName);
  335. $entities = $this->_generateEntities(1);
  336. $entity = $entities[0];
  337. $storageClient->insertEntity($tableName, $entity);
  338. $dynamicEntity = new Zend_Service_WindowsAzure_Storage_DynamicTableEntity($entity->getPartitionKey(), $entity->getRowKey());
  339. $dynamicEntity->Myproperty = 10;
  340. $dynamicEntity->Otherproperty = "Test";
  341. $dynamicEntity->Age = 0;
  342. // Set "old" etag
  343. $entity->setEtag('W/"datetime\'2009-05-27T12%3A15%3A15.3321531Z\'"');
  344. $exceptionThrown = false;
  345. try {
  346. $storageClient->mergeEntity($tableName, $dynamicEntity, true);
  347. } catch (Exception $ex) {
  348. $exceptionThrown = true;
  349. }
  350. $this->assertTrue($exceptionThrown);
  351. }
  352. }
  353. /**
  354. * Test retrieve entities, all
  355. */
  356. public function testRetrieveEntities_All()
  357. {
  358. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  359. $tableName = $this->generateName();
  360. $storageClient = $this->createStorageInstance();
  361. $storageClient->createTable($tableName);
  362. $entities = $this->_generateEntities(20);
  363. foreach ($entities as $entity)
  364. {
  365. $storageClient->insertEntity($tableName, $entity);
  366. }
  367. $result = $storageClient->retrieveEntities($tableName, 'TSTest_TestEntity');
  368. $this->assertEquals(20, count($result));
  369. }
  370. }
  371. /**
  372. * Test retrieve entities, all, DynamicTableEntity
  373. */
  374. public function testRetrieveEntities_All_DynamicTableEntity()
  375. {
  376. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  377. $tableName = $this->generateName();
  378. $storageClient = $this->createStorageInstance();
  379. $storageClient->createTable($tableName);
  380. $entities = $this->_generateEntities(20);
  381. foreach ($entities as $entity)
  382. {
  383. $storageClient->insertEntity($tableName, $entity);
  384. }
  385. $result = $storageClient->retrieveEntities($tableName);
  386. $this->assertEquals(20, count($result));
  387. foreach ($result as $item)
  388. {
  389. $this->assertType('Zend_Service_WindowsAzure_Storage_DynamicTableEntity', $item);
  390. }
  391. }
  392. }
  393. /**
  394. * Test retrieve entities, filtered
  395. */
  396. public function testRetrieveEntities_Filtered()
  397. {
  398. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  399. $tableName = $this->generateName();
  400. $storageClient = $this->createStorageInstance();
  401. $storageClient->createTable($tableName);
  402. $entities = $this->_generateEntities(5);
  403. foreach ($entities as $entity)
  404. {
  405. $storageClient->insertEntity($tableName, $entity);
  406. }
  407. $result = $storageClient->retrieveEntities($tableName, 'PartitionKey eq \'' . $entities[0]->getPartitionKey() . '\' and RowKey eq \'' . $entities[0]->getRowKey() . '\'', 'TSTest_TestEntity');
  408. $this->assertEquals(1, count($result));
  409. }
  410. }
  411. /**
  412. * Test retrieve entities, fluent interface
  413. */
  414. public function testRetrieveEntities_Fluent1()
  415. {
  416. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  417. $tableName = $this->generateName();
  418. $storageClient = $this->createStorageInstance();
  419. $storageClient->createTable($tableName);
  420. $entities = $this->_generateEntities(10);
  421. foreach ($entities as $entity)
  422. {
  423. $storageClient->insertEntity($tableName, $entity);
  424. }
  425. $result = $storageClient->retrieveEntities(
  426. $storageClient->select()
  427. ->from($tableName)
  428. ->where('Name eq ?', $entities[0]->FullName)
  429. ->andWhere('RowKey eq ?', $entities[0]->getRowKey()),
  430. 'TSTest_TestEntity'
  431. );
  432. $this->assertEquals(1, count($result));
  433. $this->assertEquals($entities[0], $result[0]);
  434. }
  435. }
  436. /**
  437. * Test retrieve entities, fluent interface
  438. */
  439. public function testRetrieveEntities_Fluent2()
  440. {
  441. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  442. $tableName = $this->generateName();
  443. $storageClient = $this->createStorageInstance();
  444. $storageClient->createTable($tableName);
  445. $entities = $this->_generateEntities(10);
  446. foreach ($entities as $entity)
  447. {
  448. $storageClient->insertEntity($tableName, $entity);
  449. }
  450. $result = $storageClient->retrieveEntities(
  451. $storageClient->select()
  452. ->from($tableName)
  453. ->where('Name eq ?', $entities[0]->FullName)
  454. ->andWhere('PartitionKey eq ?', $entities[0]->getPartitionKey()),
  455. 'TSTest_TestEntity'
  456. );
  457. $this->assertEquals(1, count($result));
  458. $this->assertEquals($entities[0], $result[0]);
  459. }
  460. }
  461. /**
  462. * Test retrieve entities, fluent interface, top specification
  463. */
  464. public function testRetrieveEntities_Fluent_Top()
  465. {
  466. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  467. $tableName = $this->generateName();
  468. $storageClient = $this->createStorageInstance();
  469. $storageClient->createTable($tableName);
  470. $entities = $this->_generateEntities(10);
  471. foreach ($entities as $entity)
  472. {
  473. $storageClient->insertEntity($tableName, $entity);
  474. }
  475. $result = $storageClient->retrieveEntities(
  476. $storageClient->select()->top(4)
  477. ->from($tableName),
  478. 'TSTest_TestEntity'
  479. );
  480. $this->assertEquals(4, count($result));
  481. }
  482. }
  483. /**
  484. * Test batch commit, success
  485. */
  486. public function testBatchCommit_Success()
  487. {
  488. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  489. $tableName = $this->generateName();
  490. $storageClient = $this->createStorageInstance();
  491. $storageClient->createTable($tableName);
  492. $entities = $this->_generateEntities(20);
  493. $entities1 = array_slice($entities, 0, 10);
  494. $entities2 = array_slice($entities, 10, 10);
  495. // Insert entities
  496. foreach ($entities1 as $entity)
  497. {
  498. $storageClient->insertEntity($tableName, $entity);
  499. }
  500. // Start batch
  501. $batch = $storageClient->startBatch();
  502. $this->assertType('Zend_Service_WindowsAzure_Storage_Batch', $batch);
  503. // Insert entities in batch
  504. foreach ($entities2 as $entity)
  505. {
  506. $storageClient->insertEntity($tableName, $entity);
  507. }
  508. // Delete entities
  509. foreach ($entities1 as $entity)
  510. {
  511. $storageClient->deleteEntity($tableName, $entity);
  512. }
  513. // Commit
  514. $batch->commit();
  515. // Verify
  516. $result = $storageClient->retrieveEntities($tableName);
  517. $this->assertEquals(10, count($result));
  518. }
  519. }
  520. /**
  521. * Test batch rollback, success
  522. */
  523. public function testBatchRollback_Success()
  524. {
  525. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  526. $tableName = $this->generateName();
  527. $storageClient = $this->createStorageInstance();
  528. $storageClient->createTable($tableName);
  529. $entities = $this->_generateEntities(10);
  530. // Start batch
  531. $batch = $storageClient->startBatch();
  532. $this->assertType('Zend_Service_WindowsAzure_Storage_Batch', $batch);
  533. // Insert entities in batch
  534. foreach ($entities as $entity)
  535. {
  536. $storageClient->insertEntity($tableName, $entity);
  537. }
  538. // Rollback
  539. $batch->rollback();
  540. // Verify
  541. $result = $storageClient->retrieveEntities($tableName);
  542. $this->assertEquals(0, count($result));
  543. }
  544. }
  545. /**
  546. * Test batch commit, fail updates
  547. */
  548. public function testBatchCommit_FailUpdates()
  549. {
  550. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  551. $tableName = $this->generateName();
  552. $storageClient = $this->createStorageInstance();
  553. $storageClient->createTable($tableName);
  554. $entities = $this->_generateEntities(10);
  555. foreach ($entities as $entity)
  556. {
  557. $storageClient->insertEntity($tableName, $entity);
  558. }
  559. // Make some entity updates with "old" etags
  560. $entities[0]->Age = 0;
  561. $entities[0]->setEtag('W/"datetime\'2009-05-27T12%3A15%3A15.3321531Z\'"');
  562. $entities[1]->Age = 0;
  563. $entities[1]->setEtag('W/"datetime\'2009-05-27T12%3A15%3A15.3321531Z\'"');
  564. $entities[2]->Age = 0;
  565. // Start batch
  566. $batch = $storageClient->startBatch();
  567. $this->assertType('Zend_Service_WindowsAzure_Storage_Batch', $batch);
  568. // Update entities in batch
  569. $storageClient->updateEntity($tableName, $entities[0], true);
  570. $storageClient->updateEntity($tableName, $entities[1], true);
  571. $storageClient->updateEntity($tableName, $entities[2], true);
  572. // Commit
  573. $exceptionThrown = false;
  574. try {
  575. $batch->commit();
  576. } catch (Exception $ex) {
  577. $exceptionThrown = true;
  578. }
  579. $this->assertTrue($exceptionThrown);
  580. }
  581. }
  582. /**
  583. * Test batch commit, fail partition
  584. */
  585. public function testBatchCommit_FailPartition()
  586. {
  587. if (TESTS_ZEND_SERVICE_WINDOWSAZURE_TABLE_RUNTESTS) {
  588. $tableName = $this->generateName();
  589. $storageClient = $this->createStorageInstance();
  590. $storageClient->createTable($tableName);
  591. $entities = $this->_generateEntities(10);
  592. // Start batch
  593. $batch = $storageClient->startBatch();
  594. $this->assertType('Zend_Service_WindowsAzure_Storage_Batch', $batch);
  595. // Insert entities in batch
  596. foreach ($entities as $entity)
  597. {
  598. $entity->setPartitionKey('partition' . rand(1, 9));
  599. $storageClient->insertEntity($tableName, $entity);
  600. }
  601. // Commit
  602. $exceptionThrown = false;
  603. try {
  604. $batch->commit();
  605. } catch (Exception $ex) {
  606. $exceptionThrown = true;
  607. }
  608. $this->assertTrue($exceptionThrown);
  609. // Verify
  610. $result = $storageClient->retrieveEntities($tableName);
  611. $this->assertEquals(0, count($result));
  612. }
  613. }
  614. /**
  615. * Generate entities
  616. *
  617. * @param int $amount Number of entities to generate
  618. * @return array Array of TSTest_TestEntity
  619. */
  620. protected function _generateEntities($amount = 1)
  621. {
  622. $returnValue = array();
  623. for ($i = 0; $i < $amount; $i++)
  624. {
  625. $entity = new TSTest_TestEntity('partition1', 'row' . ($i + 1));
  626. $entity->FullName = md5(uniqid(rand(), true));
  627. $entity->Age = rand(1, 130);
  628. $entity->Visible = rand(1,2) == 1;
  629. $returnValue[] = $entity;
  630. }
  631. return $returnValue;
  632. }
  633. }
  634. /**
  635. * Test Zend_Service_WindowsAzure_Storage_TableEntity class
  636. */
  637. class TSTest_TestEntity extends Zend_Service_WindowsAzure_Storage_TableEntity
  638. {
  639. /**
  640. * @azure Name
  641. */
  642. public $FullName;
  643. /**
  644. * @azure Age Edm.Int64
  645. */
  646. public $Age;
  647. /**
  648. * @azure Visible Edm.Boolean
  649. */
  650. public $Visible = false;
  651. }
  652. // Call Zend_Service_WindowsAzure_TableStorageTest::main() if this source file is executed directly.
  653. if (PHPUnit_MAIN_METHOD == "Zend_Service_WindowsAzure_TableStorageTest::main") {
  654. Zend_Service_WindowsAzure_TableStorageTest::main();
  655. }