public function execute()
{
if (!count($this->condition)) {
$total_rows = $this->connection->query('SELECT COUNT(*) FROM {' . $this->connection->escapeTable($this->table) . '}')->fetchField();
parent::execute();
return $total_rows;
} else {
return parent::execute();
}
}
/**
* {@inheritdoc}
*/
protected function setUp()
{
parent::setUp();
// Create a Mock database connection object.
$this->connection = $this->getMockBuilder('Drupal\\Core\\Database\\Connection')->disableOriginalConstructor()->getMock();
// Create a Mock Statement object
$this->statement = $this->getMockBuilder('Drupal\\Core\\Database\\Driver\\sqlite\\Statement')->disableOriginalConstructor()->getMock();
// Create a Mock Select object and set expectations.
$this->select = $this->getMockBuilder('Drupal\\Core\\Database\\Query\\Select')->disableOriginalConstructor()->getMock();
$this->select->expects($this->any())->method('fields')->will($this->returnSelf());
$this->select->expects($this->any())->method('condition')->will($this->returnSelf());
$this->select->expects($this->any())->method('range')->will($this->returnSelf());
$this->select->expects($this->any())->method('orderBy')->will($this->returnSelf());
$this->select->expects($this->any())->method('execute')->will($this->returnValue($this->statement));
$this->connection->expects($this->any())->method('select')->will($this->returnValue($this->select));
// Create a Mock Delete object and set expectations.
$this->delete = $this->getMockBuilder('Drupal\\Core\\Database\\Query\\Delete')->disableOriginalConstructor()->getMock();
$this->delete->expects($this->any())->method('condition')->will($this->returnSelf());
$this->delete->expects($this->any())->method('execute')->will($this->returnValue($this->statement));
}
/**
* {@inheritdoc}
*/
public function execute()
{
$this->connection->addSavepoint();
try {
$result = parent::execute();
} catch (\Exception $e) {
$this->connection->rollbackSavepoint();
throw $e;
}
$this->connection->releaseSavepoint();
return $result;
}