StateTransitionValidationTest.php in Workbench Moderation 8

Namespace

Drupal\Tests\workbench_moderation\Unit

File

tests/src/Unit/StateTransitionValidationTest.php

<?php

namespace Drupal\Tests\workbench_moderation\Unit;

use Drupal\Core\Entity\EntityStorageInterface;
use Drupal\Core\Entity\EntityTypeManagerInterface;
use Drupal\Core\Session\AccountInterface;
use Drupal\Tests\UnitTestCase;
use Drupal\workbench_moderation\ModerationStateInterface;
use Drupal\workbench_moderation\ModerationStateTransitionInterface;
use Drupal\workbench_moderation\StateTransitionValidation;
use Prophecy\Argument;

/**
 * @coversDefaultClass \Drupal\workbench_moderation\StateTransitionValidation
 * @group workbench_moderation
 */
class StateTransitionValidationTest extends UnitTestCase {

  /**
   * Builds a mock storage object for Transitions.
   *
   * @return \Drupal\Core\Entity\EntityStorageInterface
   *   Returns an entity storage config.
   */
  protected function setupTransitionStorage() {
    $entity_storage = $this
      ->prophesize(EntityStorageInterface::class);
    $list = $this
      ->setupTransitionEntityList();
    $entity_storage
      ->loadMultiple()
      ->willReturn($list);
    $entity_storage
      ->loadMultiple(Argument::type('array'))
      ->will(function ($args) use ($list) {
      $keys = $args[0];
      if (empty($keys)) {
        return $list;
      }
      $return = array_map(function ($key) use ($list) {
        return $list[$key];
      }, $keys);
      return $return;
    });
    return $entity_storage
      ->reveal();
  }

  /**
   * Builds an array of mocked Transition objects.
   *
   * @return \Drupal\workbench_moderation\ModerationStateTransitionInterface[]
   *   Returns a list.
   */
  protected function setupTransitionEntityList() {
    $transition = $this
      ->prophesize(ModerationStateTransitionInterface::class);
    $transition
      ->id()
      ->willReturn('draft__needs_review');
    $transition
      ->getFromState()
      ->willReturn('draft');
    $transition
      ->getToState()
      ->willReturn('needs_review');
    $list[$transition
      ->reveal()
      ->id()] = $transition
      ->reveal();
    $transition = $this
      ->prophesize(ModerationStateTransitionInterface::class);
    $transition
      ->id()
      ->willReturn('needs_review__staging');
    $transition
      ->getFromState()
      ->willReturn('needs_review');
    $transition
      ->getToState()
      ->willReturn('staging');
    $list[$transition
      ->reveal()
      ->id()] = $transition
      ->reveal();
    $transition = $this
      ->prophesize(ModerationStateTransitionInterface::class);
    $transition
      ->id()
      ->willReturn('staging__published');
    $transition
      ->getFromState()
      ->willReturn('staging');
    $transition
      ->getToState()
      ->willReturn('published');
    $list[$transition
      ->reveal()
      ->id()] = $transition
      ->reveal();
    $transition = $this
      ->prophesize(ModerationStateTransitionInterface::class);
    $transition
      ->id()
      ->willReturn('needs_review__draft');
    $transition
      ->getFromState()
      ->willReturn('needs_review');
    $transition
      ->getToState()
      ->willReturn('draft');
    $list[$transition
      ->reveal()
      ->id()] = $transition
      ->reveal();
    $transition = $this
      ->prophesize(ModerationStateTransitionInterface::class);
    $transition
      ->id()
      ->willReturn('draft__draft');
    $transition
      ->getFromState()
      ->willReturn('draft');
    $transition
      ->getToState()
      ->willReturn('draft');
    $list[$transition
      ->reveal()
      ->id()] = $transition
      ->reveal();
    $transition = $this
      ->prophesize(ModerationStateTransitionInterface::class);
    $transition
      ->id()
      ->willReturn('needs_review__needs_review');
    $transition
      ->getFromState()
      ->willReturn('needs_review');
    $transition
      ->getToState()
      ->willReturn('needs_review');
    $list[$transition
      ->reveal()
      ->id()] = $transition
      ->reveal();
    $transition = $this
      ->prophesize(ModerationStateTransitionInterface::class);
    $transition
      ->id()
      ->willReturn('published__published');
    $transition
      ->getFromState()
      ->willReturn('published');
    $transition
      ->getToState()
      ->willReturn('published');
    $list[$transition
      ->reveal()
      ->id()] = $transition
      ->reveal();
    return $list;
  }

  /**
   * Builds a mock storage object for States.
   *
   * @return \Drupal\Core\Entity\EntityStorageInterface
   *   Returns an entity storage config.
   */
  protected function setupStateStorage() {
    $entity_storage = $this
      ->prophesize(EntityStorageInterface::class);
    $state = $this
      ->prophesize(ModerationStateInterface::class);
    $state
      ->id()
      ->willReturn('draft');
    $state
      ->label()
      ->willReturn('Draft');
    $state
      ->isPublishedState()
      ->willReturn(FALSE);
    $state
      ->isDefaultRevisionState()
      ->willReturn(FALSE);
    $states['draft'] = $state
      ->reveal();
    $state = $this
      ->prophesize(ModerationStateInterface::class);
    $state
      ->id()
      ->willReturn('needs_review');
    $state
      ->label()
      ->willReturn('Needs Review');
    $state
      ->isPublishedState()
      ->willReturn(FALSE);
    $state
      ->isDefaultRevisionState()
      ->willReturn(FALSE);
    $states['needs_review'] = $state
      ->reveal();
    $state = $this
      ->prophesize(ModerationStateInterface::class);
    $state
      ->id()
      ->willReturn('published');
    $state
      ->label()
      ->willReturn('Published');
    $state
      ->isPublishedState()
      ->willReturn(TRUE);
    $state
      ->isDefaultRevisionState()
      ->willReturn(TRUE);
    $states['published'] = $state
      ->reveal();
    $entity_storage
      ->loadMultiple()
      ->willReturn($states);
    return $entity_storage
      ->reveal();
  }

