Example #1
0
 /**
  * Set country filter
  *
  * @param string|\Magento\Directory\Model\Country $country
  * @return \Magento\Directory\Model\ResourceModel\Country\Format\Collection
  */
 public function setCountryFilter($country)
 {
     if ($country instanceof \Magento\Directory\Model\Country) {
         $countryId = $country->getId();
     } else {
         $countryId = $country;
     }
     return $this->addFieldToFilter('country_id', $countryId);
 }
 public function getCountry()
 {
     if (is_null($this->country)) {
         $this->country = $this->countryFactory->create();
         try {
             $this->country->loadByCode($this->getData('country_code'));
         } catch (\Exception $e) {
         }
     }
     return $this->country;
 }
Example #3
0
 protected function setUp()
 {
     $this->helper = new \Magento\Framework\TestFramework\Unit\Helper\ObjectManager($this);
     $this->scope = $this->getMockBuilder('\\Magento\\Framework\\App\\Config\\ScopeConfigInterface')->disableOriginalConstructor()->getMock();
     $this->scope->expects($this->any())->method('getValue')->will($this->returnCallback([$this, 'scopeConfiggetValue']));
     $this->error = $this->getMockBuilder('\\Magento\\Quote\\Model\\Quote\\Address\\RateResult\\Error')->setMethods(['setCarrier', 'setCarrierTitle', 'setErrorMessage'])->getMock();
     $this->errorFactory = $this->getMockBuilder('Magento\\Quote\\Model\\Quote\\Address\\RateResult\\ErrorFactory')->disableOriginalConstructor()->setMethods(['create'])->getMock();
     $this->errorFactory->expects($this->any())->method('create')->willReturn($this->error);
     $this->rate = $this->getMock('Magento\\Shipping\\Model\\Rate\\Result', ['getError'], [], '', false);
     $rateFactory = $this->getMock('Magento\\Shipping\\Model\\Rate\\ResultFactory', ['create'], [], '', false);
     $rateFactory->expects($this->any())->method('create')->willReturn($this->rate);
     $this->country = $this->getMockBuilder('\\Magento\\Directory\\Model\\Country')->disableOriginalConstructor()->setMethods(['load'])->getMock();
     $this->abstractModel = $this->getMockBuilder('Magento\\Framework\\Model\\AbstractModel')->disableOriginalConstructor()->setMethods(['getData'])->getMock();
     $this->country->expects($this->any())->method('load')->willReturn($this->abstractModel);
     $this->countryFactory = $this->getMockBuilder('\\Magento\\Directory\\Model\\CountryFactory')->disableOriginalConstructor()->setMethods(['create'])->getMock();
     $this->countryFactory->expects($this->any())->method('create')->willReturn($this->country);
     $this->model = $this->helper->getObject('Magento\\Ups\\Model\\Carrier', ['scopeConfig' => $this->scope, 'rateErrorFactory' => $this->errorFactory, 'countryFactory' => $this->countryFactory, 'rateFactory' => $rateFactory]);
 }