From 9107cf950b7ed8521db9d4c29ea377b5358a5604 Mon Sep 17 00:00:00 2001 From: Sam Minnee Date: Tue, 19 Oct 2010 01:32:17 +0000 Subject: [PATCH] ENHANCEMENT: added the ability to toggle the use draft site setting (from r110757) git-svn-id: svn://svn.silverstripe.com/silverstripe/open/modules/sapphire/trunk@112844 467b73ca-7a2a-4603-9d3b-597d59a354a9 --- dev/FunctionalTest.php | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/dev/FunctionalTest.php b/dev/FunctionalTest.php index 521b769a5..668fae1b0 100644 --- a/dev/FunctionalTest.php +++ b/dev/FunctionalTest.php @@ -279,10 +279,18 @@ class FunctionalTest extends SapphireTest { /** * Use the draft (stage) site for testing. * This is helpful if you're not testing publication functionality and don't want "stage management" cluttering your test. + * + * @param bool toggle the use of the draft site */ - function useDraftSite() { - $this->session()->inst_set('readingMode', 'Stage.Stage'); - $this->session()->inst_set('unsecuredDraftSite', true); + function useDraftSite($enabled = true) { + if($enabled) { + $this->session()->inst_set('readingMode', 'Stage.Stage'); + $this->session()->inst_set('unsecuredDraftSite', true); + } + else { + $this->session()->inst_set('readingMode', 'Stage.Live'); + $this->session()->inst_set('unsecuredDraftSite', false); + } } /**