From ba94e020e7deadd5deb9b4c6744f72bec37453b3 Mon Sep 17 00:00:00 2001
From: Damian Mooyman <damian@silverstripe.com>
Date: Wed, 21 Mar 2018 16:07:31 +1300
Subject: [PATCH] Fix FunctionalTest not setting persistent versioned mode

---
 src/Dev/FunctionalTest.php | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/src/Dev/FunctionalTest.php b/src/Dev/FunctionalTest.php
index 7df93552f..fdf1894bd 100644
--- a/src/Dev/FunctionalTest.php
+++ b/src/Dev/FunctionalTest.php
@@ -406,15 +406,13 @@ class FunctionalTest extends SapphireTest implements TestOnly
      * This is helpful if you're not testing publication functionality and don't want "stage management" cluttering
      * your test.
      *
-     * @param bool $enabled toggle the use of the draft site
+     * @param bool $draft toggle the use of the draft site
      */
-    public function useDraftSite($enabled = true)
+    public function useDraftSite($draft = true)
     {
-        if ($enabled) {
-            Versioned::set_stage(Versioned::DRAFT);
-        } else {
-            Versioned::set_stage(Versioned::LIVE);
-        }
+        $stage = $draft ? Versioned::DRAFT : Versioned::LIVE;
+        Versioned::set_stage($stage);
+        Versioned::set_default_reading_mode(Versioned::get_reading_mode());
     }
 
     /**