function it_updates_a_family($attrRequiFactory, $channelRepository, FamilyTranslation $translation, FamilyInterface $family, AttributeRepositoryInterface $attributeRepository, AttributeInterface $skuAttribute, AttributeInterface $nameAttribute, AttributeInterface $descAttribute, AttributeInterface $priceAttribute, AttributeRequirementInterface $skuMobileRqrmt, AttributeRequirementInterface $nameMobileRqrmt, AttributeRequirementInterface $skuPrintRqrmt, AttributeRequirementInterface $namePrintRqrmt, AttributeRequirementInterface $descPrintRqrmt, ChannelInterface $mobileChannel, ChannelInterface $printChannel) { $values = ['code' => 'mycode', 'attributes' => ['sku', 'name', 'description', 'price'], 'attribute_as_label' => 'name', 'requirements' => ['mobile' => ['sku', 'name'], 'print' => ['name', 'description']], 'labels' => ['fr_FR' => 'Moniteurs', 'en_US' => 'PC Monitors']]; $family->getAttributeRequirements()->willReturn([$skuMobileRqrmt, $skuPrintRqrmt]); $family->getAttributes()->willReturn([$skuAttribute, $nameAttribute, $descAttribute, $priceAttribute]); $family->removeAttribute($nameAttribute)->shouldBeCalled(); $family->removeAttribute($priceAttribute)->shouldBeCalled(); $family->removeAttribute($descAttribute)->shouldBeCalled(); $family->getId()->willReturn(42); $skuAttribute->getId()->willReturn(1); $nameAttribute->getId()->willReturn(2); $descAttribute->getId()->willReturn(3); $priceAttribute->getId()->willReturn(4); $skuMobileRqrmt->getAttribute()->willReturn($skuAttribute); $skuMobileRqrmt->getChannelCode()->willReturn('mobile'); $skuPrintRqrmt->getAttribute()->willReturn($skuAttribute); $skuPrintRqrmt->getChannelCode()->willReturn('print'); $attributeRepository->findOneByIdentifier('sku')->willReturn($skuAttribute); $attributeRepository->findOneByIdentifier('name')->willReturn($nameAttribute); $attributeRepository->findOneByIdentifier('description')->willReturn($descAttribute); $attributeRepository->findOneByIdentifier('price')->willReturn($priceAttribute); $attributeRepository->getIdentifier()->willReturn($skuAttribute); $skuAttribute->getAttributeType()->willReturn('pim_catalog_identifier'); $nameAttribute->getAttributeType()->willReturn('pim_catalog_text'); $descAttribute->getAttributeType()->willReturn('pim_catalog_textarea'); $priceAttribute->getAttributeType()->willReturn('pim_catalog_price_collection'); $channelRepository->getChannelCodes()->willReturn(['mobile', 'print']); $channelRepository->findOneByIdentifier('mobile')->willReturn($mobileChannel); $channelRepository->findOneByIdentifier('print')->willReturn($printChannel); $attrRequiFactory->createAttributeRequirement($nameAttribute, $mobileChannel, true)->willReturn($nameMobileRqrmt); $attrRequiFactory->createAttributeRequirement($nameAttribute, $printChannel, true)->willReturn($namePrintRqrmt); $attrRequiFactory->createAttributeRequirement($descAttribute, $printChannel, true)->willReturn($descPrintRqrmt); $nameMobileRqrmt->getAttribute()->willReturn($nameAttribute); $namePrintRqrmt->getAttribute()->willReturn($nameAttribute); $descPrintRqrmt->getAttribute()->willReturn($descAttribute); $family->setAttributeRequirements([$skuMobileRqrmt, $skuPrintRqrmt, $nameMobileRqrmt, $namePrintRqrmt, $descPrintRqrmt])->shouldBeCalled(); $family->setCode('mycode')->shouldBeCalled(); $family->addAttribute($skuAttribute)->shouldBeCalled(); $family->addAttribute($nameAttribute)->shouldBeCalled(); $family->addAttribute($skuAttribute)->shouldBeCalled(); $family->addAttribute($skuAttribute)->shouldBeCalled(); $family->setLocale('en_US')->shouldBeCalled(); $family->setLocale('fr_FR')->shouldBeCalled(); $family->getTranslation()->willReturn($translation); $translation->setLabel('label en us'); $translation->setLabel('label fr fr'); $family->addAttribute($skuAttribute)->shouldBeCalled(); $family->addAttribute($nameAttribute)->shouldBeCalled(); $family->addAttribute($descAttribute)->shouldBeCalled(); $family->addAttribute($priceAttribute)->shouldBeCalled(); $family->setAttributeAsLabel($nameAttribute)->shouldBeCalled(); $this->update($family, $values, []); }
/** * @param FamilyInterface $family * @param array $data * * @throws \InvalidArgumentException */ protected function addAttributes(FamilyInterface $family, array $data) { foreach ($family->getAttributes() as $attribute) { if (AttributeTypes::IDENTIFIER !== $attribute->getAttributeType()) { $family->removeAttribute($attribute); } } foreach ($data as $attributeCode) { if (null !== ($attribute = $this->attributeRepository->findOneByIdentifier($attributeCode))) { $family->addAttribute($attribute); } else { throw new \InvalidArgumentException(sprintf('Attribute with "%s" code does not exist', $attributeCode)); } } }