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

diff --git a/web/core/modules/content_moderation/tests/src/Kernel/ModerationStateWidgetTest.php b/web/core/modules/content_moderation/tests/src/Kernel/ModerationStateWidgetTest.php
index c40e75395..f09496438 100644
--- a/web/core/modules/content_moderation/tests/src/Kernel/ModerationStateWidgetTest.php
+++ b/web/core/modules/content_moderation/tests/src/Kernel/ModerationStateWidgetTest.php
@@ -8,7 +8,7 @@ use Drupal\Core\Form\FormState;
 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\Plugin\Field\FieldWidget\ModerationStateWidget
@@ -16,6 +16,8 @@ use Drupal\workflows\Entity\Workflow;
  */
 class ModerationStateWidgetTest extends KernelTestBase {
 
+  use ContentModerationTestTrait;
+
   /**
    * Modules to install.
    *
@@ -46,7 +48,7 @@ class ModerationStateWidgetTest extends KernelTestBase {
       'type' => 'unmoderated',
     ])->save();
 
-    $workflow = Workflow::load('editorial');
+    $workflow = $this->createEditorialWorkflow();
     $workflow->getTypePlugin()->addEntityTypeAndBundle('node', 'moderated');
     $workflow->save();
   }