EventBaseTest.php

<?php

declare (strict_types=1);
namespace Drupal\Tests\migrate\Unit\Event;

use Drupal\migrate\Event\EventBase;
use Drupal\Tests\UnitTestCase;

/**
 * @coversDefaultClass \Drupal\migrate\Event\EventBase
 * @group migrate
 */
class EventBaseTest extends UnitTestCase {

  /**
   * Tests getMigration method.
   *
   * @covers ::__construct
   * @covers ::getMigration
   */
  public function testGetMigration() {
    $migration = $this
      ->prophesize('\\Drupal\\migrate\\Plugin\\MigrationInterface')
      ->reveal();
    $message_service = $this
      ->prophesize('\\Drupal\\migrate\\MigrateMessageInterface')
      ->reveal();
    $event = new EventBase($migration, $message_service);
    $this
      ->assertSame($migration, $event
      ->getMigration());
  }

  /**
   * Tests logging a message.
   *
   * @covers ::__construct
   * @covers ::logMessage
   */
  public function testLogMessage() {
    $migration = $this
      ->prophesize('\\Drupal\\migrate\\Plugin\\MigrationInterface')
      ->reveal();
    $message_service = $this
      ->prophesize('\\Drupal\\migrate\\MigrateMessageInterface');
    $event = new EventBase($migration, $message_service
      ->reveal());

    // Assert that the intended calls to the services happen.
    $message_service
      ->display('status message', 'status')
      ->shouldBeCalledTimes(1);
    $event
      ->logMessage('status message');
    $message_service
      ->display('warning message', 'warning')
      ->shouldBeCalledTimes(1);
    $event
      ->logMessage('warning message', 'warning');
  }

}