MAGE-1085: Add customer group replicas test

Author: damcou

Test/Integration/Config/MultiStoreConfigTest.php

@@ -5,13 +5,15 @@
 use Algolia\AlgoliaSearch\Exceptions\AlgoliaException;
 use Algolia\AlgoliaSearch\Helper\AlgoliaHelper;
 use Algolia\AlgoliaSearch\Helper\ConfigHelper;
+use Algolia\AlgoliaSearch\Test\Integration\Config\Traits\ConfigAssertionsTrait;
 use Algolia\AlgoliaSearch\Test\Integration\MultiStoreTestCase;
 
 /**
  * @magentoDataFixture Magento/Store/_files/second_website_with_two_stores.php
  */
 class MultiStoreConfigTest extends MultiStoreTestCase
 {
+    use ConfigAssertionsTrait;
 
     const ADDITIONAL_ATTRIBUTE = 'additional_attribute';
 
@@ -30,11 +32,13 @@ public function testMultiStoreIndicesCreation()
 
         $defaultStore = $this->storeRepository->get('default');
         $fixtureSecondStore = $this->storeRepository->get('fixture_second_store');
+        $fixtureThirdStore = $this->storeRepository->get('fixture_third_store');
 
         $indicesCreatedByTest = 0;
 
-        $indicesCreatedByTest += $this->countStoreIndices($defaultStore->getId());
-        $indicesCreatedByTest += $this->countStoreIndices($fixtureSecondStore->getId());
+        $indicesCreatedByTest += $this->countStoreIndices($defaultStore);
+        $indicesCreatedByTest += $this->countStoreIndices($fixtureSecondStore);
+        $indicesCreatedByTest += $this->countStoreIndices($fixtureThirdStore);
 
         // Check that the configuration created the appropriate number of indices (7 (4 mains + 3 replicas per store => 3*7=21)
         $this->assertEquals(21, $indicesCreatedByTest);
@@ -121,28 +125,6 @@ public function testMultiStoreIndicesCreation()
         $this->assertEquals(1, $fixtureProductIndexRules['nbHits']);
     }
 
-    /**
-     * @param int|null $storeId
-     * @return int
-     * @throws AlgoliaException
-     */
-    protected function countStoreIndices(int $storeId = null): int
-    {
-        $indices = $this->algoliaHelper->listIndexes($storeId);
-
-        $indicesCreatedByTest = 0;
-
-        foreach ($indices['items'] as $index) {
-            $name = $index['name'];
-
-            if (mb_strpos($name, $this->indexPrefix) === 0) {
-                $indicesCreatedByTest++;
-            }
-        }
-
-        return $indicesCreatedByTest;
-    }
-
     protected function tearDown(): void
     {
         parent::tearDown();

Test/Integration/Config/Traits/ConfigAssertionsTrait.php

@@ -0,0 +1,31 @@
+<?php
+
+namespace Algolia\AlgoliaSearch\Test\Integration\Config\Traits;
+
+use Algolia\AlgoliaSearch\Exceptions\AlgoliaException;
+use Magento\Store\Api\Data\StoreInterface;
+
+trait ConfigAssertionsTrait
+{
+    /**
+     * @param StoreInterface|null $store
+     * @return int
+     * @throws AlgoliaException
+     */
+    protected function countStoreIndices(StoreInterface $store = null): int
+    {
+        $indices = $this->algoliaHelper->listIndexes($store->getId());
+
+        $indicesCreatedByTest = 0;
+
+        foreach ($indices['items'] as $index) {
+            $name = $index['name'];
+
+            if (mb_strpos($name, $this->indexPrefix . $store->getCode()) === 0) {
+                $indicesCreatedByTest++;
+            }
+        }
+
+        return $indicesCreatedByTest;
+    }
+}

Test/Integration/Product/MultiStoreReplicaTest.php

@@ -7,6 +7,7 @@
 use Algolia\AlgoliaSearch\Console\Command\ReplicaSyncCommand;
 use Algolia\AlgoliaSearch\Helper\ConfigHelper;
 use Algolia\AlgoliaSearch\Service\Product\SortingTransformer;
+use Algolia\AlgoliaSearch\Test\Integration\Config\Traits\ConfigAssertionsTrait;
 use Algolia\AlgoliaSearch\Test\Integration\MultiStoreTestCase;
 use Algolia\AlgoliaSearch\Test\Integration\Product\Traits\ReplicaAssertionsTrait;
 use Magento\Framework\Serialize\SerializerInterface;
@@ -20,6 +21,7 @@
 class MultiStoreReplicaTest extends MultiStoreTestCase
 {
     use ReplicaAssertionsTrait;
+    use ConfigAssertionsTrait;
 
     const COLOR_ATTR = 'color';
     const CREATED_AT_ATTR = 'created_at';
@@ -63,6 +65,35 @@ public function testMultiStoreReplicaConfig()
         $this->resetAllSortings();
     }
 
+    public function testCustomerGroupsConfig()
+    {
+        $defaultStore = $this->storeRepository->get('default');
+        $fixtureSecondStore = $this->storeRepository->get('fixture_second_store');
+        $fixtureThirdStore = $this->storeRepository->get('fixture_third_store');
+
+        // Enable customer groups for second fixture store and save configuration
+        $this->setConfig( ConfigHelper::CUSTOMER_GROUPS_ENABLE, 1, $fixtureSecondStore->getCode());
+        $this->indicesConfigurator->saveConfigurationToAlgolia($fixtureSecondStore->getId());
+        $this->algoliaHelper->waitLastTask($fixtureSecondStore->getId());
+
+        // 7 indices for default store and third store:
+        // - 1 for categories
+        // - 1 for pages
+        // - 1 for suggestions
+        // - 4 for products (1 main and 3 for sorting replicas)
+        $this->assertEquals(7, $this->countStoreIndices($defaultStore));
+        $this->assertEquals(7, $this->countStoreIndices($fixtureThirdStore));
+
+        // 13 indices for second store:
+        // - 1 for categories
+        // - 1 for pages
+        // - 1 for suggestions
+        // - 10 for products (1 main and 9 replicas (2 prices sortings * 4 customer groups + 1 other sorting (created_at))
+        $this->assertEquals(13, $this->countStoreIndices($fixtureSecondStore));
+
+        $this->resetAllSortings();
+    }
+
     public function testReplicaCommands()
     {
         $defaultStore = $this->storeRepository->get('default');
@@ -218,6 +249,8 @@ protected function resetAllSortings()
                 ]),
                 $store->getCode()
             );
+
+            $this->setConfig( ConfigHelper::CUSTOMER_GROUPS_ENABLE, 0, $store->getCode());
         }
     }