trait MediaFunctionalTestCreateMediaTypeTrait in Drupal 8

trait MediaFunctionalTestCreateMediaTypeTrait {
  use MediaTypeCreationTrait {
    createMediaType as traitCreateMediaType;
  }

  /**
   * Creates a media type.
   *
   * @param array $values
   *   The media type values.
   * @param string $source
   *   (optional) The media source plugin that is responsible for additional
   *   logic related to this media type. Defaults to 'test'.
   *
   * @return \Drupal\media\MediaTypeInterface
   *   A newly created media type.
   */
  protected function createMediaType(array $values = [], $source = 'test') {
    return $this
      ->traitCreateMediaType($source, $values);
  }

}