public function PageTest::testAddVariant in Page Manager 8

Same name and namespace in other branches

1. 8.4 tests/src/Unit/PageTest.php \Drupal\Tests\page_manager\Unit\PageTest::testAddVariant()

@covers ::addVariant

File

tests/src/Unit/PageTest.php, line 79

Contains \Drupal\Tests\page_manager\Unit\PageTest.

Class

PageTest

Tests the Page entity.

Namespace

Drupal\Tests\page_manager\Unit

Code

public function testAddVariant() {
  $variant1 = $this
    ->prophesize(PageVariantInterface::class);
  $variant1
    ->id()
    ->willReturn('variant1');
  $variant1
    ->getWeight()
    ->willReturn(0);
  $variant2 = $this
    ->prophesize(PageVariantInterface::class);
  $variant2
    ->id()
    ->willReturn('variant2');
  $variant2
    ->getWeight()
    ->willReturn(-10);
  $variant3 = $this
    ->prophesize(PageVariantInterface::class);
  $variant3
    ->id()
    ->willReturn('variant3');
  $variant3
    ->getWeight()
    ->willReturn(-5);
  $entity_storage = $this
    ->prophesize(EntityStorageInterface::class);
  $entity_storage
    ->loadByProperties([
    'page' => 'the_page',
  ])
    ->willReturn([
    'variant1' => $variant1
      ->reveal(),
    'variant2' => $variant2
      ->reveal(),
  ])
    ->shouldBeCalledTimes(1);
  $entity_type_manager = $this
    ->prophesize(EntityTypeManagerInterface::class);
  $entity_type_manager
    ->getStorage('page_variant')
    ->willReturn($entity_storage);
  $container = new ContainerBuilder();
  $container
    ->set('entity_type.manager', $entity_type_manager
    ->reveal());
  \Drupal::setContainer($container);

  // Check that Variant 1 and 2 are in the right order.
  $variants = $this->page
    ->getVariants();
  $this
    ->assertSame([
    'variant2' => $variant2
      ->reveal(),
    'variant1' => $variant1
      ->reveal(),
  ], $variants);

  // Add Variant 3.
  $this->page
    ->addVariant($variant3
    ->reveal());

  // Check that Variant 1, 2 and 3 are in the right order.
  $variants = $this->page
    ->getVariants();
  $this
    ->assertSame([
    'variant2' => $variant2
      ->reveal(),
    'variant3' => $variant3
      ->reveal(),
    'variant1' => $variant1
      ->reveal(),
  ], $variants);
}