Implement referrer for Kohana driver

OpenBuildings · Jun 23, 2015 · 9a2c432 · 9a2c432
1 parent 28c68a5
commit 9a2c432
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/Openbuildings/Spiderling/Driver/Kohana/RequestFactory/Kohana.php b/src/Openbuildings/Spiderling/Driver/Kohana/RequestFactory/Kohana.php
@@ -12,6 +12,8 @@
  */
 class Driver_Kohana_RequestFactory_Kohana implements Driver_Simple_RequestFactory {
 
+	protected static $_previous_url;
+
 	protected $_request;
 	protected $_response;
 	protected $_max_redirects = 5;
@@ -65,6 +67,12 @@ public function execute($method, $url, array $post = array())
 			->post($post)
 			->body(http_build_query($post));
 
+		if (Driver_Kohana_RequestFactory_Kohana::$_previous_url) {
+			$this->_request->referrer(Driver_Kohana_RequestFactory_Kohana::$_previous_url);
+		}
+
+		Driver_Kohana_RequestFactory_Kohana::$_previous_url = $this->current_url();
+
 		\Request::$initial = $this->_request;
 
 		$this->_response = $this->_request->execute();

diff --git a/tests/tests/Driver/KohanaTest.php b/tests/tests/Driver/KohanaTest.php
@@ -48,6 +48,15 @@ public function test_redirect()
 		$this->assertSame(Request::$initial, $driver->request_factory()->request());
 	}
 
+	public function test_referrer()
+	{
+		$driver = new Driver_Kohana();
+		$driver->get('/test/index');
+		$this->assertEquals('', $driver->request_factory()->request()->referrer());
+		$driver->get('/test/action_final');
+		$this->assertEquals('http://example.com/test/index', $driver->request_factory()->request()->referrer());
+	}
+
 	public function test_too_many_redirects()
 	{
 		$this->setExpectedException('Openbuildings\Spiderling\Exception_Toomanyredirects');