Version 1

[yaffs-website] / web / modules / contrib / video_embed_field / tests / src / Kernel / EntityLoadTrait.php

diff --git a/web/modules/contrib/video_embed_field/tests/src/Kernel/EntityLoadTrait.php b/web/modules/contrib/video_embed_field/tests/src/Kernel/EntityLoadTrait.php
new file mode 100644 (file)
index 0000000..2089e89
--- /dev/null
+++ b/web/modules/contrib/video_embed_field/tests/src/Kernel/EntityLoadTrait.php
@@ -0,0 +1,28 @@
+<?php
+
+namespace Drupal\Tests\video_embed_field\Kernel;
+
+/**
+ * Test helpers for loading entities for tests.
+ */
+trait EntityLoadTrait {
+
+  /**
+   * Load an entity by it's label.
+   *
+   * @param string $label
+   *   The label of the entity to load.
+   * @param string $entity_type
+   *   The entity type to load.
+   *
+   * @return \Drupal\Core\Entity\EntityInterface
+   *   A loaded entity.
+   */
+  protected function loadEntityByLabel($label, $entity_type = 'node') {
+    $type_manager = \Drupal::entityTypeManager();
+    $label_key = $type_manager->getDefinition($entity_type)->getKey('label');
+    $entities = \Drupal::entityQuery($entity_type)->condition($label_key, $label, '=')->execute();
+    return $type_manager->getStorage($entity_type)->load(array_shift($entities));
+  }
+
+}