tonyair/silverstripe-testsession
1
0
Fork 0
mirror of https://github.com/silverstripe/silverstripe-testsession synced 2024-10-22 14:06:00 +02:00
Code Issues Projects Releases Wiki Activity

Centralized isLive checks

Browse Source
This commit is contained in:
Ingo Schommer 2012-12-19 14:30:18 +01:00
parent db290a1cf9
commit 4e75488dd3
1 changed files with 62 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

118
code/TestSessionController.php
View File

@ -12,7 +12,10 @@ class TestSessionController extends Controller {
public function init() { public function init() {
parent::init(); parent::init();
$canAccess = (Director::isDev() || Director::is_cli() || Permission::check("ADMIN")); $canAccess = (
!Director::isLive()
&& (Director::isDev() || Director::is_cli() || Permission::check("ADMIN"))
);
if(!$canAccess) return Security::permissionFailure($this); if(!$canAccess) return Security::permissionFailure($this);
} }
@ -29,71 +32,66 @@ class TestSessionController extends Controller {
* as an encrypted cookie. * as an encrypted cookie.
*/ */
public function start() { public function start() {
if(!Director::isLive()) { if(SapphireTest::using_temp_db()) {
if(SapphireTest::using_temp_db()) { $endLink = Director::baseURL() . "dev/testsession/end";
$endLink = Director::baseURL() . "dev/testsession/end"; return "<p><a id=\"end-session\" href=\"$endLink\">You're in the middle of a test session;"
return "<p><a id=\"end-session\" href=\"$endLink\">You're in the middle of a test session;" . " click here to end it.</a></p>";
. " click here to end it.</a></p>";
} else if(!isset($_GET['fixture'])) { } else if(!isset($_GET['fixture'])) {
$me = Director::baseURL() . "dev/testsession/start"; $me = Director::baseURL() . "dev/testsession/start";
return <<<HTML return <<<HTML
<form action="$me"> <form action="$me">
<p>Enter a fixture file name to start a new test session. Don't forget to visit dev/testsession/end when <p>Enter a fixture file name to start a new test session. Don't forget to visit dev/testsession/end when
you're done!</p> you're done!</p>
<p>Fixture file (leave blank to start with default set-up): <input id="fixture-file" name="fixture" /></p> <p>Fixture file (leave blank to start with default set-up): <input id="fixture-file" name="fixture" /></p>
<input type="hidden" name="flush" value="1"> <input type="hidden" name="flush" value="1">
<p><input id="start-session" value="Start test session" type="submit" /></p> <p><input id="start-session" value="Start test session" type="submit" /></p>
</form> </form>
HTML; HTML;
} else { } else {
$fixtureFile = $_GET['fixture']; $fixtureFile = $_GET['fixture'];
if($fixtureFile) { if($fixtureFile) {
// Validate fixture file // Validate fixture file
$realFile = realpath(BASE_PATH.'/'.$fixtureFile); $realFile = realpath(BASE_PATH.'/'.$fixtureFile);
$baseDir = realpath(Director::baseFolder()); $baseDir = realpath(Director::baseFolder());
if(!$realFile || !file_exists($realFile)) { if(!$realFile || !file_exists($realFile)) {
return "<p>Fixture file doesn't exist</p>"; return "<p>Fixture file doesn't exist</p>";
} else if(substr($realFile,0,strlen($baseDir)) != $baseDir) { } else if(substr($realFile,0,strlen($baseDir)) != $baseDir) {
return "<p>Fixture file must be inside $baseDir</p>"; return "<p>Fixture file must be inside $baseDir</p>";
} else if(substr($realFile,-4) != '.yml') { } else if(substr($realFile,-4) != '.yml') {
return "<p>Fixture file must be a .yml file</p>"; return "<p>Fixture file must be a .yml file</p>";
} else if(!preg_match('/^([^\/.][^\/]+)\/tests\//', $fixtureFile)) { } else if(!preg_match('/^([^\/.][^\/]+)\/tests\//', $fixtureFile)) {
return "<p>Fixture file must be inside the tests subfolder of one of your modules.</p>"; return "<p>Fixture file must be inside the tests subfolder of one of your modules.</p>";
}
} }
$dbname = SapphireTest::create_temp_db();
DB::set_alternative_database_name($dbname);
// Fixture
if($fixtureFile) {
$fixture = Injector::inst()->create('YamlFixture', $fixtureFile);
$fixture->saveIntoDatabase();
// If no fixture, then use defaults
} else {
$dataClasses = ClassInfo::subclassesFor('DataObject');
array_shift($dataClasses);
foreach($dataClasses as $dataClass) singleton($dataClass)->requireDefaultRecords();
}
return "<p>Started testing session with fixture '$fixtureFile'.
Time to start testing; where would you like to start?</p>
<ul>
<li><a id=\"home-link\" href=\"" .Director::baseURL() . "\">Homepage - published site</a></li>
<li><a id=\"draft-link\" href=\"" .Director::baseURL() . "?stage=Stage\">Homepage - draft site
</a></li>
<li><a id=\"admin-link\" href=\"" .Director::baseURL() . "admin/\">CMS Admin</a></li>
<li><a id=\"end-link\" href=\"" .Director::baseURL() . "dev/testsession/end\">
End your test session</a></li>
</ul>";
} }
} else { $dbname = SapphireTest::create_temp_db();
return "<p>startession can only be used on dev and test sites</p>";
DB::set_alternative_database_name($dbname);
// Fixture
if($fixtureFile) {
$fixture = Injector::inst()->create('YamlFixture', $fixtureFile);
$fixture->saveIntoDatabase();
// If no fixture, then use defaults
} else {
$dataClasses = ClassInfo::subclassesFor('DataObject');
array_shift($dataClasses);
foreach($dataClasses as $dataClass) singleton($dataClass)->requireDefaultRecords();
}
return "<p>Started testing session with fixture '$fixtureFile'.
Time to start testing; where would you like to start?</p>
<ul>
<li><a id=\"home-link\" href=\"" .Director::baseURL() . "\">Homepage - published site</a></li>
<li><a id=\"draft-link\" href=\"" .Director::baseURL() . "?stage=Stage\">Homepage - draft site
</a></li>
<li><a id=\"admin-link\" href=\"" .Director::baseURL() . "admin/\">CMS Admin</a></li>
<li><a id=\"end-link\" href=\"" .Director::baseURL() . "dev/testsession/end\">
End your test session</a></li>
</ul>";
} }
} }
@ -115,9 +113,6 @@ HTML;
* as an encrypted cookie. * as an encrypted cookie.
*/ */
public function setdb() { public function setdb() {
if(Director::isLive()) {
return $this->httpError(403, "dev/testsession/setdb can only be used on dev and test sites");
}
if(!isset($_GET['database'])) { if(!isset($_GET['database'])) {
return $this->httpError(400, "dev/testsession/setdb must be used with a 'database' parameter"); return $this->httpError(400, "dev/testsession/setdb must be used with a 'database' parameter");
} }
@ -162,6 +157,7 @@ HTML;
public function end() { public function end() {
SapphireTest::kill_temp_db(); SapphireTest::kill_temp_db();
DB::set_alternative_database_name(null); DB::set_alternative_database_name(null);
Session::clear('testsession');
return "<p>Test session ended.</p> return "<p>Test session ended.</p>
<ul> <ul>