public function testGetEmptyList() { $businessUnitRepository = $this->getMockBuilder('BusinessUnitRepository')->disableOriginalConstructor()->setMethods(['getQueryBuilder'])->getMock(); $qb = $this->getMockBuilder('Doctrine\\ORM\\QueryBuilder')->setMethods(['getResult'])->disableOriginalConstructor()->getMock(); $this->aclHelper->expects($this->any())->method('apply')->willReturn($qb); $businessUnitRepository->expects($this->any())->method('getQueryBuilder')->willReturn($qb); $qb->expects($this->any())->method('getResult')->willReturn([]); $this->registry->expects($this->once())->method('getRepository')->with('OroOrganizationBundle:BusinessUnit')->willReturn($businessUnitRepository); $result = $this->choiceTreeBusinessUnitProvider->getList(); $this->assertEquals([], $result); }
/** * @dataProvider getListDataProvider */ public function testGetList($userBUIds, $queryResult, $result) { $businessUnitRepos = $this->getMockBuilder('BusinessUnitRepository')->disableOriginalConstructor()->setMethods(['getQueryBuilder'])->getMock(); $qb = $this->getMockBuilder('Doctrine\\ORM\\QueryBuilder')->setMethods(['getResult', 'expr', 'setParameter'])->disableOriginalConstructor()->getMock(); $treeOwner = $this->getMockBuilder('Oro\\Bundle\\SecurityBundle\\Owner\\OwnerTree')->setMethods(['getUserSubordinateBusinessUnitIds'])->disableOriginalConstructor()->getMock(); $this->treeProvider->expects(self::once())->method('getTree')->willReturn($treeOwner); $treeOwner->expects(self::once())->method('getUserSubordinateBusinessUnitIds')->willReturn($userBUIds); $this->aclHelper->expects(self::any())->method('apply')->willReturn($qb); $businessUnitRepos->expects(self::any())->method('getQueryBuilder')->willReturn($qb); $expression = $this->getMockBuilder('Doctrine\\ORM\\Query\\Expr')->setMethods(['in'])->disableOriginalConstructor()->getMock(); $qb->expects(self::once())->method('getResult')->willReturn($queryResult); $qb->expects(self::once())->method('expr')->willReturn($expression); $qb->expects(self::once())->method('setParameter'); $tokenStorage = $this->getMock('Symfony\\Component\\Security\\Core\\Authentication\\Token\\TokenInterface'); $user = $this->getMockBuilder('Oro\\Bundle\\UserBundle\\Entity\\User')->setMethods(['getId', 'getOrganizations'])->disableOriginalConstructor()->getMock(); $organization = $this->getMock('Oro\\Bundle\\OrganizationBundle\\Entity\\OrganizationInterface'); $organization->expects(self::once())->method('getId')->willReturn(1); $user->expects(self::once())->method('getOrganizations')->willReturn([$organization]); $this->securityFacade->expects(self::once())->method('getToken')->willReturn($tokenStorage); $tokenStorage->expects(self::once())->method('getUser')->willReturn($user); $this->registry->expects(self::once())->method('getRepository')->with('OroOrganizationBundle:BusinessUnit')->willReturn($businessUnitRepos); $resultedUserBUids = $this->choiceTreeBUProvider->getList(); self::assertEquals($result, $resultedUserBUids); }