Laravel 5.1 - Using DatabaseTransactions with custom connections on Testing

DatabaseTransactions.php

<?php namespace JaTubio\Testing;

trait DatabaseTransactions {

	protected function getConnectionName()
	{
		$model = $this->getModel();

		if ( null !== $model )
		{
			return $model->getConnectionName();
		}

		return null;
	}

	/**
	 * @return \Portal\Entities\Model
	 */
	protected function getModel()
	{
		return null;
	}

	/**
	 * @before
	 */
	public function beginDatabaseTransaction()
	{
		$this->app->make('db')->setDefaultConnection($this->getConnectionName());

		$this->app->make('db')->beginTransaction();

		$this->beforeApplicationDestroyed(function ()
		{
			$this->app->make('db')->rollBack();
		});
	}
}

Use samples:

<?php

use JaTubio\Testing\DatabaseTransactions;

class ATestUsingDatabases extends TestCase {

    use DatabaseTransactions;

    protected function getModel()
    {
        return new App\Models\MyModel();
    }
}

<?php

use JaTubio\Testing\DatabaseTransactions;

class ATestUsingDatabases extends TestCase {

    use DatabaseTransactions;

    protected function getConnectionName()
    {
        return 'customconnection';
    }
}