ENGCOM-4137: Fixed issue #334 Changed hard coded values according to TYPE_CODE #340

Author: naydav

app/code/Magento/BundleGraphQl/Model/BundleProductTypeResolver.php

@@ -8,19 +8,22 @@
 namespace Magento\BundleGraphQl\Model;
 
 use Magento\Framework\GraphQl\Query\Resolver\TypeResolverInterface;
+use Magento\Bundle\Model\Product\Type as Type;
 
 /**
  * {@inheritdoc}
  */
 class BundleProductTypeResolver implements TypeResolverInterface
 {
+    const BUNDLE_PRODUCT = 'BundleProduct';
+    
     /**
      * {@inheritdoc}
      */
     public function resolveType(array $data) : string
     {
-        if (isset($data['type_id']) && $data['type_id'] == 'bundle') {
-            return 'BundleProduct';
+        if (isset($data['type_id']) && $data['type_id'] == Type::TYPE_CODE) {
+            return self::BUNDLE_PRODUCT;
         }
         return '';
     }

app/code/Magento/DownloadableGraphQl/Model/DownloadableProductTypeResolver.php

@@ -8,19 +8,21 @@
 namespace Magento\DownloadableGraphQl\Model;
 
 use Magento\Framework\GraphQl\Query\Resolver\TypeResolverInterface;
+use Magento\Downloadable\Model\Product\Type as Type;
 
 /**
  * {@inheritdoc}
  */
 class DownloadableProductTypeResolver implements TypeResolverInterface
 {
+    const DOWNLOADABLE_PRODUCT = 'DownloadableProduct';
     /**
      * {@inheritdoc}
      */
     public function resolveType(array $data) : string
     {
-        if (isset($data['type_id']) && $data['type_id'] == 'downloadable') {
-            return 'DownloadableProduct';
+        if (isset($data['type_id']) && $data['type_id'] == Type::TYPE_DOWNLOADABLE) {
+            return self::DOWNLOADABLE_PRODUCT;
         }
         return '';
     }

app/code/Magento/GroupedProductGraphQl/Model/GroupedProductTypeResolver.php

@@ -8,19 +8,20 @@
 namespace Magento\GroupedProductGraphQl\Model;
 
 use Magento\Framework\GraphQl\Query\Resolver\TypeResolverInterface;
-
+use Magento\GroupedProduct\Model\Product\Type\Grouped as Type;
 /**
  * {@inheritdoc}
  */
 class GroupedProductTypeResolver implements TypeResolverInterface
 {
+    const GROUPED_PRODUCT = 'GroupedProduct';
     /**
      * {@inheritdoc}
      */
     public function resolveType(array $data) : string
     {
-        if (isset($data['type_id']) && $data['type_id'] == 'grouped') {
-            return 'GroupedProduct';
+        if (isset($data['type_id']) && $data['type_id'] == Type::TYPE_CODE) {
+            return self::GROUPED_PRODUCT;
         }
         return '';
     }