trait ExpectDeprecationTrait in Drupal 8

trait ExpectDeprecationTrait {

  /**
   * Sets an expected deprecation message.
   *
   * @param string $message
   *   The expected deprecation message.
   */
  protected function addExpectedDeprecationMessage($message) {
    $this
      ->expectedDeprecations([
      $message,
    ]);
  }

  /**
   * Sets an expected deprecation message.
   *
   * @param string $message
   *   The expected deprecation message.
   *
   * @deprecated in drupal:8.8.5 and is removed from drupal:9.0.0. Use
   *   ::addExpectedDeprecationMessage() instead.
   *
   * @see https://www.drupal.org/node/3106024
   */
  protected function expectDeprecation($message) {
    if (strpos($message, 'ExpectDeprecationTrait::expectDeprecation') === FALSE) {
      @trigger_error('ExpectDeprecationTrait::expectDeprecation is deprecated in drupal:8.8.5 and is removed from drupal:9.0.0. Use ::addExpectedDeprecationMessage() instead. See https://www.drupal.org/node/3106024', E_USER_DEPRECATED);
    }
    $this
      ->addExpectedDeprecationMessage($message);
  }

  /**
   * Sets expected deprecation messages.
   *
   * @param string[] $messages
   *   The expected deprecation messages.
   */
  public function expectedDeprecations(array $messages) {
    if ($this instanceof TestCase) {

      // Ensure the class or method is in the legacy group.
      $groups = Test::getGroups(get_class($this), $this
        ->getName(FALSE));
      if (!in_array('legacy', $groups, TRUE)) {
        throw new AssertionFailedError('Only tests with the `@group legacy` annotation can call `setExpectedDeprecation()`.');
      }

      // If setting an expected deprecation there is no need to be strict about
      // testing nothing as this is an assertion.
      $this
        ->getTestResultObject()
        ->beStrictAboutTestsThatDoNotTestAnything(FALSE);
      if ($trait = $this
        ->getSymfonyTestListenerTrait()) {

        // Add the expected deprecation message to the class property.
        $reflection_class = new \ReflectionClass($trait);
        $expected_deprecations_property = $reflection_class
          ->getProperty('expectedDeprecations');
        $expected_deprecations_property
          ->setAccessible(TRUE);
        $expected_deprecations = $expected_deprecations_property
          ->getValue($trait);
        $expected_deprecations = array_merge($expected_deprecations, $messages);
        $expected_deprecations_property
          ->setValue($trait, $expected_deprecations);

        // Register the error handler if necessary.
        $previous_error_handler_property = $reflection_class
          ->getProperty('previousErrorHandler');
        $previous_error_handler_property
          ->setAccessible(TRUE);
        $previous_error_handler = $previous_error_handler_property
          ->getValue($trait);
        if (!$previous_error_handler) {
          $previous_error_handler = set_error_handler([
            $trait,
            'handleError',
          ]);
          $previous_error_handler_property
            ->setValue($trait, $previous_error_handler);
        }
        return;
      }
    }

    // Expected deprecations set by isolated tests need to be written to a file
    // so that the test running process can take account of them.
    if ($file = getenv('DRUPAL_EXPECTED_DEPRECATIONS_SERIALIZE')) {
      $expected_deprecations = file_get_contents($file);
      if ($expected_deprecations) {
        $expected_deprecations = array_merge(unserialize($expected_deprecations), $messages);
      }
      else {
        $expected_deprecations = $messages;
      }
      file_put_contents($file, serialize($expected_deprecations));
      return;
    }
    throw new AssertionFailedError('Can not set an expected deprecation message because the Symfony\\Bridge\\PhpUnit\\SymfonyTestsListener is not registered as a PHPUnit test listener.');
  }

  /**
   * Gets the SymfonyTestsListenerTrait.
   *
   * @return \Symfony\Bridge\PhpUnit\Legacy\SymfonyTestsListenerTrait|null
   *   The SymfonyTestsListenerTrait object or NULL is a Symfony test listener
   *   is not present.
   */
  private function getSymfonyTestListenerTrait() {
    $test_result_object = $this
      ->getTestResultObject();
    $reflection_class = new \ReflectionClass($test_result_object);
    $reflection_property = $reflection_class
      ->getProperty('listeners');
    $reflection_property
      ->setAccessible(TRUE);
    $listeners = $reflection_property
      ->getValue($test_result_object);
    foreach ($listeners as $listener) {
      if ($listener instanceof SymfonyTestsListener || $listener instanceof LegacySymfonyTestsListener) {
        $reflection_class = new \ReflectionClass($listener);
        $reflection_property = $reflection_class
          ->getProperty('trait');
        $reflection_property
          ->setAccessible(TRUE);
        return $reflection_property
          ->getValue($listener);
      }
    }
  }

}