comparison core/modules/system/src/Tests/Entity/EntityDefinitionTestTrait.php @ 14:1fec387a4317

Update Drupal core to 8.5.2 via Composer

   /**
    * Adds a new base field to the 'entity_test_update' entity type.
    *
    * @param string $type
    *   (optional) The field type for the new field. Defaults to 'string'.
-   */
-  protected function addBaseField($type = 'string') {
+   * @param string $entity_type_id
+   *   (optional) The entity type ID the base field should be attached to.
+   *   Defaults to 'entity_test_update'.
+   * @param bool $is_revisionable
+   *   (optional) If the base field should be revisionable or not. Defaults to
+   *   FALSE.
+   */
+  protected function addBaseField($type = 'string', $entity_type_id = 'entity_test_update', $is_revisionable = FALSE) {
     $definitions['new_base_field'] = BaseFieldDefinition::create($type)
       ->setName('new_base_field')
+      ->setRevisionable($is_revisionable)
       ->setLabel(t('A new base field'));
-    $this->state->set('entity_test_update.additional_base_field_definitions', $definitions);
+    $this->state->set($entity_type_id . '.additional_base_field_definitions', $definitions);
   }
 
   /**
    * Adds a long-named base field to the 'entity_test_update' entity type.
    */

     $this->state->set('entity_test_update.entity_type', $entity_type);
   }
 
   /**
    * Removes the new base field from the 'entity_test_update' entity type.
-   */
-  protected function removeBaseField() {
-    $this->state->delete('entity_test_update.additional_base_field_definitions');
+   *
+   * @param string $entity_type_id
+   *   (optional) The entity type ID the base field should be attached to.
+   */
+  protected function removeBaseField($entity_type_id = 'entity_test_update') {
+    $this->state->delete($entity_type_id . '.additional_base_field_definitions');
   }
 
   /**
    * Adds a single-field index to the base field.
    */