Fix setProductStyle method.

debuss · Mar 24, 2020 · df6cbdb · df6cbdb
1 parent cac1663
commit df6cbdb
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/Object/Product.php b/src/Object/Product.php
@@ -236,7 +236,15 @@ public function getProductStyle(): int
      */
     public function setProductStyle(int $product_style): Product
     {
-        if (!in_array($product_style, Provisionning::getInstance()->getProductStyles())) {
+        static $categories;
+        if (!is_array($categories) || !count($categories)) {
+            $categories = array_column(
+                Provisionning::getInstance()->getCategories(),
+                'code'
+            );
+        }
+
+        if (!in_array($product_style, $categories)) {
             throw InvalidArgumentException::notSupportedProductStyle($product_style);
         }