function SqlContentEntityStorageSchemaTest::testRequiresEntityStorageSchemaChanges

@covers ::requiresEntityStorageSchemaChanges

@dataProvider providerTestRequiresEntityStorageSchemaChanges

File

core/tests/Drupal/Tests/Core/Entity/Sql/SqlContentEntityStorageSchemaTest.php, line 1259

Class

SqlContentEntityStorageSchemaTest

@coversDefaultClass \Drupal\Core\Entity\Sql\SqlContentEntityStorageSchema[[api-linebreak]] @group Entity

Namespace

Code

public function testRequiresEntityStorageSchemaChanges(bool $requires_change, bool $change_schema, bool $change_shared_table, ?string $method = NULL) : void {
  $original = $this->createMock(ContentEntityTypeInterface::class);
  $original->expects($this->any())
    ->method('id')
    ->willReturn('entity_test');
  $original->expects($this->any())
    ->method('getKey')
    ->willReturn('id');
  $updated = $this->createMock(ContentEntityTypeInterface::class);
  $updated->expects($this->any())
    ->method('id')
    ->willReturn('entity_test');
  $updated->expects($this->any())
    ->method('getKey')
    ->willReturn('id');
  if ($method) {
    $original->expects($this->once())
      ->method($method)
      ->willReturn(TRUE);
    $updated->expects($this->once())
      ->method($method)
      ->willReturn(FALSE);
  }
  // Storage class changes should not impact this at all, and should not be
  // checked.
  $original->expects($this->never())
    ->method('getStorageClass');
  $updated->expects($this->never())
    ->method('getStorageClass');
  $this->entityType = new ContentEntityType([
    'id' => 'entity_test',
    'entity_keys' => [
      'id' => 'id',
    ],
  ]);
  $this->setUpStorageSchema();
  $table_mapping = new TestSqlContentDefaultTableMapping($this->entityType, $this->storageDefinitions);
  $table_mapping->setFieldNames('entity_test', array_keys($this->storageDefinitions));
  $table_mapping->setExtraColumns('entity_test', [
    'default_langcode',
  ]);
  $this->storageSchema
    ->expects($this->any())
    ->method('getTableMapping')
    ->willReturn($table_mapping);
  // Setup storage schema.
  if ($change_schema) {
    $this->storageSchema
      ->expects($this->once())
      ->method('loadEntitySchemaData')
      ->willReturn([]);
  }
  else {
    $expected = [
      'entity_test' => [
        'primary key' => [
          'id',
        ],
      ],
    ];
    $this->storageSchema
      ->expects($this->any())
      ->method('loadEntitySchemaData')
      ->willReturn($expected);
  }
  if ($change_shared_table) {
    $this->storageSchema
      ->expects($this->once())
      ->method('hasSharedTableNameChanges')
      ->willReturn(TRUE);
  }
  $this->assertEquals($requires_change, $this->storageSchema
    ->requiresEntityStorageSchemaChanges($updated, $original));
}