Init container before DataProvider is called

phpunit.xml

@@ -1,5 +1,8 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <phpunit xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="vendor/phpunit/phpunit/phpunit.xsd" bootstrap="tests/bootstrap.php" cacheResult="false" colors="true" executionOrder="random" failOnRisky="true" failOnWarning="true" failOnEmptyTestSuite="true" beStrictAboutChangesToGlobalState="true" beStrictAboutOutputDuringTests="true" cacheDirectory="tmp/.phpunit.cache" beStrictAboutCoverageMetadata="true">
+  <extensions>
+	<bootstrap class="PHPStan\Testing\PHPUnit\PHPUnitExtension"/>
+  </extensions>
   <testsuites>
     <testsuite name="PHPStan">
       <directory suffix="Test.php">tests/PHPStan</directory>

src/Testing/PHPStanTestCase.php

@@ -2,7 +2,6 @@
 
 namespace PHPStan\Testing;
 
-use Override;
 use PHPStan\Analyser\ConstantResolver;
 use PHPStan\Analyser\DirectInternalScopeFactory;
 use PHPStan\Analyser\Error;
@@ -97,12 +96,6 @@ public static function getContainer(): Container
 		return self::$containers[$cacheKey];
 	}
 
-	#[Override]
-	public static function setUpBeforeClass(): void
-	{
-		self::getContainer();
-	}
-
 	/**
 	 * @return string[]
 	 */

src/Testing/PHPUnit/InitContainerBeforeDataProviderSubscriber.php

@@ -0,0 +1,26 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Testing\PHPUnit;
+
+use Override;
+use PHPStan\Testing\PHPStanTestCase;
+use PHPUnit\Event\Test\DataProviderMethodCalled;
+use PHPUnit\Event\Test\DataProviderMethodCalledSubscriber;
+use function is_a;
+
+final class InitContainerBeforeDataProviderSubscriber implements DataProviderMethodCalledSubscriber
+{
+
+	#[Override]
+	public function notify(DataProviderMethodCalled $event): void
+	{
+		$testClassName = $event->testMethod()->className();
+
+		if (!is_a($testClassName, PhpStanTestCase::class, true)) {
+			return;
+		}
+
+		$testClassName::getContainer();
+	}
+
+}

src/Testing/PHPUnit/PHPUnitExtension.php

@@ -0,0 +1,26 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Testing\PHPUnit;
+
+use Override;
+use PHPUnit\Runner\Extension\Extension;
+use PHPUnit\Runner\Extension\Facade;
+use PHPUnit\Runner\Extension\ParameterCollection;
+use PHPUnit\TextUI\Configuration\Configuration;
+
+final class PHPUnitExtension implements Extension
+{
+
+	#[Override]
+	public function bootstrap(
+		Configuration $configuration,
+		Facade $facade,
+		ParameterCollection $parameters,
+	): void
+	{
+		$facade->registerSubscriber(
+			new InitContainerBeforeDataProviderSubscriber(),
+		);
+	}
+
+}