EnvironmentCleanerService.php

<?php

namespace Drupal\simpletest;

use Drupal\Core\Database\Connection;
use Drupal\Core\Messenger\MessengerInterface;
use Drupal\Core\StringTranslation\TranslationInterface;
use Drupal\Core\Config\ConfigFactory;
use Drupal\Core\Cache\CacheBackendInterface;
use Drupal\Core\File\FileSystem;
use Drupal\Core\Test\EnvironmentCleaner;

/**
 * Uses containerized services to perform post-test cleanup.
 */
class EnvironmentCleanerService extends EnvironmentCleaner {
    
    /**
     * Messenger service.
     *
     * @var \Drupal\Core\Messenger\MessengerInterface
     */
    protected $messenger;
    
    /**
     * The translation service.
     *
     * @var \Drupal\Core\StringTranslation\TranslationInterface
     */
    protected $translation;
    
    /**
     * The config factory.
     *
     * @var \Drupal\Core\Config\ConfigFactory
     */
    protected $configFactory;
    
    /**
     * Default cache.
     *
     * @var \Drupal\Core\Cache\CacheBackendInterface
     */
    protected $cacheDefault;
    
    /**
     * Construct an environment cleaner.
     *
     * @param string $root
     *   The path to the root of the Drupal installation.
     * @param \Drupal\Core\Database\Connection $test_database
     *   Connection to the database against which tests were run.
     * @param \Drupal\Core\Database\Connection $results_database
     *   Connection to the database where test results were stored. This could be
     *   the same as $test_database, or it could be different.
     * @param \Drupal\Core\StringTranslation\TranslationInterface|null $translation
     *   (optional) The translation service. If none is supplied, this class will
     *   attempt to discover one using \Drupal.
     */
    public function __construct($root, Connection $test_database, Connection $results_database, MessengerInterface $messenger, TranslationInterface $translation, ConfigFactory $config, CacheBackendInterface $cache_default, FileSystem $file_system) {
        $this->root = $root;
        $this->testDatabase = $test_database;
        $this->resultsDatabase = $results_database;
        $this->messenger = $messenger;
        $this->translation = $translation;
        $this->configFactory = $config;
        $this->cacheDefault = $cache_default;
        $this->fileSystem = $file_system;
    }
    
    /**
     * {@inheritdoc}
     */
    public function cleanEnvironment($clear_results = TRUE, $clear_temp_directories = TRUE, $clear_database = TRUE) {
        $results_removed = 0;
        $clear_results = $this->configFactory
            ->get('simpletest.settings')
            ->get('clear_results');
        if ($clear_database) {
            $this->cleanDatabase();
        }
        if ($clear_temp_directories) {
            $this->cleanTemporaryDirectories();
        }
        if ($clear_results) {
            $results_removed = $this->cleanResultsTable();
        }
        $this->cacheDefault
            ->delete('simpletest');
        $this->cacheDefault
            ->delete('simpletest_phpunit');
        if ($clear_results) {
            $this->messenger
                ->addMessage($this->translation
                ->formatPlural($results_removed, 'Removed 1 test result.', 'Removed @count test results.'));
        }
        else {
            $this->messenger
                ->addMessage($this->translation
                ->translate('Clear results is disabled and the test results table will not be cleared.'), 'warning');
        }
    }
    
    /**
     * {@inheritdoc}
     */
    public function cleanDatabase() {
        $tables_removed = $this->doCleanDatabase();
        if ($tables_removed > 0) {
            $this->messenger
                ->addMessage($this->translation
                ->formatPlural($tables_removed, 'Removed 1 leftover table.', 'Removed @count leftover tables.'));
        }
        else {
            $this->messenger
                ->addMessage($this->translation
                ->translate('No leftover tables to remove.'));
        }
    }
    
    /**
     * {@inheritdoc}
     */
    public function cleanTemporaryDirectories() {
        $directories_removed = $this->doCleanTemporaryDirectories();
        if ($directories_removed > 0) {
            $this->messenger
                ->addMessage($this->translation
                ->formatPlural($directories_removed, 'Removed 1 temporary directory.', 'Removed @count temporary directories.'));
        }
        else {
            $this->messenger
                ->addMessage($this->translation
                ->translate('No temporary directories to remove.'));
        }
    }

}