tonyair/silverstripe-framework
1
0
Fork 0
mirror of https://github.com/silverstripe/silverstripe-framework synced 2024-06-27 23:19:35 +02:00
Code Issues Projects Releases Wiki Activity

Extensible test session

Browse Source
This commit is contained in:
Damian Mooyman 2018-03-21 17:28:02 +13:00
parent 6fef72062b
commit 7e73ad2101
No known key found for this signature in database
GPG Key ID: 78B823A10DE27D1A
2 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/Dev/FunctionalTest.php
View File

@ -403,16 +403,17 @@ class FunctionalTest extends SapphireTest implements TestOnly
* This is helpful if you're not testing publication functionality and don't want "stage management" cluttering * This is helpful if you're not testing publication functionality and don't want "stage management" cluttering
* your test. * your test.
* *
* @param bool $draft toggle the use of the draft site * @param bool $enabled toggle the use of the draft site
*/ */
public function useDraftSite($draft = true) public function useDraftSite($enabled = true)
{ {
if (!class_exists(Versioned::class)) { if ($enabled) {
return; $this->session()->set('readingMode', 'Stage.Stage');
$this->session()->set('unsecuredDraftSite', true);
} else {
$this->session()->set('readingMode', 'Stage.Live');
$this->session()->set('unsecuredDraftSite', false);
} }
$stage = $draft ? Versioned::DRAFT : Versioned::LIVE;
Versioned::set_stage($stage);
Versioned::set_default_reading_mode(Versioned::get_reading_mode());
} }
/** /**

12
src/Dev/TestSession.php
View File

@ -2,15 +2,16 @@
namespace SilverStripe\Dev; namespace SilverStripe\Dev;
use SilverStripe\Control\Cookie_Backend; use Exception;
use SilverStripe\Control\HTTPRequest;
use SilverStripe\Control\Session;
use SilverStripe\Control\Controller; use SilverStripe\Control\Controller;
use SilverStripe\Control\Cookie_Backend;
use SilverStripe\Control\Director; use SilverStripe\Control\Director;
use SilverStripe\Control\HTTPRequest;
use SilverStripe\Control\HTTPResponse; use SilverStripe\Control\HTTPResponse;
use SilverStripe\Control\Session;
use SilverStripe\Core\Extensible;
use SilverStripe\Core\Injector\Injector; use SilverStripe\Core\Injector\Injector;
use SimpleByName; use SimpleByName;
use Exception;
use SimplePage; use SimplePage;
use SimplePageBuilder; use SimplePageBuilder;
@ -20,6 +21,7 @@ use SimplePageBuilder;
*/ */
class TestSession class TestSession
{ {
use Extensible;
/** /**
* @var Session * @var Session
@ -89,6 +91,7 @@ class TestSession
*/ */
public function get($url, $session = null, $headers = null, $cookies = null) public function get($url, $session = null, $headers = null, $cookies = null)
{ {
$this->extend('updateGetURL', $url, $session, $headers, $cookies);
$headers = (array) $headers; $headers = (array) $headers;
if ($this->lastUrl && !isset($headers['Referer'])) { if ($this->lastUrl && !isset($headers['Referer'])) {
$headers['Referer'] = $this->lastUrl; $headers['Referer'] = $this->lastUrl;
@ -123,6 +126,7 @@ class TestSession
*/ */
public function post($url, $data, $headers = null, $session = null, $body = null, $cookies = null) public function post($url, $data, $headers = null, $session = null, $body = null, $cookies = null)
{ {
$this->extend('updatePostURL', $url, $data, $headers, $session, $body, $cookies);
$headers = (array) $headers; $headers = (array) $headers;
if ($this->lastUrl && !isset($headers['Referer'])) { if ($this->lastUrl && !isset($headers['Referer'])) {
$headers['Referer'] = $this->lastUrl; $headers['Referer'] = $this->lastUrl;