public function testRemoveByEntity() { $entity = $this->createEntities(); $query = $this->getMockBuilder('Doctrine\\ORM\\AbstractQuery')->disableOriginalConstructor()->setMethods(['execute'])->getMockForAbstractClass(); $queryBuilder = $this->mockGetSnapshotDeleteQueryBuilderByEntitiesFunction($entity); $queryBuilder->expects($this->exactly(2))->method('getQuery')->will($this->returnValue($query)); $this->repository->removeByEntity(reset($entity)); }