szurubooru/tests/Dao/TransactionManagerTest.php

<?php
namespace Szurubooru\Tests\Dao;
use Szurubooru\Dao\TokenDao;
use Szurubooru\Dao\TransactionManager;
use Szurubooru\Entities\Token;
use Szurubooru\Tests\AbstractDatabaseTestCase;

final class TransactionManagerTest extends AbstractDatabaseTestCase
{
	public function testCommit()
	{
		$testEntity = $this->getTestEntity();
		$testDao = $this->getTestDao();

		$transactionManager = $this->getTransactionManager();
		$transactionManager->commit(function() use ($testDao, &$testEntity)
		{
			$testDao->save($testEntity);
			$this->assertNotNull($testEntity->getId());
		});

		$this->assertNotNull($testEntity->getId());
		$this->assertEntitiesEqual($testEntity, $testDao->findById($testEntity->getId()));
	}

	public function testRollback()
	{
		$testEntity = $this->getTestEntity();
		$testDao = $this->getTestDao();

		$transactionManager = $this->getTransactionManager();
		$transactionManager->rollback(function() use ($testDao, &$testEntity)
		{
			$testDao->save($testEntity);
			$this->assertNotNull($testEntity->getId());
		});

		//ids that could be forged in transaction get left behind after rollback
		$this->assertNotNull($testEntity->getId());

		$this->databaseConnection->getPDO()->rollBack();
		$this->databaseConnection->getPDO()->beginTransaction();

		//but entities shouldn't be saved to database
		$this->assertNull($testDao->findById($testEntity->getId()));
	}

	public function testNestedTransactions()
	{
		$testEntity = $this->getTestEntity();
		$testDao = $this->getTestDao();

		$transactionManager = $this->getTransactionManager();
		$transactionManager->commit(function() use ($transactionManager, $testDao, &$testEntity)
		{
			$transactionManager->commit(function() use ($testDao, &$testEntity)
			{
				$testDao->save($testEntity);
				$this->assertNotNull($testEntity->getId());
			});
		});

		$this->assertNotNull($testEntity->getId());
		$this->assertEntitiesEqual($testEntity, $testDao->findById($testEntity->getId()));
	}

	private function getTestEntity()
	{
		$token = new Token();
		$token->setName('yo');
		$token->setPurpose(Token::PURPOSE_ACTIVATE);
		return $token;
	}

	private function getTestDao()
	{
		return new TokenDao($this->databaseConnection);
	}

	private function getTransactionManager()
	{
		return new TransactionManager($this->databaseConnection);
	}
}