Added real Nette support via symfony/browser-kit connector

Author: enumag

composer.json

@@ -22,7 +22,7 @@
 	},
 	"autoload": {
 		"psr-0": {
-			"Codeception\\Module": "src/"
+			"Codeception": "src/"
 		}
 	}
 }

src/Codeception/Module/Nette.php

@@ -2,11 +2,15 @@
 
 namespace Codeception\Module;
 
+use Codeception\TestCase;
+use Codeception\Util\Connector\Nette as NetteConnector;
 use Codeception\Util\Framework;
 use Nette\Configurator;
-use Nette\InvalidStateException;
 use Nette\DI\Container;
 use Nette\DI\MissingServiceException;
+use Nette\Diagnostics\Debugger;
+use Nette\Environment;
+use Nette\InvalidStateException;
 use Nette\Loaders\RobotLoader;
 use Nette\Utils\Validators;
 use RecursiveDirectoryIterator;
@@ -18,12 +22,18 @@
 class Nette extends Framework
 {
 
+	/** @var Configurator */
+	protected $configurator;
+
 	/** @var Container */
 	protected $container;
 
 	/** @var RobotLoader */
 	private $robotLoader;
 
+	/** @var string */
+	private $suite;
+
 	/**
 	 * @var array $config
 	 */
@@ -49,35 +59,63 @@ public function _beforeSuite($settings = array())
 	{
 		parent::_beforeSuite($settings);
 
-		$suite = $this->detectSuiteName($settings);
-		$tempDir = $this->config['tempDir'] . DIRECTORY_SEPARATOR . $suite;
+		$this->detectSuiteName($settings);
+		$path = pathinfo($settings['path'], PATHINFO_DIRNAME);
+		$tempDir = $path . DIRECTORY_SEPARATOR . '_temp' . DIRECTORY_SEPARATOR . $this->suite;
+		Debugger::$logDirectory = $path . DIRECTORY_SEPARATOR . '_log';
 
 		self::purge($tempDir);
-		$configurator = new Configurator();
-		$configurator->setTempDirectory($tempDir);
-		$configurator->addParameters(array(
+		$this->configurator = new Configurator();
+		$this->configurator->setDebugMode(FALSE);
+		$this->configurator->setTempDirectory($tempDir);
+		$this->configurator->addParameters(array(
 			'container' => array(
-				'class' => ucfirst($suite) . 'SuiteContainer',
+				'class' => $this->getContainerClass(),
 			),
 		));
+
 		$files = $this->config['configFiles'];
 		$files[] = __DIR__ . '/config.neon';
 		foreach ($files as $file) {
-			$configurator->addConfig($file);
+			$this->configurator->addConfig($file);
 		}
-		$this->robotLoader = $configurator->createRobotLoader();
+
+		$this->robotLoader = $this->configurator->createRobotLoader();
 		foreach ($this->config['robotLoader'] as $dir) {
 			$this->robotLoader->addDirectory($dir);
 		}
 		$this->robotLoader->register();
-		$this->container = $configurator->createContainer();
 	}
 
 	public function _afterSuite()
 	{
 		$this->robotLoader->unregister();
 	}
 
+	public function _before(TestCase $test)
+	{
+		$class = $this->getContainerClass();
+		if (!class_exists($class, FALSE)) {
+			$this->container = $this->configurator->createContainer();
+		} else {
+			$this->container = new $class;
+			$this->container->initialize();
+			Environment::setContext($this->container);
+		}
+		$this->client = new NetteConnector();
+		$this->client->setContainer($this->container);
+		parent::_before($test);
+    }
+
+	public function _after(TestCase $test)
+	{
+		parent::_after($test);
+		$_SESSION = array();
+		$_GET = array();
+		$_POST = array();
+		$_COOKIE = array();
+	}
+
 	/**
 	 * @param string $service
 	 * @return object
@@ -101,7 +139,12 @@ private function detectSuiteName($settings)
 		if ($position === FALSE) {
 			throw new InvalidStateException('Could not detect suite name, path is invalid.');
 		}
-		return substr($directory, $position + 1);
+		$this->suite = substr($directory, $position + 1);
+	}
+
+	private function getContainerClass()
+	{
+		return ucfirst($this->suite) . 'SuiteContainer';
 	}
 
 	/**

src/Codeception/Util/Connector/Nette.php

@@ -0,0 +1,58 @@
+<?php
+
+namespace Codeception\Util\Connector;
+
+use Nette\DI\Container;
+use Nette\Diagnostics\Debugger;
+use Symfony\Component\BrowserKit\Client;
+use Symfony\Component\BrowserKit\Request;
+use Symfony\Component\BrowserKit\Response;
+
+/**
+ * @author Jáchym Toušek
+ */
+class Nette extends Client
+{
+
+	/** @var Container */
+	protected $container;
+
+	public function setContainer(Container $container)
+	{
+		$this->container = $container;
+	}
+
+    /**
+	 * @param Request $request
+	 * @return Response
+	 */
+	public function doRequest($request)
+	{
+		$_COOKIE = $request->getCookies();
+		$_SERVER = $request->getServer();
+		$_FILES = $request->getFiles();
+
+		$uri = str_replace('http://localhost', '', $request->getUri());
+
+		$_SERVER['REQUEST_METHOD'] = strtoupper($request->getMethod());
+		$_SERVER['REQUEST_URI'] = $uri;
+
+		ob_start();
+		try {
+			$this->container->getByType('Nette\Application\Application')->run();
+		} catch (Exception $e) {
+			ob_end_clean();
+			Debugger::log($e);
+			throw $e;
+		}
+		$content = ob_get_clean();
+
+		$httpResponse = $this->container->getByType('Nette\Http\IResponse');
+		$code = $httpResponse->getCode();
+		$headers = $httpResponse->getHeaders();
+
+		$repsonse = new Response($content, $code, $headers);
+		return $repsonse;
+	}
+
+}