Unable to add values to $databaseTraits in \Illuminate\Testing\Concerns\TestDatabases

[JakeBooher]

• Laravel Version: 8.46.0
• PHP Version: 8.0.3
• Database Driver & Version: MariaDB 10.4

Description:

We have some custom traits for handling database migrations and rollbacks for certain test classes. For example, we have one that persists the database throughout the entire class. This works fine in the normal single threaded test CLI, but once I get into parallel tests I have issues that seem to be from it using the default SQLite database instead of the one created for each thread. Adding this into \Illuminate\Testing\Concerns\TestDatabases resolves this. However, to do that I have to extend Illuminate\Foundation\Providers\FoundationServiceProvider with the following:

<?php

namespace App\Illuminate\Foundation\Providers;

use App\Illuminate\Testing\ParallelTestingServiceProvider;

class FoundationServiceProvider extends \Illuminate\Foundation\Providers\FoundationServiceProvider
{
    public function __construct($app)
    {
        $this->providers[] = ParallelTestingServiceProvider::class;
        if (($key = array_search(\Illuminate\Testing\ParallelTestingServiceProvider::class, $this->providers))) {
            unset($this->providers[$key]);
        }
        parent::__construct($app);
    }
}

and then overwrite the entire bootTestDatabase method from \Illuminate\Testing\Concerns\TestDatabases by doing this:

<?php

namespace App\Illuminate\Testing;

use Illuminate\Foundation\Testing\DatabaseMigrations;
use Illuminate\Foundation\Testing\DatabaseTransactions;
use Illuminate\Foundation\Testing\RefreshDatabase;
use Illuminate\Support\Arr;
use Illuminate\Support\Facades\ParallelTesting;
use Illuminate\Support\Facades\Schema;
use Tests\Traits\RefreshDatabaseForClass;

class ParallelTestingServiceProvider extends \Illuminate\Testing\ParallelTestingServiceProvider
{
    protected function getDatabaseTraits(): array
    {
        return [
            DatabaseMigrations::class,
            DatabaseTransactions::class,
            RefreshDatabase::class,
            RefreshDatabaseForClass::class,
        ];
    }

    protected function bootTestDatabase()
    {
        ParallelTesting::setUpProcess(function () {
            $this->whenNotUsingInMemoryDatabase(function ($database) {
                if (ParallelTesting::option('recreate_databases')) {
                    Schema::dropDatabaseIfExists(
                        $this->testDatabase($database)
                    );
                }
            });
        });

        ParallelTesting::setUpTestCase(function ($testCase) {
            $uses = array_flip(class_uses_recursive(get_class($testCase)));

            $databaseTraits = $this->getDatabaseTraits();

            if (Arr::hasAny($uses, $databaseTraits)) {
                $this->whenNotUsingInMemoryDatabase(function ($database) use ($uses) {
                    [$testDatabase, $created] = $this->ensureTestDatabaseExists($database);

                    $this->switchToDatabase($testDatabase);

                    if (isset($uses[Testing\DatabaseTransactions::class])) {
                        $this->ensureSchemaIsUpToDate();
                    }

                    if ($created) {
                        ParallelTesting::callSetUpTestDatabaseCallbacks($testDatabase);
                    }
                });
            }
        });
    }
}

Easiest way to solve this would probably be to call a getDatabaseTraits() method like I'm doing in my extension so it can be extended alone, but it may be nice to have it done some way that I don't have to extend two separate providers to get to that point

Steps To Reproduce: