class BulkDeleteTest

class BulkDeleteTest extends FieldKernelTestBase {
  
  /**
   * The fields to use in this test.
   *
   * @var array
   */
  protected $fieldStorages;
  
  /**
   * The entities to use in this test.
   *
   * @var array
   */
  protected $entities;
  
  /**
   * The entities to use in this test, keyed by bundle.
   *
   * @var array
   */
  protected $entitiesByBundles;
  
  /**
   * The bundles for the entities used in this test.
   *
   * @var array
   */
  protected $bundles;
  
  /**
   * The entity type to be used in the test classes.
   *
   * @var string
   */
  protected $entityTypeId = 'entity_test';
  
  /**
   * Tests that the expected hooks have been invoked on the expected entities.
   *
   * @param string[] $expected_hooks
   *   An array keyed by hook name, with one entry per expected invocation.
   *   Each entry is the value of the "$entity" parameter the hook is expected
   *   to have been passed.
   * @param array $actual_hooks
   *   The array of actual hook invocations recorded by
   *   FieldTestHelper::memorize().
   */
  public function checkHooksInvocations($expected_hooks, array $actual_hooks) : void {
    foreach ($expected_hooks as $hook => $invocations) {
      $actual_invocations = $actual_hooks[$hook];
      // Check that the number of invocations is correct.
      $this->assertSameSize($invocations, $actual_invocations, "{$hook}() was called the expected number of times.");
      // Check that the hook was called for each expected argument.
      foreach ($invocations as $argument) {
        $found = FALSE;
        foreach ($actual_invocations as $actual_arguments) {
          // The argument we are looking for is either an array of entities as
          // the second argument or a single entity object as the first.
          if ($argument instanceof EntityInterface && $actual_arguments[0]->id() == $argument->id()) {
            $found = TRUE;
            break;

          }
          elseif (is_array($argument) && count($actual_arguments[1]) == count($argument) && count(array_diff_key($actual_arguments[1], $argument)) == 0) {
            $found = TRUE;
            break;

          }
        }
        $this->assertTrue($found, "{$hook}() was called on expected argument");
      }
    }
  }
  
  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $this->fieldStorages = [];
    $this->entities = [];
    $this->entitiesByBundles = [];
    // Create two bundles.
    $this->bundles = [
      'bb_1' => 'bb_1',
      'bb_2' => 'bb_2',
    ];
    foreach ($this->bundles as $name => $desc) {
      EntityTestHelper::createBundle($name, $desc);
    }
    // Create two field storages.
    $field_storage = FieldStorageConfig::create([
      'field_name' => 'bf_1',
      'entity_type' => $this->entityTypeId,
      'type' => 'test_field',
      'cardinality' => 1,
    ]);
    $field_storage->save();
    $this->fieldStorages[] = $field_storage;
    $field_storage = FieldStorageConfig::create([
      'field_name' => 'bf_2',
      'entity_type' => $this->entityTypeId,
      'type' => 'test_field',
      'cardinality' => 4,
    ]);
    $field_storage->save();
    $this->fieldStorages[] = $field_storage;
    // For each bundle, create each field, and 10 entities with values for the
    // fields.
    foreach ($this->bundles as $bundle) {
      foreach ($this->fieldStorages as $field_storage) {
        FieldConfig::create([
          'field_storage' => $field_storage,
          'bundle' => $bundle,
        ])->save();
      }
      for ($i = 0; $i < 10; $i++) {
        $entity = $this->container
          ->get('entity_type.manager')
          ->getStorage($this->entityTypeId)
          ->create([
          'type' => $bundle,
        ]);
        foreach ($this->fieldStorages as $field_storage) {
          $entity->{$field_storage->getName()}
            ->setValue($this->_generateTestFieldValues($field_storage->getCardinality()));
        }
        $entity->save();
      }
    }
    $this->entities = $this->container
      ->get('entity_type.manager')
      ->getStorage($this->entityTypeId)
      ->loadMultiple();
    foreach ($this->entities as $entity) {
      // This test relies on the entities having stale field definitions
      // so that the deleted field can be accessed on them. Access the field
      // now, so that they are always loaded.
      $entity->bf_1->value;
      // Also keep track of the entities per bundle.
      $this->entitiesByBundles[$entity->bundle()][$entity->id()] = $entity;
    }
  }
  
  /**
   * Tests deleting fields.
   *
   * Verify that deleting a field leaves the field data items in the database
   * and that the appropriate Field API functions can operate on the deleted
   * data and field definition.
   *
   * This tests how EntityFieldQuery interacts with field deletion and could be
   * moved to FieldCrudTestCase, but depends on this class's setUp().
   */
  public function testDeleteField() : void {
    $bundle = reset($this->bundles);
    $field_storage = reset($this->fieldStorages);
    $field_name = $field_storage->getName();
    $storage = \Drupal::entityTypeManager()->getStorage('entity_test');
    // There are 10 entities of this bundle.
    $found = $storage->getQuery()
      ->accessCheck(FALSE)
      ->condition('type', $bundle)
      ->execute();
    $this->assertCount(10, $found, 'Correct number of entities found before deleting');
    // Delete the field.
    $field = FieldConfig::loadByName($this->entityTypeId, $bundle, $field_name);
    $field->delete();
    // The field still exists, deleted.
    $fields = \Drupal::entityTypeManager()->getStorage('field_config')
      ->loadByProperties([
      'field_storage_uuid' => $field_storage->uuid(),
      'deleted' => TRUE,
      'include_deleted' => TRUE,
    ]);
    $this->assertCount(1, $fields, 'There is one deleted field');
    $field = $fields[$field->uuid()];
    $this->assertEquals($bundle, $field->getTargetBundle(), 'The deleted field is for the correct bundle');
    // Check that the actual stored content did not change during delete.
    /** @var \Drupal\Core\Entity\Sql\DefaultTableMapping $table_mapping */
    $table_mapping = $storage->getTableMapping();
    $table = $table_mapping->getDedicatedDataTableName($field_storage);
    $column = $table_mapping->getFieldColumnName($field_storage, 'value');
    $result = Database::getConnection()->select($table, 't')
      ->fields('t')
      ->execute();
    foreach ($result as $row) {
      $this->assertEquals($row->{$column}, $this->entities[$row->entity_id]->{$field_name}->value);
    }
    // There are 0 entities of this bundle with non-deleted data.
    $found = $storage->getQuery()
      ->accessCheck(FALSE)
      ->condition('type', $bundle)
      ->condition("{$field_name}.deleted", 0)
      ->execute();
    $this->assertEmpty($found, 'No entities found after deleting');
    // There are 10 entities of this bundle when deleted fields are allowed, and
    // their values are correct.
    $found = $storage->getQuery()
      ->accessCheck(FALSE)
      ->condition('type', $bundle)
      ->condition("{$field_name}.deleted", 1)
      ->sort('id')
      ->execute();
    $this->assertCount(10, $found, 'Correct number of entities found after deleting');
    $this->assertEmpty(array_diff($found, array_keys($this->entities)));
  }
  
  /**
   * Tests that recreating a field with the name as a deleted field works.
   */
  public function testPurgeWithDeletedAndActiveField() : void {
    $bundle = reset($this->bundles);
    // Create another field storage.
    $field_name = 'bf_3';
    $deleted_field_storage = FieldStorageConfig::create([
      'field_name' => $field_name,
      'entity_type' => $this->entityTypeId,
      'type' => 'test_field',
      'cardinality' => 1,
    ]);
    $deleted_field_storage->save();
    // Create the field.
    FieldConfig::create([
      'field_storage' => $deleted_field_storage,
      'bundle' => $bundle,
    ])->save();
    for ($i = 0; $i < 20; $i++) {
      $entity = $this->container
        ->get('entity_type.manager')
        ->getStorage($this->entityTypeId)
        ->create([
        'type' => $bundle,
      ]);
      $entity->{$field_name}
        ->setValue($this->_generateTestFieldValues(1));
      $entity->save();
    }
    // Delete the field.
    $deleted_field = FieldConfig::loadByName($this->entityTypeId, $bundle, $field_name);
    $deleted_field->delete();
    $deleted_field_uuid = $deleted_field->uuid();
    // Reload the field storage.
    $field_storages = \Drupal::entityTypeManager()->getStorage('field_storage_config')
      ->loadByProperties([
      'uuid' => $deleted_field_storage->uuid(),
      'include_deleted' => TRUE,
    ]);
    $deleted_field_storage = reset($field_storages);
    // Create the field again.
    $field_storage = FieldStorageConfig::create([
      'field_name' => $field_name,
      'entity_type' => $this->entityTypeId,
      'type' => 'test_field',
      'cardinality' => 1,
    ]);
    $field_storage->save();
    FieldConfig::create([
      'field_storage' => $field_storage,
      'bundle' => $bundle,
    ])->save();
    // The field still exists, deleted, with the same field name.
    $fields = \Drupal::entityTypeManager()->getStorage('field_config')
      ->loadByProperties([
      'uuid' => $deleted_field_uuid,
      'include_deleted' => TRUE,
    ]);
    $this->assertArrayHasKey($deleted_field_uuid, $fields);
    $this->assertTrue($fields[$deleted_field_uuid]->isDeleted());
    $this->assertSame($field_name, $fields[$deleted_field_uuid]->getName());
    for ($i = 0; $i < 10; $i++) {
      $entity = $this->container
        ->get('entity_type.manager')
        ->getStorage($this->entityTypeId)
        ->create([
        'type' => $bundle,
      ]);
      $entity->{$field_name}
        ->setValue($this->_generateTestFieldValues(1));
      $entity->save();
    }
    // Check that the two field storages have different tables.
    $storage = \Drupal::entityTypeManager()->getStorage($this->entityTypeId);
    /** @var \Drupal\Core\Entity\Sql\DefaultTableMapping $table_mapping */
    $table_mapping = $storage->getTableMapping();
    $deleted_table_name = $table_mapping->getDedicatedDataTableName($deleted_field_storage, TRUE);
    $active_table_name = $table_mapping->getDedicatedDataTableName($field_storage);
    \Drupal::service(FieldPurger::class)->purgeBatch(50);
    // Ensure the new field still has its table and the deleted one has been
    // removed.
    $this->assertTrue(\Drupal::database()->schema()
      ->tableExists($active_table_name));
    $this->assertFalse(\Drupal::database()->schema()
      ->tableExists($deleted_table_name));
    // The field has been removed from the system.
    $fields = \Drupal::entityTypeManager()->getStorage('field_config')
      ->loadByProperties([
      'field_storage_uuid' => $deleted_field_storage->uuid(),
      'deleted' => TRUE,
      'include_deleted' => TRUE,
    ]);
    $this->assertCount(0, $fields, 'The field is gone');
    // Verify there are still 10 entries in the main table.
    $count = \Drupal::database()->select('entity_test__' . $field_name, 'f')
      ->fields('f', [
      'entity_id',
    ])
      ->condition('bundle', $bundle)
      ->countQuery()
      ->execute()
      ->fetchField();
    $this->assertEquals(10, $count);
  }
  
  /**
   * Tests purging fields.
   *
   * Verify that field data items and fields are purged when a field storage is
   * deleted.
   */
  public function testPurgeField() : void {
    // Start recording hook invocations.
    FieldTestHelper::memorize();
    $entity_field_purger = \Drupal::service(FieldPurger::class);
    $bundle = reset($this->bundles);
    $field_storage = reset($this->fieldStorages);
    $field_name = $field_storage->getName();
    // Delete the field.
    $field = FieldConfig::loadByName($this->entityTypeId, $bundle, $field_name);
    $field->delete();
    // No field hooks were called.
    $mem = FieldTestHelper::memorize();
    $this->assertCount(0, $mem, 'No field hooks were called');
    $batch_size = 2;
    for ($count = 8; $count >= 0; $count -= $batch_size) {
      // Purge two entities.
      $entity_field_purger->purgeBatch($batch_size);
      // There are $count deleted entities left.
      $found = \Drupal::entityQuery('entity_test')->accessCheck(FALSE)
        ->condition('type', $bundle)
        ->condition($field_name . '.deleted', 1)
        ->execute();
      $this->assertCount($count, $found, 'Correct number of entities found after purging 2');
    }
    // Check hooks invocations.
    // FieldItemInterface::delete() should have been called once for each entity
    // in the bundle.
    $actual_hooks = FieldTestHelper::memorize();
    $hooks = [];
    $hooks['field_test_field_delete'] = $this->entitiesByBundles[$bundle];
    $this->checkHooksInvocations($hooks, $actual_hooks);
    // The field still exists, deleted.
    $fields = \Drupal::entityTypeManager()->getStorage('field_config')
      ->loadByProperties([
      'field_storage_uuid' => $field_storage->uuid(),
      'deleted' => TRUE,
      'include_deleted' => TRUE,
    ]);
    $this->assertCount(1, $fields, 'There is one deleted field');
    // Purge the field.
    $entity_field_purger->purgeBatch($batch_size);
    // The field is gone.
    $fields = \Drupal::entityTypeManager()->getStorage('field_config')
      ->loadByProperties([
      'field_storage_uuid' => $field_storage->uuid(),
      'deleted' => TRUE,
      'include_deleted' => TRUE,
    ]);
    $this->assertCount(0, $fields, 'The field is gone');
    // The field storage still exists, not deleted, because it has a second
    // field.
    $storages = \Drupal::entityTypeManager()->getStorage('field_storage_config')
      ->loadByProperties([
      'uuid' => $field_storage->uuid(),
      'include_deleted' => TRUE,
    ]);
    $this->assertTrue(isset($storages[$field_storage->uuid()]), 'The field storage exists and is not deleted');
  }
  
  /**
   * Tests purging field storages.
   *
   * Verify that field storages are preserved and purged correctly as multiple
   * fields are deleted and purged.
   */
  public function testPurgeFieldStorage() : void {
    // Start recording hook invocations.
    FieldTestHelper::memorize();
    $entity_field_purger = \Drupal::service(FieldPurger::class);
    $field_storage = reset($this->fieldStorages);
    $field_name = $field_storage->getName();
    // Delete the first field.
    $bundle = reset($this->bundles);
    $field = FieldConfig::loadByName($this->entityTypeId, $bundle, $field_name);
    $field->delete();
    // Assert that FieldItemInterface::delete() was not called yet.
    $mem = FieldTestHelper::memorize();
    $this->assertCount(0, $mem, 'No field hooks were called.');
    // Purge the data.
    $entity_field_purger->purgeBatch(10);
    // Check hooks invocations.
    // FieldItemInterface::delete() should have been called once for each entity in the
    // bundle.
    $actual_hooks = FieldTestHelper::memorize();
    $hooks = [];
    $hooks['field_test_field_delete'] = $this->entitiesByBundles[$bundle];
    $this->checkHooksInvocations($hooks, $actual_hooks);
    // The field still exists, deleted.
    $fields = \Drupal::entityTypeManager()->getStorage('field_config')
      ->loadByProperties([
      'uuid' => $field->uuid(),
      'include_deleted' => TRUE,
    ]);
    $this->assertArrayHasKey($field->uuid(), $fields);
    $this->assertTrue($fields[$field->uuid()]
      ->isDeleted());
    // Purge again to purge the field.
    $entity_field_purger->purgeBatch(0);
    // The field is gone.
    $fields = \Drupal::entityTypeManager()->getStorage('field_config')
      ->loadByProperties([
      'uuid' => $field->uuid(),
      'include_deleted' => TRUE,
    ]);
    $this->assertCount(0, $fields, 'The field is purged.');
    // The field storage still exists, not deleted.
    $storages = \Drupal::entityTypeManager()->getStorage('field_storage_config')
      ->loadByProperties([
      'uuid' => $field_storage->uuid(),
      'include_deleted' => TRUE,
    ]);
    $this->assertArrayHasKey($field_storage->uuid(), $storages);
    $this->assertFalse($storages[$field_storage->uuid()]
      ->isDeleted());
    // Delete the second field.
    $bundle = next($this->bundles);
    $field = FieldConfig::loadByName($this->entityTypeId, $bundle, $field_name);
    $field->delete();
    // Assert that FieldItemInterface::delete() was not called yet.
    $mem = FieldTestHelper::memorize();
    $this->assertCount(0, $mem, 'No field hooks were called.');
    // Purge the data.
    $entity_field_purger->purgeBatch(10);
    // Check hooks invocations (same as above, for the 2nd bundle).
    $actual_hooks = FieldTestHelper::memorize();
    $hooks = [];
    $hooks['field_test_field_delete'] = $this->entitiesByBundles[$bundle];
    $this->checkHooksInvocations($hooks, $actual_hooks);
    // The field and the storage still exist, deleted.
    $fields = \Drupal::entityTypeManager()->getStorage('field_config')
      ->loadByProperties([
      'uuid' => $field->uuid(),
      'include_deleted' => TRUE,
    ]);
    $this->assertArrayHasKey($field->uuid(), $fields);
    $this->assertTrue($fields[$field->uuid()]
      ->isDeleted());
    $storages = \Drupal::entityTypeManager()->getStorage('field_storage_config')
      ->loadByProperties([
      'uuid' => $field_storage->uuid(),
      'include_deleted' => TRUE,
    ]);
    $this->assertArrayHasKey($field_storage->uuid(), $storages);
    $this->assertTrue($storages[$field_storage->uuid()]
      ->isDeleted());
    // Purge again to purge the field and the storage.
    $entity_field_purger->purgeBatch(0);
    // The field and the storage are gone.
    $fields = \Drupal::entityTypeManager()->getStorage('field_config')
      ->loadByProperties([
      'uuid' => $field->uuid(),
      'include_deleted' => TRUE,
    ]);
    $this->assertCount(0, $fields, 'The field is purged.');
    $storages = \Drupal::entityTypeManager()->getStorage('field_storage_config')
      ->loadByProperties([
      'uuid' => $field_storage->uuid(),
      'include_deleted' => TRUE,
    ]);
    $this->assertCount(0, $storages, 'The field storage is purged.');
  }

}