Added step to remove media.

Author: Alex Skrypnyk

src/MediaTrait.php

@@ -74,6 +74,28 @@ public function mediaCreate($type, TableNode $nodesTable) {
     }
   }
 
+  /**
+   * Remove media defined by provided properties.
+   *
+   * @code
+   * Given no "image" media:
+   * | name               |
+   * | Media item         |
+   * | Another media item |
+   * @endcode
+   *
+   * @Given /^no ([a-zA-z0-9_-]+) media:$/
+   */
+  public function mediaDelete($type, TableNode $nodesTable) {
+    foreach ($nodesTable->getHash() as $nodeHash) {
+      $ids = $this->mediaLoadMultiple($type, $nodeHash);
+
+      $controller = \Drupal::entityTypeManager()->getStorage('media');
+      $entities = $controller->loadMultiple($ids);
+      $controller->delete($entities);
+    }
+  }
+
   /**
    * Create a single media item.
    */

tests/behat/features/media.feature

@@ -7,14 +7,25 @@ Feature: Check that MediaTrait works for or D9
       | path                 |
       | example_document.pdf |
 
-    And "image" media:
+    And no image media:
       | name             | field_media_image |
       | Test media image | example_image.png |
+
+    And "image" media:
+      | name              | field_media_image |
+      | Test media image  | example_image.png |
+      | Test media image2 | example_image.png |
+
+    And no image media:
+      | name              |
+      | Test media image2 |
+
     And "document" media:
       | name                | field_media_document |
       | Test media document | example_document.pdf |
 
     And I am logged in as a user with the "administrator" role
     When I visit "/admin/content/media"
     Then I should see the text "Test media image"
+    Then I should not see the text "Test media image2"
     And I should see the text "Test media document"