function let(DocumentManager $manager, ChannelInterface $ecommerce, ChannelInterface $mobile, LocaleInterface $enUs, LocaleInterface $frFr, CategoryInterface $category, ChannelManager $channelManager, CategoryRepositoryInterface $categoryRepository, ProductRepository $productRepository, QueryBuilder $ormQb, Builder $odmQb, Query $odmQuery, Cursor $cursor)
{
$enUs->getCode()->willReturn('en_US');
$frFr->getCode()->willReturn('fr_FR');
$ecommerce->getCode()->willReturn('ecommerce');
$ecommerce->getLabel()->willReturn('ECommerce');
$ecommerce->getLocales()->willReturn(array($enUs, $frFr));
$ecommerce->getCategory()->willReturn($category);
$mobile->getCode()->willReturn('mobile');
$mobile->getLabel()->willReturn('Mobile');
$mobile->getLocales()->willReturn(array($enUs));
$mobile->getCategory()->willReturn($category);
$odmQuery->execute()->willReturn($cursor);
$productRepository->createQueryBuilder()->willReturn($odmQb);
$odmQb->hydrate(Argument::any())->willReturn($odmQb);
$odmQb->field(Argument::any())->willReturn($odmQb);
$odmQb->in(Argument::any())->willReturn($odmQb);
$odmQb->equals(Argument::any())->willReturn($odmQb);
$odmQb->select('_id')->willReturn($odmQb);
$odmQb->getQuery()->willReturn($odmQuery);
$categoryRepository->getAllChildrenQueryBuilder($category, true)->willReturn($ormQb);
$categoryRepository->getCategoryIds($category, $ormQb)->willReturn(array(1, 2, 3));
$channelManager->getFullChannels()->willReturn(array($ecommerce, $mobile));
$manager->getRepository('pim_product_class')->willReturn($productRepository);
$this->beConstructedWith($manager, $channelManager, $categoryRepository, 'pim_product_class');
}
/**
* Get product ids linked to a category or its children.
* You can define if you just want to get the property of the actual node or with its children with the direct
* parameter
*
* @param CategoryInterface $category the requested node
* @param boolean $inChildren true to take children not into account
*
* @return array
*/
public function getProductIdsInCategory(CategoryInterface $category, $inChildren = false)
{
$categoryQb = null;
if ($inChildren) {
$categoryQb = $this->categoryRepository->getAllChildrenQueryBuilder($category, true);
}
return $this->productRepository->getProductIdsInCategory($category, $categoryQb);
}
/**
* {@inheritdoc}
*/
public function buildByChannelAndCompleteness(ChannelInterface $channel)
{
$qb = $this->createQueryBuilder('p');
foreach ($channel->getLocales() as $locale) {
$qb->addOr($qb->expr()->field(sprintf('normalizedData.completenesses.%s-%s', $channel->getCode(), $locale->getCode()))->equals(100));
}
$categoryIds = $this->categoryRepository->getAllChildrenIds($channel->getCategory());
$qb->addAnd($qb->expr()->field('categoryIds')->in($categoryIds));
return $qb;
}
/**
* Get children category ids
*
* @param integer[] $categoryIds
*
* @return integer[]
*/
protected function getAllChildrenIds(array $categoryIds)
{
$allChildrenIds = [];
foreach ($categoryIds as $categoryId) {
$category = $this->categoryRepository->find($categoryId);
$childrenIds = $this->categoryRepository->getAllChildrenIds($category);
$childrenIds[] = $category->getId();
$allChildrenIds = array_merge($allChildrenIds, $childrenIds);
}
return $allChildrenIds;
}
/**
* {@inheritdoc}
*/
public function getCompleteProductsCountPerChannels()
{
$channels = $this->channelManager->getFullChannels();
$productRepo = $this->documentManager->getRepository($this->productClass);
$productsCount = array();
foreach ($channels as $channel) {
$category = $channel->getCategory();
$categoryQb = $this->categoryRepository->getAllChildrenQueryBuilder($category, true);
$categoryIds = $this->categoryRepository->getCategoryIds($category, $categoryQb);
foreach ($channel->getLocales() as $locale) {
$data = array();
$compSuffix = $channel->getCode() . '-' . $locale->getCode();
$qb = $productRepo->createQueryBuilder()->hydrate(false)->field('categoryIds')->in($categoryIds)->field('enabled')->equals(true)->field('normalizedData.completenesses.' . $compSuffix)->equals(100)->select('_id');
$localeCount = $qb->getQuery()->execute()->count();
$data['locale'] = $locale->getCode();
$data['label'] = $channel->getLabel();
$data['total'] = $localeCount;
$productsCount[] = $data;
}
}
return $productsCount;
}
function let(CategoryManager $categoryManager, CategoryRepositoryInterface $categoryRepository)
{
$categoryRepository->findBy(['parent' => null])->willReturn(['this', 'is', 'a', 'category', 'tree']);
$categoryManager->getEntityRepository()->willReturn($categoryRepository);
$this->beConstructedWith($categoryManager, 'Pim\\Bundle\\CatalogBundle\\Entity\\Category', 'Pim\\Bundle\\EnrichBundle\\MassEditAction\\Operation\\Classify');
}