/** * testGetTotalRecords */ public function testGetTotalRecords() { $expectedResults = 2; Phake::when($this->queryBuilder)->getEntityManager()->thenReturn($this->em); Phake::when($this->queryBuilder)->getQuery()->thenReturn($this->query); Phake::when($this->em)->createNativeQuery(Phake::anyParameters())->thenReturn($this->query); Phake::when($this->queryBuilder)->getParameters()->thenReturn($this->arrayCollection); Phake::when($this->queryBuilder)->select(Phake::anyParameters())->thenReturn($this->queryBuilder); Phake::when($this->query)->getSingleResult()->thenReturn(array('result' => $expectedResults)); Phake::when($this->arrayCollection)->toArray()->thenReturn(array()); $result = $this->service->getTotalRecords($this->queryBuilder, 's'); Phake::verify($this->query)->getSingleResult(); $this->assertEquals($expectedResults, $result); }