public function testEnableWithExistingContent()
{
foreach ($this->entityTypes as $entity_type_id => $values) {
$storage = \Drupal::entityTypeManager()->getStorage($entity_type_id);
if ($entity_type_id == 'user') {
$this->createUser(['administer nodes']);
// There should now be 3 users in total, including the initial anonymous
// and admin users.
$count = 3;
} else {
$count = 2;
for ($i = 0; $i < $count; $i++) {
$storage->create($values)->save();
}
}
$count_before[$entity_type_id] = $count;
}
// Installing Multiversion will trigger the migration of existing content.
$this->moduleInstaller->install(['multiversion']);
$this->multiversionManager = \Drupal::service('multiversion.manager');
// Check if all updates have been applied.
$this->assertFalse(\Drupal::service('entity.definition_update_manager')->needsUpdates(), 'All compatible entity types have been updated.');
$ids_after = [];
// Now check that the previously created entities still exist, have the
// right IDs and are multiversion enabled. That means profit. Big profit.
foreach ($this->entityTypes as $entity_type_id => $values) {
$manager = \Drupal::entityTypeManager();
$entity_type = $manager->getDefinition($entity_type_id);
$storage = $manager->getStorage($entity_type_id);
$id_key = $entity_type->getKey('id');
$this->assertTrue($this->multiversionManager->isEnabledEntityType($entity_type), "{$entity_type_id} was enabled for Multiversion.");
$this->assertTrue($storage instanceof ContentEntityStorageInterface, "{$entity_type_id} got the correct storage handler assigned.");
$this->assertTrue($storage->getQuery() instanceof QueryInterface, "{$entity_type_id} got the correct query handler assigned.");
$ids_after[$entity_type_id] = $storage->getQuery()->execute();
$this->assertEqual($count_before[$entity_type_id], count($ids_after[$entity_type_id]), "All {$entity_type_id}s were migrated.");
foreach ($ids_after[$entity_type_id] as $revision_id => $entity_id) {
$rev = (int) $storage->getQuery()->condition($id_key, $entity_id)->condition('_rev', 'NULL', '<>')->count()->execute();
$workspace = (int) $storage->getQuery()->condition($id_key, $entity_id)->condition('workspace', 1)->count()->execute();
$deleted = (int) $storage->getQuery()->condition($id_key, $entity_id)->condition('_deleted', 0)->count()->execute();
$this->assertEqual($rev, 1, "{$entity_type_id} {$entity_id} has a revision hash in database");
$this->assertEqual($workspace, 1, "{$entity_type_id} {$entity_id} has correct workspace in database");
$this->assertEqual($deleted, 1, "{$entity_type_id} {$entity_id} is not marked as deleted in database");
}
}
// Now install a module with an entity type AFTER the migration and assert
// that is being returned as supported and enabled.
$this->moduleInstaller->install(['taxonomy']);
$entity_type = \Drupal::entityTypeManager()->getDefinition('taxonomy_term');
$this->assertTrue($this->multiversionManager->isEnabledEntityType($entity_type), 'Newly installed entity types gets enabled as well.');
$this->assertFalse(\Drupal::service('entity.definition_update_manager')->needsUpdates(), 'There are not new updates to apply.');
}
/**
* @param \Drupal\Core\Entity\ContentEntityInterface $entity
* @return array
*/
protected function buildRecord(ContentEntityInterface $entity)
{
return array('entity_type_id' => $entity->getEntityTypeId(), 'entity_id' => $entity->id(), 'entity_uuid' => $entity->uuid(), 'revision_id' => $entity->getRevisionId(), 'deleted' => $entity->_deleted->value, 'rev' => $entity->_rev->value, 'seq' => $this->multiversionManager->newSequenceId(), 'local' => (bool) $entity->getEntityType()->get('local'), 'is_stub' => (bool) $entity->_rev->is_stub);
}
