Support for Data Providers

Closes #25

Author: Alexander Obuhovich

library/aik099/PHPUnit/TestSuite/AbstractTestSuite.php

@@ -77,21 +77,37 @@ public function addTestMethods($class_name)
 	 * @param SessionStrategyManager       $session_strategy_manager      Session strategy manager.
 	 * @param IBrowserConfigurationFactory $browser_configuration_factory Browser configuration factory.
 	 * @param RemoteCoverageHelper         $remote_coverage_helper        Remote coverage helper.
+	 * @param array                        $tests                         Tests to process.
 	 *
 	 * @return self
 	 */
 	public function setTestDependencies(
 		SessionStrategyManager $session_strategy_manager,
 		IBrowserConfigurationFactory $browser_configuration_factory,
-		RemoteCoverageHelper $remote_coverage_helper
+		RemoteCoverageHelper $remote_coverage_helper,
+		array $tests = null
 	)
 	{
-		/* @var $test BrowserTestCase */
-		foreach ( $this->tests() as $test ) {
-			$test->setEventDispatcher($this->_eventDispatcher);
-			$test->setSessionStrategyManager($session_strategy_manager);
-			$test->setBrowserConfigurationFactory($browser_configuration_factory);
-			$test->setRemoteCoverageHelper($remote_coverage_helper);
+		if ( !isset($tests) ) {
+			$tests = $this->tests();
+		}
+
+		foreach ( $tests as $test ) {
+			if ( $test instanceof \PHPUnit_Framework_TestSuite_DataProvider ) {
+				$this->setTestDependencies(
+					$session_strategy_manager,
+					$browser_configuration_factory,
+					$remote_coverage_helper,
+					$test->tests()
+				);
+			}
+			else {
+				/** @var BrowserTestCase $test */
+				$test->setEventDispatcher($this->_eventDispatcher);
+				$test->setSessionStrategyManager($session_strategy_manager);
+				$test->setBrowserConfigurationFactory($browser_configuration_factory);
+				$test->setRemoteCoverageHelper($remote_coverage_helper);
+			}
 		}
 
 		return $this;
@@ -100,14 +116,24 @@ public function setTestDependencies(
 	/**
 	 * Report back suite ending to each it's test.
 	 *
+	 * @param array $tests Tests to process.
+	 *
 	 * @return void
 	 */
-	protected function tearDown()
+	protected function tearDown(array $tests = null)
 	{
-		/* @var $test BrowserTestCase */
+		if ( !isset($tests) ) {
+			$tests = $this->tests();
+		}
 
-		foreach ( $this->tests() as $test ) {
-			$test->onTestSuiteEnded();
+		foreach ( $tests as $test ) {
+			if ( $test instanceof \PHPUnit_Framework_TestSuite_DataProvider ) {
+				$this->tearDown($test->tests());
+			}
+			else {
+				/* @var $test BrowserTestCase */
+				$test->onTestSuiteEnded();
+			}
 		}
 	}
 

library/aik099/PHPUnit/TestSuite/BrowserTestSuite.php

@@ -43,14 +43,24 @@ public function nameFromBrowser(array $browser)
 	 * Sets given browser to be used in each underlying test cases and test suites.
 	 *
 	 * @param array $browser Browser configuration.
+	 * @param array $tests   Tests to process.
 	 *
 	 * @return self
 	 */
-	public function setBrowserFromConfiguration(array $browser)
+	public function setBrowserFromConfiguration(array $browser, array $tests = null)
 	{
-		/* @var $test BrowserTestCase */
-		foreach ( $this->tests() as $test ) {
-			$test->setBrowserFromConfiguration($browser);
+		if ( !isset($tests) ) {
+			$tests = $this->tests();
+		}
+
+		foreach ( $tests as $test ) {
+			if ( $test instanceof \PHPUnit_Framework_TestSuite_DataProvider ) {
+				$this->setBrowserFromConfiguration($browser, $test->tests());
+			}
+			else {
+				/* @var $test BrowserTestCase */
+				$test->setBrowserFromConfiguration($browser);
+			}
 		}
 
 		return $this;

tests/aik099/PHPUnit/Integration/DataProviderTest.php

@@ -0,0 +1,48 @@
+<?php
+/**
+ * This file is part of the phpunit-mink library.
+ * For the full copyright and license information, please view
+ * the LICENSE file that was distributed with this source code.
+ *
+ * @copyright Alexander Obuhovich <aik.bold@gmail.com>
+ * @link      https://github.com/aik099/phpunit-mink
+ */
+
+namespace tests\aik099\PHPUnit\Integration;
+
+
+use aik099\PHPUnit\BrowserTestCase;
+use Mockery as m;
+
+class DataProviderTest extends BrowserTestCase
+{
+
+	public function sampleDataProvider()
+	{
+		return array(
+			array('case1'),
+			array('case2'),
+		);
+	}
+
+	/**
+	 * @dataProvider sampleDataProvider
+	 */
+	public function testDataProvider($case)
+	{
+		$this->customMethod();
+
+		if ( $case === 'case1' || $case === 'case2' ) {
+			$this->assertTrue(true);
+		}
+		else {
+			$this->fail('Unknown $case: ' . $case);
+		}
+	}
+
+	protected function customMethod()
+	{
+		return 5;
+	}
+
+}