Skip to content

Commit

Permalink
[BrowserKit] added the component
Browse files Browse the repository at this point in the history
  • Loading branch information
fabpot committed Apr 19, 2010
1 parent e09f730 commit 79b3915
Show file tree
Hide file tree
Showing 12 changed files with 1,659 additions and 0 deletions.
376 changes: 376 additions & 0 deletions src/Symfony/Components/BrowserKit/Client.php
Original file line number Diff line number Diff line change
@@ -0,0 +1,376 @@
<?php

namespace Symfony\Components\BrowserKit;

use Symfony\Components\DomCrawler\Crawler;
use Symfony\Components\DomCrawler\Link;
use Symfony\Components\DomCrawler\Form;
use Symfony\Components\Process\PhpProcess;

/*
* This file is part of the symfony package.
*
* (c) Fabien Potencier <fabien.potencier@symfony-project.com>
*
* For the full copyright and license information, please view the LICENSE
* file that was distributed with this source code.
*/

/**
* Client simulates a browser.
*
* To make the actual request, you need to implement the doRequest() method.
*
* If you want to be able to run requests in their own process (insulated flag),
* you need to also implement the getScript() method.
*
* @package Symfony
* @subpackage Components_BrowserKit
* @author Fabien Potencier <fabien.potencier@symfony-project.com>
*/
abstract class Client
{
protected $history;
protected $cookieJar;
protected $server;
protected $request;
protected $response;
protected $crawler;
protected $insulated;
protected $redirect;

/**
* Constructor.
*
* @param array $server The server parameters (equivalent of $_SERVER)
* @param Symfony\Components\BrowserKit\History $history A History instance to store the browser history
* @param Symfony\Components\BrowserKit\CookieJar $cookieJar A CookieJar instance to store the cookies
*/
public function __construct(array $server = array(), History $history = null, CookieJar $cookieJar = null)
{
$this->setServerParameters($server);
$this->history = null === $history ? new History() : $history;
$this->cookieJar = null === $cookieJar ? new CookieJar() : $cookieJar;
$this->insulated = false;
}

/**
* Sets the insulated flag.
*
* @param Boolean $insulated Whether to insulate the requests or not
*/
public function insulate($insulated = true)
{
if (!class_exists('Symfony\\Components\\Process\\Process'))
{
// @codeCoverageIgnoreStart
throw new \RuntimeException('Unable to isolate requests as the Symfony Process Component is not installed.');
// @codeCoverageIgnoreEnd
}

$this->insulated = (Boolean) $insulated;
}

/**
* Sets server parameters.
*
* @param array $server An array of server parameters
*/
public function setServerParameters(array $server)
{
$this->server = array_merge(array(
'HTTP_HOST' => 'localhost',
'HTTP_USER_AGENT' => 'Mozilla/5.0 (Macintosh; U; Intel Mac OS X 10.6; en-US; rv:1.9.2.3) Gecko/20100401 Firefox/3.6.3',
), $server);
}

/**
* Returns the History instance.
*
* @return Symfony\Components\BrowserKit\History A History instance
*/
public function getHistory()
{
return $this->history;
}

/**
* Returns the CookieJar instance.
*
* @return Symfony\Components\BrowserKit\CookieJar A CookieJar instance
*/
public function getCookieJar()
{
return $this->cookieJar;
}

/**
* Returns the current Crawler instance.
*
* @return Symfony\Components\DomCrawler\Crawler A Crawler instance
*/
public function getCrawler()
{
return $this->crawler;
}

/**
* Returns the current Response instance.
*
* @return Symfony\Components\BrowserKit\Response A Response instance
*/
public function getResponse()
{
return $this->response;
}

/**
* Returns the current Request instance.
*
* @return Symfony\Components\BrowserKit\Request A Request instance
*/
public function getRequest()
{
return $this->request;
}

/**
* Clicks on a given link.
*
* @param Symfony\Components\BrowserKit\Link $link A Link instance
*/
public function click(Link $link)
{
return $this->request($link->getMethod(), $link->getUri());
}

/**
* Submits a form.
*
* @param Symfony\Components\BrowserKit\Form $form A Form instance
* @param array $values An array of form field values
*/
public function submit(Form $form, array $values = array())
{
$form->setValues($values);

return $this->request($form->getMethod(), $form->getUri(), $form->getPhpValues(), array(), $form->getPhpFiles());
}

/**
* Calls a URI.
*
* @param string $method The request method
* @param string $uri The URI to fetch
* @param array $parameters The Request parameters
* @param array $headers The headers
* @param array $files The files
* @param array $server The server parameters
* @param Boolean $changeHistory Whether to update the history or not (only used internally for back(), forward(), and reload())
*
* @return Symfony\Components\DomCrawler\Crawler
*/
public function request($method, $uri, $parameters = array(), $headers = array(), $files = array(), $server = array(), $changeHistory = true)
{
$uri = $this->getAboluteUri($uri);

$server = array_merge($this->server, $server);
if (!$this->history->isEmpty())
{
$server['HTTP_REFERER'] = $this->history->current()->getUri();
}
$server['HTTP_HOST'] = parse_url($uri, PHP_URL_HOST);
$server['HTTPS'] = 'https' == parse_url($uri, PHP_URL_SCHEME);

$request = new Request($uri, $method, $parameters, $files, $this->cookieJar->getValues($uri), $server);

$this->request = $this->filterRequest($request);

if (true === $changeHistory)
{
$this->history->add($request);
}

if ($this->insulated)
{
$this->response = $this->doRequestInProcess($this->request);
}
else
{
$this->response = $this->doRequest($this->request);
}

$response = $this->filterResponse($this->response);

$this->cookieJar->updateFromResponse($response);

$this->redirect = $response->getHeader('Location');

return $this->crawler = $this->createCrawlerFromContent($request->getUri(), $response->getContent(), $response->getHeader('Content-Type'));
}

/**
* Makes a request in another process.
*
* @param Request $request A Request instance
*
* @param Response $response A Response instance
*/
protected function doRequestInProcess($request)
{
$process = new PhpProcess($this->getScript($request));
$process->run();

if ($process->getExitCode() > 0)
{
throw new \RuntimeException($process->getErrorOutput());
}

return unserialize($process->getOutput());
}

/**
* Makes a request.
*
* @param Request $request A Request instance
*
* @param Response $response A Response instance
*/
abstract protected function doRequest($request);

/**
* Returns the script to execute when the request must be insulated.
*
* @param Request $request A Request instance
*/
protected function getScript($request)
{
// @codeCoverageIgnoreStart
throw new \LogicException('To insulate requests, you need to override the getScript() method.');
// @codeCoverageIgnoreEnd
}

protected function filterRequest(Request $request)
{
return $request;
}

protected function filterResponse($response)
{
return $response;
}

protected function createCrawlerFromContent($uri, $content, $type)
{
$crawler = new Crawler(null, $uri);
$crawler->addContent($content, $type);

return $crawler;
}

/**
* Goes back in the browser history.
*/
public function back()
{
return $this->requestFromRequest($this->history->back(), false);
}

/**
* Goes forward in the browser history.
*/
public function forward()
{
return $this->requestFromRequest($this->history->forward(), false);
}

/**
* Reloads the current browser.
*/
public function reload()
{
return $this->requestFromRequest($this->history->current(), false);
}

/**
* Follow redirects?
*
* @throws sfException If request was not a redirect
*
* @return Symfony\Components\BrowserKit\Client
*/
public function followRedirect()
{
if (null === $this->redirect)
{
throw new \LogicException('The request was not redirected.');
}

return $this->request('get', $this->redirect);
}

/**
* Restarts the client.
*
* It flushes all cookies.
*/
public function restart()
{
$this->cookieJar->clear();
$this->history->clear();
}

protected function getAboluteUri($uri)
{
// already absolute?
if ('http' === substr($uri, 0, 4))
{
return $uri;
}

if (!$this->history->isEmpty())
{
$currentUri = $this->history->current()->getUri();
}
else
{
$currentUri = sprintf('http%s://%s/',
isset($this->server['HTTPS']) ? 's' : '',
isset($this->server['HTTP_HOST']) ? $this->server['HTTP_HOST'] : 'localhost'
);
}

// anchor?
if (!$uri || '#' == $uri[0])
{
return preg_replace('/#.*?$/', '', $currentUri).$uri;
}

if ('/' !== $uri[0])
{
$path = parse_url($currentUri, PHP_URL_PATH);

if ('/' !== substr($path, -1))
{
$path = substr($path, 0, strrpos($path, '/') + 1);
}

$uri = $path.$uri;
}

return preg_replace('#^(.*?//[^/]+)\/.*$#', '$1', $currentUri).$uri;
}

/**
* Makes a request from a Request object directly.
*
* @param Symfony\Components\BrowserKit\Request $request A Request instance
* @param Boolean $changeHistory Whether to update the history or not (only used internally for back(), forward(), and reload())
*
* @return Symfony\Components\DomCrawler\Crawler
*/
protected function requestFromRequest(Request $request, $changeHistory = true)
{
return $this->request($request->getMethod(), $request->getUri(), $request->getParameters(), $request->getFiles(), array(), $request->getServer(), $changeHistory);
}
}
Loading

0 comments on commit 79b3915

Please sign in to comment.