ProductVariationStorageTest.php in Commerce Core 8.2

Namespace

Drupal\Tests\commerce_product\Kernel

File

modules/product/tests/src/Kernel/ProductVariationStorageTest.php

<?php

namespace Drupal\Tests\commerce_product\Kernel;

use Drupal\commerce_product\Entity\Product;
use Drupal\commerce_product\Entity\ProductVariation;
use Drupal\Tests\commerce\Kernel\CommerceKernelTestBase;
use Symfony\Component\HttpFoundation\Request;

/**
 * Tests the product variation storage.
 *
 * @group commerce
 */
class ProductVariationStorageTest extends CommerceKernelTestBase {

  /**
   * The product variation storage.
   *
   * @var \Drupal\commerce_product\ProductVariationStorageInterface
   */
  protected $variationStorage;

  /**
   * Modules to enable.
   *
   * @var array
   */
  public static $modules = [
    'path',
    'commerce_product',
  ];

  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $this
      ->installEntitySchema('commerce_product_variation');
    $this
      ->installEntitySchema('commerce_product');
    $this
      ->installConfig([
      'commerce_product',
    ]);
    $this->variationStorage = $this->container
      ->get('entity_type.manager')
      ->getStorage('commerce_product_variation');
    $user = $this
      ->createUser([], [
      'administer commerce_product',
    ]);
    $this->container
      ->get('current_user')
      ->setAccount($user);
  }

  /**
   * Tests loading variations by SKU.
   */
  public function testLoadBySku() {
    $sku = strtolower($this
      ->randomMachineName());
    $variation = ProductVariation::create([
      'type' => 'default',
      'sku' => $sku,
      'title' => $this
        ->randomString(),
    ]);
    $variation
      ->save();
    $product = Product::create([
      'type' => 'default',
      'variations' => [
        $variation,
      ],
    ]);
    $product
      ->save();
    $result = $this->variationStorage
      ->loadBySku('FAKE');
    $this
      ->assertNull($result);
    $result = $this->variationStorage
      ->loadBySku($sku);
    $this
      ->assertEquals($result
      ->id(), $variation
      ->id());
  }

  /**
   * Tests loadEnabled() function.
   */
  public function testLoadEnabled() {
    $variations = [];
    for ($i = 1; $i <= 3; $i++) {
      $variation = ProductVariation::create([
        'type' => 'default',
        'sku' => strtolower($this
          ->randomMachineName()),
        'title' => $this
          ->randomString(),
        'status' => $i % 2,
      ]);
      $variation
        ->save();
      $variations[] = $variation;
    }
    $variations = array_reverse($variations);
    $product = Product::create([
      'type' => 'default',
      'variations' => $variations,
    ]);
    $product
      ->save();
    $variationsFiltered = $this->variationStorage
      ->loadEnabled($product);
    $this
      ->assertEquals(2, count($variationsFiltered), '2 out of 3 variations are enabled');
    $this
      ->assertEquals(reset($variations)
      ->getSku(), reset($variationsFiltered)
      ->getSku(), 'The sort order of the variations remains the same');
  }

  /**
   * Tests loadFromContext() method.
   */
  public function testLoadFromContext() {
    $variations = [];
    for ($i = 1; $i <= 3; $i++) {
      $variation = ProductVariation::create([
        'type' => 'default',
        'sku' => strtolower($this
          ->randomMachineName()),
        'title' => $this
          ->randomString(),
      ]);
      $variation
        ->save();
      $variations[] = $variation;
    }
    $variations = array_reverse($variations);
    $product = Product::create([
      'type' => 'default',
      'variations' => $variations,
    ]);
    $product
      ->save();
    $request = Request::create('');
    $request->query
      ->add([
      'v' => end($variations)
        ->id(),
    ]);

    // Push the request to the request stack so `current_route_match` works.
    $this->container
      ->get('request_stack')
      ->push($request);
    $this
      ->assertNotEquals($request->query
      ->get('v'), $product
      ->getDefaultVariation()
      ->id());
    $context_variation = $this->variationStorage
      ->loadFromContext($product);
    $this
      ->assertEquals($request->query
      ->get('v'), $context_variation
      ->id());

    // Invalid variation ID returns default variation.
    $request = Request::create('');
    $request->query
      ->add([
      'v' => '1111111',
    ]);

    // Push the request to the request stack so `current_route_match` works.
    $this->container
      ->get('request_stack')
      ->push($request);
    $this
      ->assertNotEquals($request->query
      ->get('v'), $product
      ->getDefaultVariation()
      ->id());
    $context_variation = $this->variationStorage
      ->loadFromContext($product);
    $this
      ->assertEquals($product
      ->getDefaultVariation()
      ->id(), $context_variation
      ->id());
  }

}

Classes

Name	Description
ProductVariationStorageTest	Tests the product variation storage.