X-Git-Url: http://aleph1.co.uk/gitweb/?a=blobdiff_plain;f=web%2Fcore%2Fmodules%2Fcontent_moderation%2Ftests%2Fsrc%2FFunctional%2FModerationStateAccessTest.php;h=86269193e3f67dcfd99b289512ed72c8efd9fe1e;hb=4f1b9b4ab48a8498afac9e2213a02a23ccf4a06c;hp=74c457b0bf9752608d58936494d032b2f70b232d;hpb=9917807b03b64faf00f6a1f29dcb6eafc454efa5;p=yaffs-website

diff --git a/web/core/modules/content_moderation/tests/src/Functional/ModerationStateAccessTest.php b/web/core/modules/content_moderation/tests/src/Functional/ModerationStateAccessTest.php
index 74c457b0b..86269193e 100644
--- a/web/core/modules/content_moderation/tests/src/Functional/ModerationStateAccessTest.php
+++ b/web/core/modules/content_moderation/tests/src/Functional/ModerationStateAccessTest.php
@@ -5,7 +5,7 @@ namespace Drupal\Tests\content_moderation\Functional;
 use Drupal\node\Entity\Node;
 use Drupal\node\Entity\NodeType;
 use Drupal\Tests\BrowserTestBase;
-use Drupal\workflows\Entity\Workflow;
+use Drupal\Tests\content_moderation\Traits\ContentModerationTestTrait;
 
 /**
  * Tests the view access control handler for moderation state entities.
@@ -14,6 +14,8 @@ use Drupal\workflows\Entity\Workflow;
  */
 class ModerationStateAccessTest extends BrowserTestBase {
 
+  use ContentModerationTestTrait;
+
   /**
    * {@inheritdoc}
    */
@@ -99,7 +101,7 @@ class ModerationStateAccessTest extends BrowserTestBase {
     ]);
     $node_type->save();
 
-    $workflow = Workflow::load('editorial');
+    $workflow = $this->createEditorialWorkflow();
     $workflow->getTypePlugin()->addEntityTypeAndBundle('node', $machine_name);
     $workflow->save();
     return $node_type;