diff -u b/core/modules/migrate_drupal/tests/src/Traits/CreateTestContentEntitiesTrait.php b/core/modules/migrate_drupal/tests/src/Traits/CreateTestContentEntitiesTrait.php --- b/core/modules/migrate_drupal/tests/src/Traits/CreateTestContentEntitiesTrait.php +++ b/core/modules/migrate_drupal/tests/src/Traits/CreateTestContentEntitiesTrait.php @@ -60,129 +60,164 @@ * Create several pieces of generic content. */ protected function createContent() { + $entityTypeManager = \Drupal::entityTypeManager(); + // Create an aggregator feed. - $feed = Feed::create([ - 'title' => 'feed', - 'url' => 'http://www.example.com', - ]); - $feed->save(); + if ($entityTypeManager->hasDefinition('aggregator_feed')) { + $feed = Feed::create([ + 'title' => 'feed', + 'url' => 'http://www.example.com', + ]); + $feed->save(); + } // Create an aggregator feed item. - $item = Item::create([ - 'title' => 'feed item', - 'fid' => $feed->id(), - 'link' => 'http://www.example.com', - ]); - $item->save(); + if ($entityTypeManager->hasDefinition('aggregator_item')) { + $item = Item::create([ + 'title' => 'feed item', + 'fid' => $feed->id(), + 'link' => 'http://www.example.com', + ]); + $item->save(); + } // Create a block content. - $block = BlockContent::create([ - 'info' => 'block', - 'type' => 'block', - ]); - $block->save(); + if ($entityTypeManager->hasDefinition('block_content')) { + $block = BlockContent::create([ + 'info' => 'block', + 'type' => 'block', + ]); + $block->save(); + } // Create a node. - $node = Node::create([ - 'type' => 'page', - 'title' => 'page', - ]); - $node->save(); + if ($entityTypeManager->hasDefinition('node')) { + $node = Node::create([ + 'type' => 'page', + 'title' => 'page', + ]); + $node->save(); + } // Create a comment. - $comment = Comment::create([ - 'comment_type' => 'comment', - 'field_name' => 'comment', - 'entity_type' => 'node', - 'entity_id' => $node->id(), - ]); - $comment->save(); + if ($entityTypeManager->hasDefinition('comment')) { + $comment = Comment::create([ + 'comment_type' => 'comment', + 'field_name' => 'comment', + 'entity_type' => 'node', + 'entity_id' => $node->id(), + ]); + $comment->save(); + } // Create a file. - $file = File::create([ - 'uri' => 'public://example.txt', - ]); - $file->save(); + if ($entityTypeManager->hasDefinition('file')) { + $file = File::create([ + 'uri' => 'public://example.txt', + ]); + $file->save(); + } // Create a menu link. - $menu_link = MenuLinkContent::create([ - 'title' => 'menu link', - 'link' => ['uri' => 'http://www.example.com'], - 'menu_name' => 'tools', - ]); - $menu_link->save(); + if ($entityTypeManager->hasDefinition('menu_link_content')) { + $menu_link = MenuLinkContent::create([ + 'title' => 'menu link', + 'link' => ['uri' => 'http://www.example.com'], + 'menu_name' => 'tools', + ]); + $menu_link->save(); + } // Create a taxonomy term. - $term = Term::create([ - 'name' => 'term', - 'vid' => 'term', - ]); - $term->save(); + if ($entityTypeManager->hasDefinition('taxonomy_term')) { + $term = Term::create([ + 'name' => 'term', + 'vid' => 'term', + ]); + $term->save(); + } // Create a user. - $user = User::create([ - 'uid' => 2, - 'name' => 'user', - 'mail' => 'user@example.com', - ]); - $user->save(); + if ($entityTypeManager->hasDefinition('user')) { + $user = User::create([ + 'uid' => 2, + 'name' => 'user', + 'mail' => 'user@example.com', + ]); + $user->save(); + } } /** * Create several pieces of generic content. */ protected function createContentPostUpgrade() { + $entityTypeManager = \Drupal::entityTypeManager(); + // Create a block content. - $block = BlockContent::create([ - 'info' => 'Post upgrade block', - 'type' => 'block', - ]); - $block->save(); + if ($entityTypeManager->hasDefinition('block_content')) { + $block = BlockContent::create([ + 'info' => 'Post upgrade block', + 'type' => 'block', + ]); + $block->save(); + } // Create a node. - $node = Node::create([ - 'type' => 'page', - 'title' => 'Post upgrade page', - ]); - $node->save(); + if ($entityTypeManager->hasDefinition('node')) { + $node = Node::create([ + 'type' => 'page', + 'title' => 'Post upgrade page', + ]); + $node->save(); + } // Create a comment. - $comment = Comment::create([ - 'comment_type' => 'comment', - 'field_name' => 'comment', - 'entity_type' => 'node', - 'entity_id' => $node->id(), - ]); - $comment->save(); - + if ($entityTypeManager->hasDefinition('comment')) { + $comment = Comment::create([ + 'comment_type' => 'comment', + 'field_name' => 'comment', + 'entity_type' => 'node', + 'entity_id' => $node->id(), + ]); + $comment->save(); + } // Create a file. - $file = File::create([ - 'uri' => 'public://post_upgrade_example.txt', - ]); - $file->save(); + if ($entityTypeManager->hasDefinition('file')) { + $file = File::create([ + 'uri' => 'public://post_upgrade_example.txt', + ]); + $file->save(); + } // Create a menu link. - $menu_link = MenuLinkContent::create([ - 'title' => 'post upgrade menu link', - 'link' => ['uri' => 'http://www.drupal.org'], - 'menu_name' => 'tools', - ]); - $menu_link->save(); + if ($entityTypeManager->hasDefinition('menu_link_content')) { + $menu_link = MenuLinkContent::create([ + 'title' => 'post upgrade menu link', + 'link' => ['uri' => 'http://www.drupal.org'], + 'menu_name' => 'tools', + ]); + $menu_link->save(); + } // Create a taxonomy term. - $term = Term::create([ - 'name' => 'post upgrade term', - 'vid' => 'term', - ]); - $term->save(); + if ($entityTypeManager->hasDefinition('taxonomy_term')) { + $term = Term::create([ + 'name' => 'post upgrade term', + 'vid' => 'term', + ]); + $term->save(); + } // Create a user. - $user = User::create([ - 'uid' => 42, - 'name' => 'universe', - 'mail' => 'universe@example.com', - ]); - $user->save(); + if ($entityTypeManager->hasDefinition('user')) { + $user = User::create([ + 'uid' => 42, + 'name' => 'universe', + 'mail' => 'universe@example.com', + ]); + $user->save(); + } } }