migrate_conditions-1.0.0-beta1/tests/src/Unit/condition/EntityExistsTest.php

<?php

namespace Drupal\Tests\migrate_conditions\Unit\condition;

use Drupal\Component\Plugin\Exception\PluginNotFoundException;
use Drupal\migrate\MigrateException;
use Drupal\migrate_conditions\Plugin\migrate_conditions\condition\EntityExists;
use Drupal\Tests\UnitTestCase;

/**
 * Tests the entity_exists condition plugin.
 *
 * @group migrate_conditions
 * @coversDefaultClass \Drupal\migrate_conditions\Plugin\migrate_conditions\condition\EntityExists
 */
class EntityExistsTest extends UnitTestCase {

  /**
   * @covers ::row
   * @covers ::process
   * @dataProvider providerTestEvaluate
   */
  public function testEvaluate($source, $load, $configuration, $expected) {
    $row = $this->createMock('Drupal\migrate\Row');
    $storage = $this->createMock('\Drupal\Core\Entity\EntityStorageInterface');
    $storage->expects($this->once())
      ->method('load')
      ->willReturnMap([[$source, $load]]);
    $entity_type_manager = $this->createMock('\Drupal\Core\Entity\EntityTypeManagerInterface');
    $entity_type_manager->expects($this->once())
      ->method('getStorage')
      ->willReturnMap([[$configuration['entity_type'], $storage]]);

    $condition = new EntityExists($configuration, 'entity_exists', [], $entity_type_manager);
    $this->assertSame($expected, $condition->evaluate($source, $row));
  }

  /**
   * Data provider for ::testEvaluate().
   */
  public function providerTestEvaluate() {
    $mock_entity = $this->createMock('\Drupal\Core\Entity\EntityInterface');
    return [
      [
        'source' => 123,
        'load' => NULL,
        'configuration' => [
          'entity_type' => 'foo',
        ],
        'expected' => FALSE,
      ],
      [
        'source' => 123,
        'load' => $mock_entity,
        'configuration' => [
          'entity_type' => 'foo',
        ],
        'expected' => TRUE,
      ],
    ];
  }

  /**
   * @covers ::evaluate
   */
  public function testEvaluateExceptions() {
    $row = $this->createMock('Drupal\migrate\Row');
    $storage = $this->createMock('\Drupal\Core\Entity\EntityStorageInterface');
    $entity_type_manager = $this->createMock('\Drupal\Core\Entity\EntityTypeManagerInterface');
    $entity_type_manager->expects($this->once())
      ->method('getStorage')
      ->willReturn($storage);

    $configuration = [
      'entity_type' => 'foo',
    ];
    $condition = new EntityExists($configuration, 'entity_exists', [], $entity_type_manager);

    $this->expectException(MigrateException::class);
    $this->expectExceptionMessage("The source value for entity_exists must be an integer or a string.");
    $condition->evaluate(['an array'], $row);
  }

  /**
   * @covers ::__construct
   */
  public function testConstructorExceptions() {
    $row = $this->createMock('Drupal\migrate\Row');
    $entity_type_manager = $this->createMock('\Drupal\Core\Entity\EntityTypeManagerInterface');
    $entity_type_manager->expects($this->once())
      ->method('getStorage')
      ->willThrowException(new PluginNotFoundException('foo'));

    $configuration = [
      'entity_type' => 'foo',
    ];

    $this->expectException(\InvalidArgumentException::class);
    $this->expectExceptionMessage('The entity_type configured for entity_exists could not be loaded.');
    $condition = new EntityExists($configuration, 'entity_exists', [], $entity_type_manager);
  }

}