52 lines
1.2 KiB
PHP
52 lines
1.2 KiB
PHP
<?php
|
|
|
|
namespace Pterodactyl\Tests\Traits;
|
|
|
|
use PDO;
|
|
use Mockery;
|
|
use Illuminate\Database\Eloquent\Model;
|
|
use Illuminate\Database\MySqlConnection;
|
|
use Illuminate\Database\ConnectionResolver;
|
|
|
|
trait MocksPdoConnection
|
|
{
|
|
/**
|
|
* @var \Illuminate\Database\ConnectionResolverInterface|null
|
|
*/
|
|
private static $initialResolver;
|
|
|
|
/**
|
|
* Generates a mock PDO connection and injects it into the models so that any actual
|
|
* DB call can be properly intercepted.
|
|
*
|
|
* @return \Mockery\MockInterface
|
|
*/
|
|
protected function mockPdoConnection()
|
|
{
|
|
self::$initialResolver = Model::getConnectionResolver();
|
|
|
|
Model::unsetConnectionResolver();
|
|
|
|
$connection = new MySqlConnection($mock = Mockery::mock(PDO::class), 'testing_mock');
|
|
$resolver = new ConnectionResolver(['mocked' => $connection]);
|
|
$resolver->setDefaultConnection('mocked');
|
|
|
|
Model::setConnectionResolver($resolver);
|
|
|
|
return $mock;
|
|
}
|
|
|
|
/**
|
|
* Resets the mock state.
|
|
*/
|
|
protected function tearDownPdoMock()
|
|
{
|
|
if (! self::$initialResolver) {
|
|
return;
|
|
}
|
|
|
|
Model::setConnectionResolver(self::$initialResolver);
|
|
|
|
self::$initialResolver = null;
|
|
}
|
|
}
|