function entity_schema_test_entity_type_alter

Implements hook_entity_type_alter().

File

core/modules/system/tests/modules/entity_schema_test/entity_schema_test.module, line 18

Code

function entity_schema_test_entity_type_alter(array &$entity_types) {
  // Allow a test to tell us whether or not to alter the entity type.
  if (\Drupal::state()->get('entity_schema_update')) {
    $entity_type = $entity_types['entity_test_update'];
    if ($entity_type instanceof ContentEntityTypeInterface) {
      $entity_type->set('translatable', TRUE);
      $entity_type->set('data_table', 'entity_test_update_data');
      // Update the keys with a revision ID.
      $keys = $entity_type->getKeys();
      $keys['revision'] = 'revision_id';
      $entity_type->set('entity_keys', $keys);
      // We can't use getRevisionMetadataKeys() because it causes recursion.
      $revision_metadata_keys = $entity_type->get('revision_metadata_keys');
      $revision_metadata_keys = array_merge($revision_metadata_keys, [
        'revision_log_message' => 'revision_log',
      ]);
      $entity_type->set('revision_metadata_keys', $revision_metadata_keys);
    }
  }
}