tonyair
/
silverstripe-subsites
mirror of https://github.com/silverstripe/silverstripe-subsites
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

MINOR pass tests

This commit is contained in:
Tom Rix 2010-03-02 01:15:11 +00:00
parent 3dd9b2a1e4
commit fa21df693f
2 changed files with 11 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
code/SiteTreeSubsites.php
View File

@ -48,27 +48,20 @@ class SiteTreeSubsites extends SiteTreeDecorator {
);
}
/**
* Check if we're currently looking at the main site.
* @return boolean TRUE main site | FALSE sub-site
*/
function isMainSite() {
if($this->owner->SubsiteID == 0) return true;
return false;
}
/**
* Update any requests to limit the results to the current site
*/
function augmentSQL(SQLQuery &$query) {
if(Subsite::$disable_subsite_filter) return;
$q = defined('DB::USE_ANSI_SQL') ? '"' : '`';
// Don't run on delete queries, since they are always tied to
// a specific ID.
if ($query->delete) return;
// If you're querying by ID, ignore the sub-site - this is a bit ugly...
// if(!$query->where || (strpos($query->where[0], ".\"ID\" = ") === false && strpos($query->where[0], ".`ID` = ") === false && strpos($query->where[0], ".ID = ") === false && strpos($query->where[0], "ID = ") !== 0)) {
if (!$query->where || (!preg_match('/\.(\'|"|`|)ID(\'|"|`|)( ?)=/', $query->where[0]))) {
if (Subsite::$force_subsite) $subsiteID = Subsite::$force_subsite;
else {
if($context = DataObject::context_obj()) $subsiteID = (int)$context->SubsiteID;
@ -77,11 +70,9 @@ class SiteTreeSubsites extends SiteTreeDecorator {
// The foreach is an ugly way of getting the first key :-)
foreach($query->from as $tableName => $info) {
$where = "{$q}$tableName{$q}.{$q}SubsiteID{$q} IN ($subsiteID)";
// The tableName should be SiteTree or SiteTree_Live...
if(strpos($tableName,'SiteTree') === false) break;
$query->where[] = $where;
$query->where[] = "\"$tableName\".\"SubsiteID\" IN ($subsiteID)";
break;
}
}
@ -97,11 +88,8 @@ class SiteTreeSubsites extends SiteTreeDecorator {
protected $nextWriteDoesntCustomise = false;
function augmentBeforeWrite() {
// if the page hasn't been written to a database table yet and no subsite has been set, then give it a subsite
if((!is_numeric($this->owner->ID) || $this->owner->ID == 0) && !$this->owner->SubsiteID) {
$this->owner->SubsiteID = Subsite::currentSubsiteID();
}
function onBeforeWrite() {
if(!$this->owner->ID && !$this->owner->SubsiteID) $this->owner->SubsiteID = Subsite::currentSubsiteID();
// If the content has been changed, then the page should be marked as 'custom content'
if(!$this->nextWriteDoesntCustomise && $this->owner->ID && $this->owner->MasterPageID && !$this->owner->CustomContent) {
@ -117,6 +105,8 @@ class SiteTreeSubsites extends SiteTreeDecorator {
}
$this->nextWriteDoesntCustomise = false;
parent::onBeforeWrite();
}
function onAfterWrite(&$original) {

5
tests/SubsitesVirtualPageTest.php
View File

@ -27,7 +27,7 @@ class SubsitesVirtualPageTest extends SapphireTest {
$this->assertEquals($svp->SubsiteID, $subsite->ID);
$this->assertEquals($svp->Title, $linky->Title);
}
/**
* Test custom metadata. Reloading Content should not
* obliterate our custom fields
@ -62,7 +62,7 @@ class SubsitesVirtualPageTest extends SapphireTest {
function testFileLinkRewritingOnVirtualPages() {
// File setup
$this->logInWithPermssion('ADMIN');
$this->logInWithPermission('ADMIN');
touch(Director::baseFolder() . '/assets/testscript-test-file.pdf');
// Publish the source page
@ -178,7 +178,6 @@ class SubsitesVirtualPageTest extends SapphireTest {
$page->doUnpublish();
Subsite::changeSubsite($vp1->SubsiteID);
$_REQUEST['showqueries']=1;
$onLive = Versioned::get_one_by_stage('SubsitesVirtualPage', 'Live', "SiteTree_Live.ID = ".$vp1->ID);
$this->assertFalse($onLive, 'SVP has been removed from live');