silverstripe-dms/tests/cms/DMSGridFieldAddNewButtonTest.php
Robbie Averill d0c2558a1f FIX Get page ID from document set and include it in $NewLink for add document buttons
This helps the module to determine where to return a user to after they've uploaded a file, e.g. either back to a document set in a page context, back to a document set in the ModelAdmin context or back to the ModelAdmin for documents with no context.
2017-06-14 10:57:00 +12:00

80 lines
2.4 KiB
PHP

<?php
class DMSGridFieldAddNewButtonTest extends SapphireTest
{
protected static $fixture_file = 'dms/tests/dmstest.yml';
/**
* @var DMSGridFieldAddNewButton
*/
protected $button;
/**
* @var GridField
*/
protected $gridField;
public function setUp()
{
parent::setUp();
$fakeList = DMSDocument::get();
$this->gridField = GridField::create('TestGridField', false, $fakeList);
$this->button = new DMSGridFieldAddNewButton;
}
/**
* Test that when no document set ID is present then it is not added to the URL for "add document"
*/
public function testNoDocumentSetIdInAddUrlWhenNotProvided()
{
$this->assertNotContains('?dsid', $this->getButtonHtml());
}
/**
* Test that when a document set ID is provided, it is added onto the "add document" link
*/
public function testDocumentSetIdAddedToLinkWhenProvided()
{
$this->button->setDocumentSetId(123);
$this->assertContains('?dsid=123', $this->getButtonHtml());
}
/**
* If a set is saved and associated to a page, that page's ID should be added to the "add document" link to help
* to ensure the user gets redirected back to the correct place afterwards
*/
public function testPageIdIsAddedWhenAvailableViaDocumentSetRelationship()
{
$set = $this->objFromFixture('DMSDocumentSet', 'ds1');
$this->button->setDocumentSetId($set->ID);
$controller = new ContentController;
$controller->pushCurrent();
$result = $this->getButtonHtml();
$this->assertContains('dsid=' . $set->ID, $result, 'Add new button contains document set ID');
$this->assertNotContains('page_id=' . $set->Page()->ID, $result, 'No page ID when not editing in page context');
$controller = new CMSPageEditController;
$controller->pushCurrent();
$this->assertContains(
'page_id=' . $set->Page()->ID,
$this->getButtonHtml(),
'Button contains page ID when in edit page context'
);
}
/**
* Returns the HTML result of the "add new" button, used for DRY in test class
*
* @return string
*/
protected function getButtonHtml()
{
$fragments = $this->button->getHTMLFragments($this->gridField);
return array_pop($fragments)->getValue();
}
}