MigrateFileTest.php in Zircon Profile 8

Contains \Drupal\file\Tests\Migrate\d6\MigrateFileTest.

Namespace

Drupal\file\Tests\Migrate\d6

File

core/modules/file/src/Tests/Migrate/d6/MigrateFileTest.php

<?php

/**
 * @file
 * Contains \Drupal\file\Tests\Migrate\d6\MigrateFileTest.
 */
namespace Drupal\file\Tests\Migrate\d6;

use Drupal\Component\Utility\Random;
use Drupal\file\Entity\File;
use Drupal\file\FileInterface;
use Drupal\migrate\Entity\Migration;
use Drupal\migrate\Tests\MigrateDumpAlterInterface;
use Drupal\Core\Database\Database;
use Drupal\migrate_drupal\Tests\d6\MigrateDrupal6TestBase;
use Drupal\simpletest\TestBase;

/**
 * file migration.
 *
 * @group migrate_drupal_6
 */
class MigrateFileTest extends MigrateDrupal6TestBase implements MigrateDumpAlterInterface {

  /**
   * The filename of a file used to test temporary file migration.
   *
   * @var string
   */
  protected static $tempFilename;

  /**
   * {@inheritdoc}
   */
  protected function setUp() {
    parent::setUp();
    $this
      ->installEntitySchema('file');
    $this
      ->installConfig([
      'file',
    ]);

    /** @var \Drupal\migrate\Entity\MigrationInterface $migration */
    $migration = Migration::load('d6_file');
    $source = $migration
      ->get('source');
    $source['site_path'] = 'core/modules/simpletest';
    $migration
      ->set('source', $source);
    $this
      ->executeMigration($migration);
  }

  /**
   * Asserts a file entity.
   *
   * @param integer $fid
   *  The file ID.
   * @param string $name
   *  The expected file name.
   * @param integer $size
   *  The expected file size.
   * @param string $uri
   *  The expected file URI.
   * @param string $type
   *  The expected MIME type.
   * @param integer $uid
   *  The expected file owner ID.
   */
  protected function assertEntity($fid, $name, $size, $uri, $type, $uid) {

    /** @var \Drupal\file\FileInterface $file */
    $file = File::load($fid);
    $this
      ->assertTrue($file instanceof FileInterface);
    $this
      ->assertIdentical($name, $file
      ->getFilename());
    $this
      ->assertIdentical($size, $file
      ->getSize());
    $this
      ->assertIdentical($uri, $file
      ->getFileUri());
    $this
      ->assertIdentical($type, $file
      ->getMimeType());
    $this
      ->assertIdentical($uid, $file
      ->getOwnerId());
  }

  /**
   * Tests the Drupal 6 files to Drupal 8 migration.
   */
  public function testFiles() {
    $this
      ->assertEntity(1, 'Image1.png', '39325', 'public://image-1.png', 'image/png', '1');
    $this
      ->assertEntity(2, 'Image2.jpg', '1831', 'public://image-2.jpg', 'image/jpeg', '1');
    $this
      ->assertEntity(3, 'Image-test.gif', '183', 'public://image-test.gif', 'image/jpeg', '1');
    $this
      ->assertEntity(5, 'html-1.txt', '24', 'public://html-1.txt', 'text/plain', '1');

    // Test that we can re-import and also test with file_directory_path set.
    \Drupal::database()
      ->truncate(Migration::load('d6_file')
      ->getIdMap()
      ->mapTableName())
      ->execute();

    // Update the file_directory_path.
    Database::getConnection('default', 'migrate')
      ->update('variable')
      ->fields(array(
      'value' => serialize('files/test'),
    ))
      ->condition('name', 'file_directory_path')
      ->execute();
    Database::getConnection('default', 'migrate')
      ->update('variable')
      ->fields(array(
      'value' => serialize($this
        ->getTempFilesDirectory()),
    ))
      ->condition('name', 'file_directory_temp')
      ->execute();
    $migration = \Drupal::entityManager()
      ->getStorage('migration')
      ->loadUnchanged('d6_file');
    $this
      ->executeMigration($migration);
    $file = File::load(2);
    $this
      ->assertIdentical('public://core/modules/simpletest/files/image-2.jpg', $file
      ->getFileUri());

    // Ensure that a temporary file has been migrated.
    $file = File::load(6);
    $this
      ->assertIdentical('temporary://' . static::getUniqueFilename(), $file
      ->getFileUri());

    // File 7, created in static::migrateDumpAlter(), shares a path with
    // file 5, which means it should be skipped entirely.
    $this
      ->assertNull(File::load(7));
  }

  /**
   * @return string
   *   A filename based upon the test.
   */
  public static function getUniqueFilename() {
    return static::$tempFilename;
  }

  /**
   * {@inheritdoc}
   */
  public static function migrateDumpAlter(TestBase $test) {

    // Creates a random filename and updates the source database.
    $random = new Random();
    $temp_directory = $test
      ->getTempFilesDirectory();
    file_prepare_directory($temp_directory, FILE_CREATE_DIRECTORY);
    static::$tempFilename = $test
      ->getDatabasePrefix() . $random
      ->name() . '.jpg';
    $file_path = $temp_directory . '/' . static::$tempFilename;
    file_put_contents($file_path, '');
    $db = Database::getConnection('default', 'migrate');
    $db
      ->update('files')
      ->condition('fid', 6)
      ->fields(array(
      'filename' => static::$tempFilename,
      'filepath' => $file_path,
    ))
      ->execute();
    $file = (array) $db
      ->select('files')
      ->fields('files')
      ->condition('fid', 5)
      ->execute()
      ->fetchObject();
    unset($file['fid']);
    $db
      ->insert('files')
      ->fields($file)
      ->execute();
    return static::$tempFilename;
  }

}

Classes

Name	Description
MigrateFileTest	file migration.