  /**
   * Builds a mocked Entity Type Manager.
   *
   * @return \Drupal\Core\Entity\EntityTypeManagerInterface
   *   Returns an entity type manager.
   */
  protected function setupEntityTypeManager() {
    $entityTypeManager = $this
      ->prophesize(EntityTypeManagerInterface::class);
    $entityTypeManager
      ->getStorage('moderation_state')
      ->willReturn($this
      ->setupStateStorage());
    $entityTypeManager
      ->getStorage('moderation_state_transition')
      ->willReturn($this
      ->setupTransitionStorage());
    return $entityTypeManager
      ->reveal();
  }

  /**
   * @covers ::isTransitionAllowed
   * @covers ::calculatePossibleTransitions
   */
  public function testIsTransitionAllowedWithValidTransition() {
    $state_transition_validation = new StateTransitionValidation($this
      ->setupEntityTypeManager());
    $this
      ->assertTrue($state_transition_validation
      ->isTransitionAllowed('draft', 'draft'));
    $this
      ->assertTrue($state_transition_validation
      ->isTransitionAllowed('draft', 'needs_review'));
    $this
      ->assertTrue($state_transition_validation
      ->isTransitionAllowed('needs_review', 'needs_review'));
    $this
      ->assertTrue($state_transition_validation
      ->isTransitionAllowed('needs_review', 'staging'));
    $this
      ->assertTrue($state_transition_validation
      ->isTransitionAllowed('staging', 'published'));
    $this
      ->assertTrue($state_transition_validation
      ->isTransitionAllowed('needs_review', 'draft'));
  }

  /**
   * @covers ::isTransitionAllowed
   * @covers ::calculatePossibleTransitions
   */
  public function testIsTransitionAllowedWithInValidTransition() {
    $state_transition_validation = new StateTransitionValidation($this
      ->setupEntityTypeManager());
    $this
      ->assertFalse($state_transition_validation
      ->isTransitionAllowed('published', 'needs_review'));
    $this
      ->assertFalse($state_transition_validation
      ->isTransitionAllowed('published', 'staging'));
    $this
      ->assertFalse($state_transition_validation
      ->isTransitionAllowed('staging', 'needs_review'));
    $this
      ->assertFalse($state_transition_validation
      ->isTransitionAllowed('staging', 'staging'));
    $this
      ->assertFalse($state_transition_validation
      ->isTransitionAllowed('needs_review', 'published'));
    $this
      ->assertFalse($state_transition_validation
      ->isTransitionAllowed('published', 'archived'));
    $this
      ->assertFalse($state_transition_validation
      ->isTransitionAllowed('archived', 'published'));
  }

  /**
   * Verifies user-aware transition validation.
   *
   * @param string $from
   *   The state to transition from.
   * @param string $to
   *   The state to transition to.
   * @param string $permission
   *   The permission to give the user, or not.
   * @param bool $allowed
   *   Whether or not to grant a user this permission.
   * @param bool $result
   *   Whether userMayTransition() is expected to return TRUE or FALSE.
   *
   * @dataProvider userTransitionsProvider
   */
  public function testUserSensitiveValidTransitions($from, $to, $permission, $allowed, $result) {
    $user = $this
      ->prophesize(AccountInterface::class);

    // The one listed permission will be returned as instructed; Any others are
    // always denied.
    $user
      ->hasPermission($permission)
      ->willReturn($allowed);
    $user
      ->hasPermission(Argument::type('string'))
      ->willReturn(FALSE);
    $validator = new Validator($this
      ->setupEntityTypeManager());
    $this
      ->assertEquals($result, $validator
      ->userMayTransition($from, $to, $user
      ->reveal()));
  }

  /**
   * Data provider for the user transition test.
   *
   * @return array
   *   Returns an array.
   */
  public function userTransitionsProvider() {

    // The user has the right permission, so let it through.
    $ret[] = [
      'draft',
      'draft',
      'use draft__draft transition',
      TRUE,
      TRUE,
    ];

    // The user doesn't have the right permission, block it.
    $ret[] = [
      'draft',
      'draft',
      'use draft__draft transition',
      FALSE,
      FALSE,
    ];

    // The user has some other permission that doesn't matter.
    $ret[] = [
      'draft',
      'draft',
      'use draft__needs_review transition',
      TRUE,
      FALSE,
    ];

    // The user has permission, but the transition isn't allowed anyway.
    $ret[] = [
      'published',
      'needs_review',
      'use published__needs_review transition',
      TRUE,
      FALSE,
    ];
    return $ret;
  }

}

/**
 * Testable subclass for selected tests.
 *
 * EntityQuery is beyond untestable, so we have to subclass and override the
 * method that uses it.
 */
class Validator extends StateTransitionValidation {

  /**
   * {@inheritdoc}
   */
  protected function getTransitionFromStates($from, $to) {
    if ($from == 'draft' && $to == 'draft') {
      return $this
        ->transitionStorage()
        ->loadMultiple([
        'draft__draft',
      ])[0];
    }
  }

}

Classes

Name	Description
StateTransitionValidationTest	@coversDefaultClass \Drupal\workbench_moderation\StateTransitionValidation @group workbench_moderation
Validator	Testable subclass for selected tests.