Question

When running unit tests, I am using a test database (so I don’t touch the production one). How can I clean the test database after unit tests are finished?

I have the entity manager object. Is there some method to delete all rows from all tables?

Answer 1

It was actually pretty easy. I have created a test listener:

<?php

class TestDbCleanupListener implements PHPUnit_Framework_TestListener
{

    private $_em;

    private function _getEntityManager()
    {
        if (null === $this->_em) {
            $this->_em = Bootstrap::getServiceManager()->get('doctrine.entitymanager.orm_default');
        }
        return $this->_em;
    }

    /**
     * called when test is started - starts transaction
     * (non-PHPdoc)
     * @see PHPUnit_Framework_TestListener::startTest()
     */
    public function startTest(PHPUnit_Framework_Test $test)
    {
        $this->_getEntityManager()->beginTransaction();
    }

    /**
     * called when test is ended - rolls back the transaction 
     * @param PHUnit_Framework_Test $test
     * @param float $length the length of time for the test
     */
    public function endTest(PHPUnit_Framework_Test $test, $length)
    {
        $this->_getEntityManager()->rollback();
    }

    /**
     * Required for Interface
     * (non-PHPdoc)
     * @see PHPUnit_Framework_TestListener::addError()
     */
    public function addError(PHPUnit_Framework_Test $test, Exception $e, $time)
    {

    }

    /**
     * Required for Interface
     * (non-PHPdoc)
     * @see PHPUnit_Framework_TestListener::addFailure()
     */
    public function addFailure(PHPUnit_Framework_Test $test, PHPUnit_Framework_AssertionFailedError $e, $time)
    {

    }

    /**
     * Required for Interface
     * (non-PHPdoc)
     * @see PHPUnit_Framework_TestListener::addError()
     */
    public function addIncompleteTest(PHPUnit_Framework_Test $test, Exception $e, $time)
    {

    }

    /**
     * Required for Interface
     * (non-PHPdoc)
     * @see PHPUnit_Framework_TestListener::addSkippedTest()
     */
    public function addSkippedTest(PHPUnit_Framework_Test $test, Exception $e, $time)
    {

    }

    /**
     * Required for Interface
     * (non-PHPdoc)
     * @see PHPUnit_Framework_TestListener::startTestSuite()
     */
    public function startTestSuite(PHPUnit_Framework_TestSuite $suite)
    {

    }

    /**
     * Required for Interface
     * (non-PHPdoc)
     * @see PHPUnit_Framework_TestListener::endTestSuite()
     */
    public function endTestSuite(PHPUnit_Framework_TestSuite $suite)
    {

    }

}