X-Git-Url: http://www.aleph1.co.uk/gitweb/?a=blobdiff_plain;f=web%2Fcore%2Fmodules%2Fmenu_ui%2Ftests%2Fsrc%2FFunctional%2FMenuUiContentModerationTest.php;h=b5d4b43c8224dc8824cc2bbfb58855bf1d8641e2;hb=0bf8d09d2542548982e81a441b1f16e75873a04f;hp=be018f4ce113ed2beda8e067bd5006ab42805c48;hpb=9917807b03b64faf00f6a1f29dcb6eafc454efa5;p=yaffs-website

diff --git a/web/core/modules/menu_ui/tests/src/Functional/MenuUiContentModerationTest.php b/web/core/modules/menu_ui/tests/src/Functional/MenuUiContentModerationTest.php
index be018f4ce..b5d4b43c8 100644
--- a/web/core/modules/menu_ui/tests/src/Functional/MenuUiContentModerationTest.php
+++ b/web/core/modules/menu_ui/tests/src/Functional/MenuUiContentModerationTest.php
@@ -3,7 +3,7 @@
 namespace Drupal\Tests\menu_ui\Functional;
 
 use Drupal\Tests\BrowserTestBase;
-use Drupal\workflows\Entity\Workflow;
+use Drupal\Tests\content_moderation\Traits\ContentModerationTestTrait;
 
 /**
  * Tests Menu UI and Content Moderation integration.
@@ -12,6 +12,8 @@ use Drupal\workflows\Entity\Workflow;
  */
 class MenuUiContentModerationTest extends BrowserTestBase {
 
+  use ContentModerationTestTrait;
+
   /**
    * Modules to install.
    *
@@ -34,7 +36,7 @@ class MenuUiContentModerationTest extends BrowserTestBase {
       'display_submitted' => FALSE,
     ]);
 
-    $workflow = Workflow::load('editorial');
+    $workflow = $this->createEditorialWorkflow();
     $workflow->getTypePlugin()->addEntityTypeAndBundle('node', 'page');
     $workflow->save();
   }