diff --git a/devel_generate/tests/src/Functional/DevelGenerateCommandsTest.php b/devel_generate/tests/src/Functional/DevelGenerateCommandsTest.php
index 9771fa6ab2425dde9346f3a2410972fbf6a7eefa..98ac00aa6021f0ca8d4dda7f4b275fb2d3d06b2f 100644
--- a/devel_generate/tests/src/Functional/DevelGenerateCommandsTest.php
+++ b/devel_generate/tests/src/Functional/DevelGenerateCommandsTest.php
@@ -12,75 +12,92 @@ use Drupal\Tests\devel_generate\Traits\DevelGenerateSetupTrait;
 use Drupal\user\Entity\User;
 use Drush\TestTraits\DrushTestTrait;
 
-/**
- * Note: Drush must be in the Composer project. See https://cgit.drupalcode.org/devel/tree/drupalci.yml?h=8.x-2.x and its docs at
- * https://www.drupal.org/drupalorg/docs/drupal-ci/customizing-drupalci-testing-for-projects
- */
-
-/**
- * @coversDefaultClass \Drupal\devel_generate\Commands\DevelGenerateCommands
- * @group devel-generate
- */
-class DevelGenerateCommandsTest extends BrowserTestBase
-{
-  use DrushTestTrait;
-  use DevelGenerateSetupTrait;
+if (trait_exists('DrushTestTrait')) {
 
   /**
-   * Modules to enable.
-   *
-   * @var array
+   * Note: Drush must be in the Composer project. See https://cgit.drupalcode.org/devel/tree/drupalci.yml?h=8.x-2.x and its docs at
+   * https://www.drupal.org/drupalorg/docs/drupal-ci/customizing-drupalci-testing-for-projects
    */
-  public static $modules = array('menu_ui', 'node', 'comment', 'taxonomy', 'path', 'devel_generate');
 
   /**
-   * Prepares the testing environment
+   * @coversDefaultClass \Drupal\devel_generate\Commands\DevelGenerateCommands
+   * @group devel-generate
    */
-  public function setUp() {
-    parent::setUp();
-    $this->setUpData();
-  }
+  class DevelGenerateCommandsTest extends BrowserTestBase {
+
+    use DrushTestTrait;
+    use DevelGenerateSetupTrait;
 
-  public function testGeneration() {
-    // Make sure users get created, and with correct roles.
-    $this->drush('devel-generate-users', [55], ['kill' => null, 'roles' => 'administrator']);
-    $user = User::load(55);
-    $this->assertTrue($user->hasRole('administrator'));
+    /**
+     * Modules to enable.
+     *
+     * @var array
+     */
+    public static $modules = [
+      'menu_ui',
+      'node',
+      'comment',
+      'taxonomy',
+      'path',
+      'devel_generate'
+    ];
+
+    /**
+     * Prepares the testing environment
+     */
+    public function setUp() {
+      parent::setUp();
+      $this->setUpData();
+    }
 
-    // Make sure terms get created, and with correct vocab.
-    $this->drush('devel-generate-terms', [$this->vocabulary->id(), 55], ['kill' => null]);
-    $term = Term::load(55);
-    $this->assertEquals($this->vocabulary->id(), $term->bundle());
+    public function testGeneration() {
+      // Make sure users get created, and with correct roles.
+      $this->drush('devel-generate-users', [55], [
+        'kill' => NULL,
+        'roles' => 'administrator'
+      ]);
+      $user = User::load(55);
+      $this->assertTrue($user->hasRole('administrator'));
 
-    // Make sure vocabs get created.
-    $this->drush('devel-generate-vocabs', [5], ['kill' => null]);
-    $vocabs = Vocabulary::loadMultiple();
-    $this->assertGreaterThan(4, count($vocabs));
-    $vocab = array_pop($vocabs);
-    $this->assertNotEmpty($vocab);
+      // Make sure terms get created, and with correct vocab.
+      $this->drush('devel-generate-terms', [
+        $this->vocabulary->id(),
+        55
+      ], ['kill' => NULL]);
+      $term = Term::load(55);
+      $this->assertEquals($this->vocabulary->id(), $term->bundle());
 
-    // Make sure menus, and with correct properties.
-    $this->drush('devel-generate-menus', [1, 5], ['kill' => null]);
-    $menus = Menu::loadMultiple();
-    foreach ($menus as $key => $menu) {
-      if (strstr($menu->id(), 'devel-') !== FALSE) {
-        // We have a menu that we created.
-        break;
+      // Make sure vocabs get created.
+      $this->drush('devel-generate-vocabs', [5], ['kill' => NULL]);
+      $vocabs = Vocabulary::loadMultiple();
+      $this->assertGreaterThan(4, count($vocabs));
+      $vocab = array_pop($vocabs);
+      $this->assertNotEmpty($vocab);
+
+      // Make sure menus, and with correct properties.
+      $this->drush('devel-generate-menus', [1, 5], ['kill' => NULL]);
+      $menus = Menu::loadMultiple();
+      foreach ($menus as $key => $menu) {
+        if (strstr($menu->id(), 'devel-') !== FALSE) {
+          // We have a menu that we created.
+          break;
+        }
       }
-    }
-    $link = MenuLinkContent::load(5);
-    $this->assertEquals($menu->id(), $link->getMenuName());
+      $link = MenuLinkContent::load(5);
+      $this->assertEquals($menu->id(), $link->getMenuName());
 
-    // Make sure content gets created, with comments.
-    $this->drush('devel-generate-content', [21, 9], ['kill' => null]);
-    $node = Node::load(3);
-    $this->assertNotEmpty($node);
-    $comment = Comment::load(1);
-    $this->assertNotEmpty($comment);
+      // Make sure content gets created, with comments.
+      $this->drush('devel-generate-content', [21, 9], ['kill' => NULL]);
+      $node = Node::load(3);
+      $this->assertNotEmpty($node);
+      $comment = Comment::load(1);
+      $this->assertNotEmpty($comment);
 
-    // Do same, but with a higher number that triggers batch running.
-    $this->drush('devel-generate-content', [55], ['kill' => null]);
-    $node = Node::load(55);
-    $this->assertNotEmpty($node);
+      // Do same, but with a higher number that triggers batch running.
+      $this->drush('devel-generate-content', [55], ['kill' => NULL]);
+      $node = Node::load(55);
+      $this->assertNotEmpty($node);
+    }
   }
+
 }
diff --git a/tests/src/Functional/DevelCommandsTest.php b/tests/src/Functional/DevelCommandsTest.php
index 1004ae49d3614337b5249f06f215de948a1ffd16..aac6b4358b9a2832afacedad52f855ddeafecc7f 100644
--- a/tests/src/Functional/DevelCommandsTest.php
+++ b/tests/src/Functional/DevelCommandsTest.php
@@ -5,38 +5,42 @@ namespace Drupal\Tests\devel\Functional;
 use Drupal\Tests\BrowserTestBase;
 use Drush\TestTraits\DrushTestTrait;
 
-/**
- * Note: Drush must be installed. See
- * https://cgit.drupalcode.org/devel/tree/drupalci.yml?h=8.x-2.x and its docs
- * at
- * https://www.drupal.org/drupalorg/docs/drupal-ci/customizing-drupalci-testing-for-projects
- */
-
-/**
- * @coversDefaultClass \Drupal\devel\Commands\DevelCommands
- * @group devel
- */
-class DevelCommandsTest extends BrowserTestBase {
-
-  use DrushTestTrait;
+if (trait_exists('DrushTestTrait')) {
 
   /**
-   * {@inheritdoc}
+   * Note: Drush must be installed. See
+   * https://cgit.drupalcode.org/devel/tree/drupalci.yml?h=8.x-2.x and its docs
+   * at
+   * https://www.drupal.org/drupalorg/docs/drupal-ci/customizing-drupalci-testing-for-projects
    */
-  public static $modules = ['devel'];
 
   /**
-   * Tests drush commands.
+   * @coversDefaultClass \Drupal\devel\Commands\DevelCommands
+   * @group devel
    */
-  public function testCommands() {
-    $this->drush('devel:token', [], ['format' => 'json']);
-    $output = $this->getOutputFromJSON();
-    $tokens = array_column($output, 'token');
-    $this->assertContains('account-name', $tokens);
-
-    $this->drush('devel:services', [], ['format' => 'json']);
-    $output = $this->getOutputFromJSON();
-    $this->assertContains('current_user', $output);
+  class DevelCommandsTest extends BrowserTestBase {
+
+    use DrushTestTrait;
+
+    /**
+     * {@inheritdoc}
+     */
+    public static $modules = ['devel'];
+
+    /**
+     * Tests drush commands.
+     */
+    public function testCommands() {
+      $this->drush('devel:token', [], ['format' => 'json']);
+      $output = $this->getOutputFromJSON();
+      $tokens = array_column($output, 'token');
+      $this->assertContains('account-name', $tokens);
+
+      $this->drush('devel:services', [], ['format' => 'json']);
+      $output = $this->getOutputFromJSON();
+      $this->assertContains('current_user', $output);
+    }
+
   }
 
 }