- added non proxy class of the Marello Product Unit as without this i…

…t can cause issues when the 'normal' class is used to send templates instead of the Proxy class of the product unit
marellocommerce · Aug 7, 2020 · d0d640d · d0d640d
1 parent ad177b0
commit d0d640d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/Marello/Bundle/ProductBundle/Twig/Provider/ProductUnitVariablesProvider.php b/src/Marello/Bundle/ProductBundle/Twig/Provider/ProductUnitVariablesProvider.php
@@ -2,7 +2,8 @@
 
 namespace Marello\Bundle\ProductBundle\Twig\Provider;
 
-use Proxies\__CG__\Extend\Entity\EV_Marello_Product_Unit;
+use Extend\Entity\EV_Marello_Product_Unit;
+use Proxies\__CG__\Extend\Entity\EV_Marello_Product_Unit as PROX_EV_Marello_Product_Unit;
 use Oro\Bundle\EntityBundle\Twig\Sandbox\EntityVariablesProviderInterface;
 
 class ProductUnitVariablesProvider implements EntityVariablesProviderInterface
@@ -24,6 +25,10 @@ public function getVariableGetters(): array
             EV_Marello_Product_Unit::class => [
                 'name' => 'getName',
                 'id' => 'getId'
+            ],
+            PROX_EV_Marello_Product_Unit::class => [
+                'name' => 'getName',
+                'id' => 'getId'
             ]
         ];
     }