PriceListTreeHandler TypeErrors fix

oroinc · Sep 29, 2020 · 5636e49 · 5636e49
1 parent eee9924
commit 5636e49
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/Oro/Bundle/PricingBundle/Model/PriceListTreeHandler.php b/src/Oro/Bundle/PricingBundle/Model/PriceListTreeHandler.php
@@ -75,13 +75,22 @@ public function __construct(
      * @param Customer|null $customer
      * @param Website|null $website
      * @return CombinedPriceList|null
+     *
+     * @throws \LogicException
      */
     public function getPriceList(Customer $customer = null, Website $website = null)
     {
         if (!$website) {
             $website = $this->websiteManager->getCurrentWebsite();
         }
 
+        if (null === $website) {
+            throw new \LogicException(\sprintf(
+                'When "%s" is called not in a storefront context, a website must be passed as a parameter',
+                __METHOD__
+            ));
+        }
+
         $key = $this->getUniqueKey($customer, $website);
         if (array_key_exists($key, $this->priceLists)) {
             return $this->priceLists[$key];
@@ -123,8 +132,8 @@ protected function getPriceListByCustomer(Customer $customer, Website $website)
     }
 
     /**
-     * @param Customer|null $customer
-     * @param Website|null $website
+     * @param Customer $customer
+     * @param Website $website
      * @return null|CombinedPriceList
      */
     protected function getPriceListByCustomerGroup(Customer $customer, Website $website)
@@ -139,7 +148,7 @@ protected function getPriceListByCustomerGroup(Customer $customer, Website $webs
     }
 
     /**
-     * @param Website|null $website
+     * @param Website $website
      * @return null|CombinedPriceList
      */
     protected function getPriceListByAnonymousCustomerGroup(Website $website)