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; }