/** * Sets up required mocks and the CommentStatistics service under test. */ protected function setUp() { $this->statement = $this->getMockBuilder('Drupal\\Core\\Database\\Driver\\sqlite\\Statement')->disableOriginalConstructor()->getMock(); $this->statement->expects($this->any())->method('fetchObject')->will($this->returnCallback(array($this, 'fetchObjectCallback'))); $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('execute')->will($this->returnValue($this->statement)); $this->database = $this->getMockBuilder('Drupal\\Core\\Database\\Connection')->disableOriginalConstructor()->getMock(); $this->database->expects($this->once())->method('select')->will($this->returnValue($this->select)); $this->commentStatistics = new CommentStatistics($this->database, $this->getMock('Drupal\\Core\\Session\\AccountInterface'), $this->getMock('Drupal\\Core\\Entity\\EntityManagerInterface'), $this->getMock('Drupal\\Core\\State\\StateInterface')); }
/** * {@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)); }