move tests to test_tools

1 files changed, 490 insertions, 0 deletions

diff --git a/test_tools/simpletest/user_agent.php b/test_tools/simpletest/user_agent.php
new file mode 100644
index 00000000..c65d5110
--- /dev/null
+++ b/test_tools/simpletest/user_agent.php
@@ -0,0 +1,490 @@
+<?php
+    /**
+     *	Base include file for SimpleTest
+     *	@package	SimpleTest
+     *	@subpackage	WebTester
+     *	@version	$Id: user_agent.php,v 1.43 2005/01/02 22:46:08 lastcraft Exp $
+     */
+
+    /**#@+
+     *	include other SimpleTest class files
+     */
+    require_once(dirname(__FILE__) . '/http.php');
+    require_once(dirname(__FILE__) . '/encoding.php');
+    require_once(dirname(__FILE__) . '/authentication.php');
+    /**#@-*/
+   
+    define('DEFAULT_MAX_REDIRECTS', 3);
+    define('DEFAULT_CONNECTION_TIMEOUT', 15);
+    
+    /**
+     *    Repository for cookies. This stuff is a
+     *    tiny bit browser dependent.
+	 *    @package SimpleTest
+	 *    @subpackage WebTester
+     */
+    class SimpleCookieJar {
+        protected $_cookies;
+        
+        /**
+         *    Constructor. Jar starts empty.
+         *    @access public
+         */
+        function SimpleCookieJar() {
+            $this->_cookies = array();
+        }
+        
+        /**
+         *    Removes expired and temporary cookies as if
+         *    the browser was closed and re-opened.
+         *    @param string/integer $now   Time to test expiry against.
+         *    @access public
+         */
+        function restartSession($date = false) {
+            $surviving_cookies = array();
+            for ($i = 0; $i < count($this->_cookies); $i++) {
+                if (! $this->_cookies[$i]->getValue()) {
+                    continue;
+                }
+                if (! $this->_cookies[$i]->getExpiry()) {
+                    continue;
+                }
+                if ($date && $this->_cookies[$i]->isExpired($date)) {
+                    continue;
+                }
+                $surviving_cookies[] = $this->_cookies[$i];
+            }
+            $this->_cookies = $surviving_cookies;
+        }
+        
+        /**
+         *    Ages all cookies in the cookie jar.
+         *    @param integer $interval     The old session is moved
+         *                                 into the past by this number
+         *                                 of seconds. Cookies now over
+         *                                 age will be removed.
+         *    @access public
+         */
+        function agePrematurely($interval) {
+            for ($i = 0; $i < count($this->_cookies); $i++) {
+                $this->_cookies[$i]->agePrematurely($interval);
+            }
+        }
+        
+        /**
+         *    Adds a cookie to the jar. This will overwrite
+         *    cookies with matching host, paths and keys.
+         *    @param SimpleCookie $cookie        New cookie.
+         *    @access public
+         */
+        function setCookie($cookie) {
+            for ($i = 0; $i < count($this->_cookies); $i++) {
+                $is_match = $this->_isMatch(
+                        $cookie,
+                        $this->_cookies[$i]->getHost(),
+                        $this->_cookies[$i]->getPath(),
+                        $this->_cookies[$i]->getName());
+                if ($is_match) {
+                    $this->_cookies[$i] = $cookie;
+                    return;
+                }
+            }
+            $this->_cookies[] = $cookie;
+        }
+        
+        /**
+         *    Fetches a hash of all valid cookies filtered
+         *    by host, path and keyed by name
+         *    Any cookies with missing categories will not
+         *    be filtered out by that category. Expired
+         *    cookies must be cleared by restarting the session.
+         *    @param string $host   Host name requirement.
+         *    @param string $path   Path encompassing cookies.
+         *    @return hash          Valid cookie objects keyed
+         *                          on the cookie name.
+         *    @access public
+         */
+        function getValidCookies($host = false, $path = "/") {
+            $valid_cookies = array();
+            foreach ($this->_cookies as $cookie) {
+                if ($this->_isMatch($cookie, $host, $path, $cookie->getName())) {
+                    $valid_cookies[] = $cookie;
+                }
+            }
+            return $valid_cookies;
+        }
+        
+        /**
+         *    Tests cookie for matching against search
+         *    criteria.
+         *    @param SimpleTest $cookie    Cookie to test.
+         *    @param string $host          Host must match.
+         *    @param string $path          Cookie path must be shorter than
+         *                                 this path.
+         *    @param string $name          Name must match.
+         *    @return boolean              True if matched.
+         *    @access private
+         */
+        function _isMatch($cookie, $host, $path, $name) {
+            if ($cookie->getName() != $name) {
+                return false;
+            }
+            if ($host && $cookie->getHost() && !$cookie->isValidHost($host)) {
+                return false;
+            }
+            if (! $cookie->isValidPath($path)) {
+                return false;
+            }
+            return true;
+        }
+        
+        /**
+         *    Adds the current cookies to a request.
+         *    @param SimpleHttpRequest $request    Request to modify.
+         *    @param SimpleUrl $url                Cookie selector.
+         *    @access private
+         */
+        function addHeaders($request, $url) {
+            $cookies = $this->getValidCookies($url->getHost(), $url->getPath());
+            foreach ($cookies as $cookie) {
+                $request->setCookie($cookie);
+            }
+        }
+    }
+
+    /**
+     *    Fetches web pages whilst keeping track of
+     *    cookies and authentication.
+	 *    @package SimpleTest
+	 *    @subpackage WebTester
+     */
+    class SimpleUserAgent {
+        protected $_cookie_jar;
+        protected $_authenticator;
+        protected $_max_redirects;
+        protected $_proxy;
+        protected $_proxy_username;
+        protected $_proxy_password;
+        protected $_connection_timeout;
+        protected $_additional_headers;
+        
+        /**
+         *    Starts with no cookies, realms or proxies.
+         *    @access public
+         */
+        function SimpleUserAgent() {
+            $this->_cookie_jar = new SimpleCookieJar();
+            $this->_authenticator = new SimpleAuthenticator();
+            $this->setMaximumRedirects(DEFAULT_MAX_REDIRECTS);
+            $this->_proxy = false;
+            $this->_proxy_username = false;
+            $this->_proxy_password = false;
+            $this->setConnectionTimeout(DEFAULT_CONNECTION_TIMEOUT);
+            $this->_additional_headers = array();
+        }
+        
+        /**
+         *    Removes expired and temporary cookies as if
+         *    the browser was closed and re-opened. Authorisation
+         *    has to be obtained again as well.
+         *    @param string/integer $date   Time when session restarted.
+         *                                  If omitted then all persistent
+         *                                  cookies are kept.
+         *    @access public
+         */
+        function restart($date = false) {
+            $this->_cookie_jar->restartSession($date);
+            $this->_authenticator->restartSession();
+        }
+        
+        /**
+         *    Adds a header to every fetch.
+         *    @param string $header       Header line to add to every
+         *                                request until cleared.
+         *    @access public
+         */
+        function addHeader($header) {
+            $this->_additional_headers[] = $header;
+        }
+        
+        /**
+         *    Ages the cookies by the specified time.
+         *    @param integer $interval    Amount in seconds.
+         *    @access public
+         */
+        function ageCookies($interval) {
+            $this->_cookie_jar->agePrematurely($interval);
+        }
+        
+        /**
+         *    Sets an additional cookie. If a cookie has
+         *    the same name and path it is replaced.
+         *    @param string $name            Cookie key.
+         *    @param string $value           Value of cookie.
+         *    @param string $host            Host upon which the cookie is valid.
+         *    @param string $path            Cookie path if not host wide.
+         *    @param string $expiry          Expiry date.
+         *    @access public
+         */
+        function setCookie($name, $value, $host = false, $path = '/', $expiry = false) {
+            $cookie = new SimpleCookie($name, $value, $path, $expiry);
+            if ($host) {
+                $cookie->setHost($host);
+            }
+            $this->_cookie_jar->setCookie($cookie);
+        }
+        
+        /**
+         *    Reads the most specific cookie value from the
+         *    browser cookies.
+         *    @param string $host        Host to search.
+         *    @param string $path        Applicable path.
+         *    @param string $name        Name of cookie to read.
+         *    @return string             False if not present, else the
+         *                               value as a string.
+         *    @access public
+         */
+        function getCookieValue($host, $path, $name) {
+            $longest_path = '';
+            foreach ($this->_cookie_jar->getValidCookies($host, $path) as $cookie) {
+                if ($name == $cookie->getName()) {
+                    if (strlen($cookie->getPath()) > strlen($longest_path)) {
+                        $value = $cookie->getValue();
+                        $longest_path = $cookie->getPath();
+                    }
+                }
+            }
+            return (isset($value) ? $value : false);
+        }
+        
+        /**
+         *    Reads the current cookies within the base URL.
+         *    @param string $name     Key of cookie to find.
+         *    @param SimpleUrl $base  Base URL to search from.
+         *    @return string          Null if there is no base URL, false
+         *                            if the cookie is not set.
+         *    @access public
+         */
+        function getBaseCookieValue($name, $base) {
+            if (! $base) {
+                return null;
+            }
+            return $this->getCookieValue($base->getHost(), $base->getPath(), $name);
+        }
+        
+        /**
+         *    Sets the socket timeout for opening a connection.
+         *    @param integer $timeout      Maximum time in seconds.
+         *    @access public
+         */
+        function setConnectionTimeout($timeout) {
+            $this->_connection_timeout = $timeout;
+        }
+        
+        /**
+         *    Sets the maximum number of redirects before
+         *    a page will be loaded anyway.
+         *    @param integer $max        Most hops allowed.
+         *    @access public
+         */
+        function setMaximumRedirects($max) {
+            $this->_max_redirects = $max;
+        }
+        
+        /**
+         *    Sets proxy to use on all requests for when
+         *    testing from behind a firewall. Set URL
+         *    to false to disable.
+         *    @param string $proxy        Proxy URL.
+         *    @param string $username     Proxy username for authentication.
+         *    @param string $password     Proxy password for authentication.
+         *    @access public
+         */
+        function useProxy($proxy, $username, $password) {
+            if (! $proxy) {
+                $this->_proxy = false;
+                return;
+            }
+            if (strncmp($proxy, 'http://', 7) != 0) {
+                $proxy = 'http://'. $proxy;
+            }
+            $this->_proxy = new SimpleUrl($proxy);
+            $this->_proxy_username = $username;
+            $this->_proxy_password = $password;
+        }
+        
+        /**
+         *    Test to see if the redirect limit is passed.
+         *    @param integer $redirects        Count so far.
+         *    @return boolean                  True if over.
+         *    @access private
+         */
+        function _isTooManyRedirects($redirects) {
+            return ($redirects > $this->_max_redirects);
+        }
+        
+        /**
+         *    Sets the identity for the current realm.
+         *    @param string $host        Host to which realm applies.
+         *    @param string $realm       Full name of realm.
+         *    @param string $username    Username for realm.
+         *    @param string $password    Password for realm.
+         *    @access public
+         */
+        function setIdentity($host, $realm, $username, $password) {
+            $this->_authenticator->setIdentityForRealm($host, $realm, $username, $password);
+        }
+        
+        /**
+         *    Fetches a URL as a response object. Will keep trying if redirected.
+         *    It will also collect authentication realm information.
+         *    @param string $method                   GET, POST, etc.
+         *    @param string/SimpleUrl $url            Target to fetch.
+         *    @param SimpleFormEncoding $parameters   Additional parameters for request.
+         *    @return SimpleHttpResponse              Hopefully the target page.
+         *    @access public
+         */
+        function fetchResponse($method, $url, $parameters = false) {
+            if ($method != 'POST') {
+                $url->addRequestParameters($parameters);
+                $parameters = false;
+            }
+            $response = $this->_fetchWhileRedirected($method, $url, $parameters);
+            if ($headers = $response->getHeaders()) {
+                if ($headers->isChallenge()) {
+                    $this->_authenticator->addRealm(
+                            $url,
+                            $headers->getAuthentication(),
+                            $headers->getRealm());
+                }
+            }
+            return $response;
+        }
+        
+        /**
+         *    Fetches the page until no longer redirected or
+         *    until the redirect limit runs out.
+         *    @param string $method                  GET, POST, etc.
+         *    @param SimpleUrl $url                  Target to fetch.
+         *    @param SimpelFormEncoding $parameters  Additional parameters for request.
+         *    @return SimpleHttpResponse             Hopefully the target page.
+         *    @access private
+         */
+        function _fetchWhileRedirected($method, $url, $parameters) {
+            $redirects = 0;
+            do {
+                $response = $this->_fetch($method, $url, $parameters);
+                if ($response->isError()) {
+                    return $response;
+                }
+                $headers = $response->getHeaders();
+                $location = new SimpleUrl($headers->getLocation());
+                $url = $location->makeAbsolute($url);
+                $this->_addCookiesToJar($url, $headers->getNewCookies());
+                if (! $headers->isRedirect()) {
+                    break;
+                }
+                $method = 'GET';
+                $parameters = false;
+            } while (! $this->_isTooManyRedirects(++$redirects));
+            return $response;
+        }
+        
+        /**
+         *    Actually make the web request.
+         *    @param string $method                   GET, POST, etc.
+         *    @param SimpleUrl $url                   Target to fetch.
+         *    @param SimpleFormEncoding $parameters   Additional parameters for request.
+         *    @return SimpleHttpResponse              Headers and hopefully content.
+         *    @access protected
+         */
+        function _fetch($method, $url, $parameters) {
+            if (! $parameters) {
+                $parameters = new SimpleFormEncoding();
+            }
+            $request = $this->_createRequest($method, $url, $parameters);
+            return $request->fetch($this->_connection_timeout);
+        }
+        
+        /**
+         *    Creates a full page request.
+         *    @param string $method                   Fetching method.
+         *    @param SimpleUrl $url                   Target to fetch as url object.
+         *    @param SimpleFormEncoding $parameters   POST/GET parameters.
+         *    @return SimpleHttpRequest               New request.
+         *    @access private
+         */
+        function _createRequest($method, $url, $parameters) {
+            $request = $this->_createHttpRequest($method, $url, $parameters);
+            $this->_addAdditionalHeaders($request);
+            $this->_cookie_jar->addHeaders($request, $url);
+            $this->_authenticator->addHeaders($request, $url);
+            return $request;
+        }
+        
+        /**
+         *    Builds the appropriate HTTP request object.
+         *    @param string $method                  Fetching method.
+         *    @param SimpleUrl $url                  Target to fetch as url object.
+         *    @param SimpleFormEncoding $parameters  POST/GET parameters.
+         *    @return SimpleHttpRequest              New request object.
+         *    @access protected
+         */
+        function _createHttpRequest($method, $url, $parameters) {
+            if ($method == 'POST') {
+                $request = new SimpleHttpRequest(
+                        $this->_createRoute($url),
+                        'POST',
+                        $parameters);
+                return $request;
+            }
+            if ($parameters) {
+                $url->addRequestParameters($parameters);
+            }
+            return new SimpleHttpRequest($this->_createRoute($url), $method);
+        }
+        
+        /**
+         *    Sets up either a direct route or via a proxy.
+         *    @param SimpleUrl $url   Target to fetch as url object.
+         *    @return SimpleRoute     Route to take to fetch URL.
+         *    @access protected
+         */
+        function _createRoute($url) {
+            if ($this->_proxy) {
+                return new SimpleProxyRoute(
+                        $url,
+                        $this->_proxy,
+                        $this->_proxy_username,
+                        $this->_proxy_password);
+            }
+            return new SimpleRoute($url);
+        }
+        
+        /**
+         *    Adds additional manual headers.
+         *    @param SimpleHttpRequest $request    Outgoing request.
+         *    @access private
+         */
+        function _addAdditionalHeaders($request) {
+            foreach ($this->_additional_headers as $header) {
+                $request->addHeaderLine($header);
+            }
+        }
+        
+        /**
+         *    Extracts new cookies into the cookie jar.
+         *    @param SimpleUrl $url     Target to fetch as url object.
+         *    @param array $cookies     New cookies.
+         *    @access private
+         */
+        function _addCookiesToJar($url, $cookies) {
+            foreach ($cookies as $cookie) {
+                if ($url->getHost()) {
+                    $cookie->setHost($url->getHost());
+                }
+                $this->_cookie_jar->setCookie($cookie);
+            }
+        }
+    }
+?>
\ No newline at end of file