<?php
/*
* This file is part of PHPUnit.
*
* (c) Sebastian Bergmann <sebastian@phpunit.de>
*
* For the full copyright and license information, please view the LICENSE
* file that was distributed with this source code.
*/
namespace PHPUnit\Framework;

use Iterator;
use IteratorAggregate;
use PHPUnit\Runner\BaseTestRunner;
use PHPUnit\Runner\Filter\Factory;
use PHPUnit\Runner\PhptTestCase;
use PHPUnit\Util\Fileloader;
use PHPUnit\Util\InvalidArgumentHelper;
use ReflectionClass;
use ReflectionMethod;
use Throwable;

/**
* A TestSuite is a composite of Tests. It runs a collection of test cases.
*/
class TestSuite implements Test, SelfDescribing, IteratorAggregate
{
/**
* Last count of tests in this suite.
*
* @var int|null
*/
private $cachedNumTests;

/**
* Enable or disable the backup and restoration of the $GLOBALS array.
*
* @var bool
*/
protected $backupGlobals;

/**
* Enable or disable the backup and restoration of static attributes.
*
* @var bool
*/
protected $backupStaticAttributes;

/**
* @var bool
*/
private $beStrictAboutChangesToGlobalState;

/**
* @var bool
*/
protected $runTestInSeparateProcess = false;

/**
* The name of the test suite.
*
* @var string
*/
protected $name = '';

/**
* The test groups of the test suite.
*
* @var array
*/
protected $groups = [];

/**
* The tests in the test suite.
*
* @var TestCase[]
*/
protected $tests = [];

/**
* The number of tests in the test suite.
*
* @var int
*/
protected $numTests = -1;

/**
* @var bool
*/
protected $testCase = false;

/**
* @var array
*/
protected $foundClasses = [];

/**
* @var Factory
*/
private $iteratorFilter;

/**
* @var string[]
*/
private $declaredClasses;

/**
* Constructs a new TestSuite:
*
* - PHPUnit\Framework\TestSuite() constructs an empty TestSuite.
*
* - PHPUnit\Framework\TestSuite(ReflectionClass) constructs a
* TestSuite from the given class.
*
* - PHPUnit\Framework\TestSuite(ReflectionClass, String)
* constructs a TestSuite from the given class with the given
* name.
*
* - PHPUnit\Framework\TestSuite(String) either constructs a
* TestSuite from the given class (if the passed string is the
* name of an existing class) or constructs an empty TestSuite
* with the given name.
*
* @param mixed $theClass
* @param string $name
*
* @throws Exception
*/
public function __construct($theClass = '', $name = '')
{
$this->declaredClasses = \get_declared_classes();

$argumentsValid = false;

if (\is_object($theClass) &&
$theClass instanceof ReflectionClass) {
$argumentsValid = true;
} elseif (\is_string($theClass) &&
$theClass !== '' &&
\class_exists($theClass, false)) {
$argumentsValid = true;

if ($name == '') {
$name = $theClass;
}

$theClass = new ReflectionClass($theClass);
} elseif (\is_string($theClass)) {
$this->setName($theClass);

return;
}

if (!$argumentsValid) {
throw new Exception;
}

if (!$theClass->isSubclassOf(TestCase::class)) {
throw new Exception(
'Class "' . $theClass->name . '" does not extend PHPUnit\Framework\TestCase.'
);
}

if ($name != '') {
$this->setName($name);
} else {
$this->setName($theClass->getName());
}

$constructor = $theClass->getConstructor();

if ($constructor !== null &&
!$constructor->isPublic()) {
$this->addTest(
self::warning(
\sprintf(
'Class "%s" has no public constructor.',
$theClass->getName()
)
)
);

return;
}

foreach ($theClass->getMethods() as $method) {
$this->addTestMethod($theClass, $method);
}

if (empty($this->tests)) {
$this->addTest(
self::warning(
\sprintf(
'No tests found in class "%s".',
$theClass->getName()
)
)
);
}

$this->testCase = true;
}

/**
* Returns a string representation of the test suite.
*
* @return string
*/
public function toString()
{
return $this->getName();
}

/**
* Adds a test to the suite.
*
* @param Test $test
* @param array $groups
*/
public function addTest(Test $test, $groups = [])
{
$class = new ReflectionClass($test);

if (!$class->isAbstract()) {
$this->tests[] = $test;
$this->numTests = -1;

if ($test instanceof self && empty($groups)) {
$groups = $test->getGroups();
}

if (empty($groups)) {
$groups = ['default'];
}

foreach ($groups as $group) {
if (!isset($this->groups[$group])) {
$this->groups[$group] = [$test];
} else {
$this->groups[$group][] = $test;
}
}

if ($test instanceof TestCase) {
$test->setGroups($groups);
}
}
}

/**
* Adds the tests from the given class to the suite.
*
* @param mixed $testClass
*
* @throws Exception
*/
public function addTestSuite($testClass)
{
if (\is_string($testClass) && \class_exists($testClass)) {
$testClass = new ReflectionClass($testClass);
}

if (!\is_object($testClass)) {
throw InvalidArgumentHelper::factory(
1,
'class name or object'
);
}

if ($testClass instanceof self) {
$this->addTest($testClass);
} elseif ($testClass instanceof ReflectionClass) {
$suiteMethod = false;

if (!$testClass->isAbstract() && $testClass->hasMethod(BaseTestRunner::SUITE_METHODNAME)) {
$method = $testClass->getMethod(
BaseTestRunner::SUITE_METHODNAME
);

if ($method->isStatic()) {
$this->addTest(
$method->invoke(null, $testClass->getName())
);

$suiteMethod = true;
}
}

if (!$suiteMethod && !$testClass->isAbstract() && $testClass->isSubclassOf(TestCase::class)) {
$this->addTest(new self($testClass));
}
} else {
throw new Exception;
}
}

/**
* Wraps both <code>addTest()</code> and <code>addTestSuite</code>
* as well as the separate import statements for the user's convenience.
*
* If the named file cannot be read or there are no new tests that can be
* added, a <code>PHPUnit\Framework\WarningTestCase</code> will be created instead,
* leaving the current test run untouched.
*
* @param string $filename
*
* @throws Exception
*/
public function addTestFile($filename)
{
if (!\is_string($filename)) {
throw InvalidArgumentHelper::factory(1, 'string');
}

if (\file_exists($filename) && \substr($filename, -5) == '.phpt') {
$this->addTest(
new PhptTestCase($filename)
);

return;
}

// The given file may contain further stub classes in addition to the
// test class itself. Figure out the actual test class.
$filename = Fileloader::checkAndLoad($filename);
$newClasses = \array_diff(\get_declared_classes(), $this->declaredClasses);

// The diff is empty in case a parent class (with test methods) is added
// AFTER a child class that inherited from it. To account for that case,
// accumulate all discovered classes, so the parent class may be found in
// a later invocation.
if (!empty($newClasses)) {
// On the assumption that test classes are defined first in files,
// process discovered classes in approximate LIFO order, so as to
// avoid unnecessary reflection.
$this->foundClasses = \array_merge($newClasses, $this->foundClasses);
$this->declaredClasses = \get_declared_classes();
}

// The test class's name must match the filename, either in full, or as
// a PEAR/PSR-0 prefixed shortname ('NameSpace_ShortName'), or as a
// PSR-1 local shortname ('NameSpace\ShortName'). The comparison must be
// anchored to prevent false-positive matches (e.g., 'OtherShortName').
$shortname = \basename($filename, '.php');
$shortnameRegEx = '/(?:^|_|\\\\)' . \preg_quote($shortname, '/') . '$/';

foreach ($this->foundClasses as $i => $className) {
if (\preg_match($shortnameRegEx, $className)) {
$class = new ReflectionClass($className);

if ($class->getFileName() == $filename) {
$newClasses = [$className];
unset($this->foundClasses[$i]);

break;
}
}
}

foreach ($newClasses as $className) {
$class = new ReflectionClass($className);

if (\dirname($class->getFileName()) === __DIR__) {
continue;
}

if (!$class->isAbstract()) {
if ($class->hasMethod(BaseTestRunner::SUITE_METHODNAME)) {
$method = $class->getMethod(
BaseTestRunner::SUITE_METHODNAME
);

if ($method->isStatic()) {
$this->addTest($method->invoke(null, $className));
}
} elseif ($class->implementsInterface(Test::class)) {
$this->addTestSuite($class);
}
}
}

$this->numTests = -1;
}

/**
* Wrapper for addTestFile() that adds multiple test files.
*
* @param array|Iterator $filenames
*
* @throws Exception
*/
public function addTestFiles($filenames)
{
if (!(\is_array($filenames) ||
(\is_object($filenames) && $filenames instanceof Iterator))) {
throw InvalidArgumentHelper::factory(
1,
'array or iterator'
);
}

foreach ($filenames as $filename) {
$this->addTestFile((string) $filename);
}
}

/**
* Counts the number of test cases that will be run by this test.
*
* @param bool $preferCache Indicates if cache is preferred.
*
* @return int
*/
public function count($preferCache = false)
{
if ($preferCache && $this->cachedNumTests !== null) {
return $this->cachedNumTests;
}

$numTests = 0;

foreach ($this as $test) {
$numTests += \count($test);
}

$this->cachedNumTests = $numTests;

return $numTests;
}

/**
* @param ReflectionClass $theClass
* @param string $name
*
* @return Test
*
* @throws Exception
*/
public static function createTest(ReflectionClass $theClass, $name)
{
$className = $theClass->getName();

if (!$theClass->isInstantiable()) {
return self::warning(
\sprintf('Cannot instantiate class "%s".', $className)
);
}

$backupSettings = \PHPUnit\Util\Test::getBackupSettings(
$className,
$name
);

$preserveGlobalState = \PHPUnit\Util\Test::getPreserveGlobalStateSettings(
$className,
$name
);

$runTestInSeparateProcess = \PHPUnit\Util\Test::getProcessIsolationSettings(
$className,
$name
);

$runClassInSeparateProcess = \PHPUnit\Util\Test::getClassProcessIsolationSettings(
$className,
$name
);

$constructor = $theClass->getConstructor();

if ($constructor !== null) {
$parameters = $constructor->getParameters();

// TestCase() or TestCase($name)
if (\count($parameters) < 2) {
$test = new $className;
} // TestCase($name, $data)
else {
try {
$data = \PHPUnit\Util\Test::getProvidedData(
$className,
$name
);
} catch (IncompleteTestError $e) {
$message = \sprintf(
'Test for %s::%s marked incomplete by data provider',
$className,
$name
);

$_message = $e->getMessage();

if (!empty($_message)) {
$message .= "\n" . $_message;
}

$data = self::incompleteTest($className, $name, $message);
} catch (SkippedTestError $e) {
$message = \sprintf(
'Test for %s::%s skipped by data provider',
$className,
$name
);

$_message = $e->getMessage();

if (!empty($_message)) {
$message .= "\n" . $_message;
}

$data = self::skipTest($className, $name, $message);
} catch (Throwable $_t) {
$t = $_t;
} catch (Exception $_t) {
$t = $_t;
}

if (isset($t)) {
$message = \sprintf(
'The data provider specified for %s::%s is invalid.',
$className,
$name
);

$_message = $t->getMessage();

if (!empty($_message)) {
$message .= "\n" . $_message;
}

$data = self::warning($message);
}

// Test method with @dataProvider.
if (isset($data)) {
$test = new DataProviderTestSuite(
$className . '::' . $name
);

if (empty($data)) {
$data = self::warning(
\sprintf(
'No tests found in suite "%s".',
$test->getName()
)
);
}

$groups = \PHPUnit\Util\Test::getGroups($className, $name);

if ($data instanceof WarningTestCase ||
$data instanceof SkippedTestCase ||
$data instanceof IncompleteTestCase) {
$test->addTest($data, $groups);
} else {
foreach ($data as $_dataName => $_data) {
$_test = new $className($name, $_data, $_dataName);

if ($runTestInSeparateProcess) {
$_test->setRunTestInSeparateProcess(true);

if ($preserveGlobalState !== null) {
$_test->setPreserveGlobalState($preserveGlobalState);
}
}

if ($runClassInSeparateProcess) {
$_test->setRunClassInSeparateProcess(true);

if ($preserveGlobalState !== null) {
$_test->setPreserveGlobalState($preserveGlobalState);
}
}

if ($backupSettings['backupGlobals'] !== null) {
$_test->setBackupGlobals(
$backupSettings['backupGlobals']
);
}

if ($backupSettings['backupStaticAttributes'] !== null) {
$_test->setBackupStaticAttributes(
$backupSettings['backupStaticAttributes']
);
}

$test->addTest($_test, $groups);
}
}
} else {
$test = new $className;
}
}
}

if (!isset($test)) {
throw new Exception('No valid test provided.');
}

if ($test instanceof TestCase) {
$test->setName($name);

if ($runTestInSeparateProcess) {
$test->setRunTestInSeparateProcess(true);

if ($preserveGlobalState !== null) {
$test->setPreserveGlobalState($preserveGlobalState);
}
}

if ($backupSettings['backupGlobals'] !== null) {
$test->setBackupGlobals($backupSettings['backupGlobals']);
}

if ($backupSettings['backupStaticAttributes'] !== null) {
$test->setBackupStaticAttributes(
$backupSettings['backupStaticAttributes']
);
}
}

return $test;
}

/**
* Creates a default TestResult object.
*
* @return TestResult
*/
protected function createResult()
{
return new TestResult;
}

/**
* Returns the name of the suite.
*
* @return string
*/
public function getName()
{
return $this->name;
}

/**
* Returns the test groups of the suite.
*
* @return array
*/
public function getGroups()
{
return \array_keys($this->groups);
}

public function getGroupDetails()
{
return $this->groups;
}

/**
* Set tests groups of the test case
*
* @param array $groups
*/
public function setGroupDetails(array $groups)
{
$this->groups = $groups;
}

/**
* Runs the tests and collects their result in a TestResult.
*
* @param TestResult $result
*
* @return TestResult
*/
public function run(TestResult $result = null)
{
if ($result === null) {
$result = $this->createResult();
}

if (\count($this) == 0) {
return $result;
}

$hookMethods = \PHPUnit\Util\Test::getHookMethods($this->name);

$result->startTestSuite($this);

try {
$this->setUp();

foreach ($hookMethods['beforeClass'] as $beforeClassMethod) {
if ($this->testCase === true &&
\class_exists($this->name, false) &&
\method_exists($this->name, $beforeClassMethod)) {
if ($missingRequirements = \PHPUnit\Util\Test::getMissingRequirements($this->name, $beforeClassMethod)) {
$this->markTestSuiteSkipped(\implode(PHP_EOL, $missingRequirements));
}

\call_user_func([$this->name, $beforeClassMethod]);
}
}
} catch (SkippedTestSuiteError $e) {
$numTests = \count($this);

for ($i = 0; $i < $numTests; $i++) {
$result->startTest($this);
$result->addFailure($this, $e, 0);
$result->endTest($this, 0);
}

$this->tearDown();
$result->endTestSuite($this);

return $result;
} catch (Throwable $_t) {
$t = $_t;
} catch (Exception $_t) {
$t = $_t;
}

if (isset($t)) {
$numTests = \count($this);

for ($i = 0; $i < $numTests; $i++) {
if ($result->shouldStop()) {
break;
}

$result->startTest($this);
$result->addError($this, $t, 0);
$result->endTest($this, 0);
}

$this->tearDown();
$result->endTestSuite($this);

return $result;
}

foreach ($this as $test) {
if ($result->shouldStop()) {
break;
}

if ($test instanceof TestCase || $test instanceof self) {
$test->setBeStrictAboutChangesToGlobalState($this->beStrictAboutChangesToGlobalState);
$test->setBackupGlobals($this->backupGlobals);
$test->setBackupStaticAttributes($this->backupStaticAttributes);
$test->setRunTestInSeparateProcess($this->runTestInSeparateProcess);
}

$test->run($result);
}

foreach ($hookMethods['afterClass'] as $afterClassMethod) {
if ($this->testCase === true && \class_exists($this->name, false) && \method_exists($this->name, $afterClassMethod)) {
\call_user_func([$this->name, $afterClassMethod]);
}
}

$this->tearDown();

$result->endTestSuite($this);

return $result;
}

/**
* @param bool $runTestInSeparateProcess
*
* @throws Exception
*/
public function setRunTestInSeparateProcess($runTestInSeparateProcess)
{
if (\is_bool($runTestInSeparateProcess)) {
$this->runTestInSeparateProcess = $runTestInSeparateProcess;
} else {
throw InvalidArgumentHelper::factory(1, 'boolean');
}
}

/**
* Runs a test.
*
* @deprecated
*
* @param Test $test
* @param TestResult $result
*/
public function runTest(Test $test, TestResult $result)
{
$test->run($result);
}

/**
* Sets the name of the suite.
*
* @param string
*/
public function setName($name)
{
$this->name = $name;
}

/**
* Returns the test at the given index.
*
* @param int|false
*
* @return Test|false
*/
public function testAt($index)
{
if (isset($this->tests[$index])) {
return $this->tests[$index];
}

return false;
}

/**
* Returns the tests as an enumeration.
*
* @return array
*/
public function tests()
{
return $this->tests;
}

/**
* Set tests of the test suite
*
* @param array $tests
*/
public function setTests(array $tests)
{
$this->tests = $tests;
}

/**
* Mark the test suite as skipped.
*
* @param string $message
*
* @throws SkippedTestSuiteError
*/
public function markTestSuiteSkipped($message = '')
{
throw new SkippedTestSuiteError($message);
}

/**
* @param ReflectionClass $class
* @param ReflectionMethod $method
*/
protected function addTestMethod(ReflectionClass $class, ReflectionMethod $method)
{
if (!$this->isTestMethod($method)) {
return;
}

$name = $method->getName();

if (!$method->isPublic()) {
$this->addTest(
self::warning(
\sprintf(
'Test method "%s" in test class "%s" is not public.',
$name,
$class->getName()
)
)
);

return;
}

$test = self::createTest($class, $name);

if ($test instanceof TestCase || $test instanceof DataProviderTestSuite) {
$test->setDependencies(
\PHPUnit\Util\Test::getDependencies($class->getName(), $name)
);
}

$this->addTest(
$test,
\PHPUnit\Util\Test::getGroups($class->getName(), $name)
);
}

/**
* @param ReflectionMethod $method
*
* @return bool
*/
public static function isTestMethod(ReflectionMethod $method)
{
if (\strpos($method->name, 'test') === 0) {
return true;
}

// @scenario on TestCase::testMethod()
// @test on TestCase::testMethod()
$docComment = $method->getDocComment();

return \strpos($docComment, '@test') !== false ||
\strpos($docComment, '@scenario') !== false;
}

/**
* @param string $message
*
* @return WarningTestCase
*/
protected static function warning($message)
{
return new WarningTestCase($message);
}

/**
* @param string $class
* @param string $methodName
* @param string $message
*
* @return SkippedTestCase
*/
protected static function skipTest($class, $methodName, $message)
{
return new SkippedTestCase($class, $methodName, $message);
}

/**
* @param string $class
* @param string $methodName
* @param string $message
*
* @return IncompleteTestCase
*/
protected static function incompleteTest($class, $methodName, $message)
{
return new IncompleteTestCase($class, $methodName, $message);
}

/**
* @param bool $beStrictAboutChangesToGlobalState
*/
public function setBeStrictAboutChangesToGlobalState($beStrictAboutChangesToGlobalState)
{
if (null === $this->beStrictAboutChangesToGlobalState && \is_bool($beStrictAboutChangesToGlobalState)) {
$this->beStrictAboutChangesToGlobalState = $beStrictAboutChangesToGlobalState;
}
}

/**
* @param bool $backupGlobals
*/
public function setBackupGlobals($backupGlobals)
{
if (null === $this->backupGlobals && \is_bool($backupGlobals)) {
$this->backupGlobals = $backupGlobals;
}
}

/**
* @param bool $backupStaticAttributes
*/
public function setBackupStaticAttributes($backupStaticAttributes)
{
if (null === $this->backupStaticAttributes && \is_bool($backupStaticAttributes)) {
$this->backupStaticAttributes = $backupStaticAttributes;
}
}

/**
* Returns an iterator for this test suite.
*
* @return TestSuiteIterator
*/
public function getIterator()
{
$iterator = new TestSuiteIterator($this);

if ($this->iteratorFilter !== null) {
$iterator = $this->iteratorFilter->factory($iterator, $this);
}

return $iterator;
}

public function injectFilter(Factory $filter)
{
$this->iteratorFilter = $filter;
foreach ($this as $test) {
if ($test instanceof self) {
$test->injectFilter($filter);
}
}
}

/**
* Template Method that is called before the tests
* of this test suite are run.
*/
protected function setUp()
{
}

/**
* Template Method that is called after the tests
* of this test suite have finished running.
*/
protected function tearDown()
{
}
}