X-Git-Url: http://www.aleph1.co.uk/gitweb/?a=blobdiff_plain;ds=sidebyside;f=web%2Fcore%2Fmodules%2Fcontent_moderation%2Ftests%2Fsrc%2FKernel%2FEntityOperationsTest.php;fp=web%2Fcore%2Fmodules%2Fcontent_moderation%2Ftests%2Fsrc%2FKernel%2FEntityOperationsTest.php;h=0a93924ab42fdc807ee53379e1ca7d5d87531bf5;hb=0bf8d09d2542548982e81a441b1f16e75873a04f;hp=63f2c9a7321f836c95e8fcca0a8bfd0636cc7203;hpb=74df008bdbb3a11eeea356744f39b802369bda3c;p=yaffs-website

diff --git a/web/core/modules/content_moderation/tests/src/Kernel/EntityOperationsTest.php b/web/core/modules/content_moderation/tests/src/Kernel/EntityOperationsTest.php
index 63f2c9a73..0a93924ab 100644
--- a/web/core/modules/content_moderation/tests/src/Kernel/EntityOperationsTest.php
+++ b/web/core/modules/content_moderation/tests/src/Kernel/EntityOperationsTest.php
@@ -5,7 +5,7 @@ namespace Drupal\Tests\content_moderation\Kernel;
 use Drupal\KernelTests\KernelTestBase;
 use Drupal\node\Entity\Node;
 use Drupal\node\Entity\NodeType;
-use Drupal\workflows\Entity\Workflow;
+use Drupal\Tests\content_moderation\Traits\ContentModerationTestTrait;
 
 /**
  * @coversDefaultClass \Drupal\content_moderation\EntityOperations
@@ -14,6 +14,8 @@ use Drupal\workflows\Entity\Workflow;
  */
 class EntityOperationsTest extends KernelTestBase {
 
+  use ContentModerationTestTrait;
+
   /**
    * {@inheritdoc}
    */
@@ -48,7 +50,7 @@ class EntityOperationsTest extends KernelTestBase {
       'label' => 'Page',
     ]);
     $node_type->save();
-    $workflow = Workflow::load('editorial');
+    $workflow = $this->createEditorialWorkflow();
     $workflow->getTypePlugin()->addEntityTypeAndBundle('node', 'page');
     $workflow->save();
   }