silverstripe-dms/tests/DMSShortcodeTest.php

<?php
/**
 * Tests DMS shortcode linking functionality.
 */
class DMSShortcodeTest extends SapphireTest {

	public function testShortcodeOperation() {
		$file = 'dms/tests/DMS-test-lorum-file.pdf';
		$document = DMS::inst()->storeDocument($file);

		$result = ShortcodeParser::get('default')->parse(sprintf(
			'<p><a href="[dms_document_link,id=\'%d\']">Document</a></p>', $document->ID
		));

		$value = Injector::inst()->create('HTMLValue', $result);
		$link = $value->query('//a')->item(0);

		$this->assertStringEndsWith("/dmsdocument/$document->ID", $link->getAttribute('href'));
		$this->assertEquals($document->getExtension(), $link->getAttribute('data-ext'));
		$this->assertEquals($document->getFileSizeFormatted(), $link->getAttribute('data-size'));
	}

}