silverstripe-dms/tests/DMSTest.php

<?php
class DMSTest extends SapphireTest {

	static $testFile = 'dms/tests/DMS-test-lorum-file.pdf';

	//store values to reset back to after this test runs
	static $dmsFolderOld;
	static $dmsFolderSizeOld;

	function setUp() {
		self::$dmsFolderOld = DMS::$dmsFolder;
		self::$dmsFolderSizeOld = DMS::$dmsFolderSize;

		//use a test DMS folder, so we don't overwrite the live one
		DMS::$dmsFolder = 'dms-assets-test-1234';

		//clear out the test folder (in case a broken test doesn't delete it)
		$this->delete(BASE_PATH . DIRECTORY_SEPARATOR . DMS::$dmsFolder);
	}

	function tearDown() {
		$d = DataObject::get("DMSDocument");
		foreach($d as $d1) {
			$d1->delete();
		}
		$t = DataObject::get("DMSTag");
		foreach($t as $t1) {
			$t1->delete();
		}

		//delete the test folder after the test runs
		$this->delete(BASE_PATH . DIRECTORY_SEPARATOR . DMS::$dmsFolder);

		//set the old DMS folder back again
		DMS::$dmsFolder = self::$dmsFolderOld;
		DMS::$dmsFolderSize = self::$dmsFolderSizeOld;
	}

	public function delete($path) {
		if (file_exists($path) || is_dir($path)) {
		    $it = new RecursiveIteratorIterator(
		        new RecursiveDirectoryIterator($path),
		        RecursiveIteratorIterator::CHILD_FIRST
		    );
		    foreach ($it as $file) {
		        if (in_array($file->getBasename(), array('.', '..'))) {
		            continue;
		        } elseif ($file->isDir()) {
		            rmdir($file->getPathname());
		        } elseif ($file->isFile() || $file->isLink()) {
		            unlink($file->getPathname());
		        }
		    }
		    rmdir($path);
		}
	}


	function testDMSStorage() {
		$dms = DMS::getDMSInstance();

		$file = self::$testFile;
		$document = $dms->storeDocument($file);

		$this->assertNotNull($document, "Document object created");
		$this->assertTrue(file_exists(DMS::$dmsPath . DIRECTORY_SEPARATOR . $document->Folder . DIRECTORY_SEPARATOR . $document->Filename),"Document file copied into DMS folder");

		//$title = $document->getTag('title');
	}

	function testDMSFolderSpanning() {
		DMS::$dmsFolderSize = 5;
		$dms = DMS::getDMSInstance();

		$file = self::$testFile;

		$documents = array();
		for($i = 0; $i <= 16; $i++) {
			$document = $dms->storeDocument($file);
			$this->assertNotNull($document, "Document object created on run number: $i");
			$this->assertTrue(file_exists($document->getFullPath()));
			$documents[] = $document;
		}

		//test document objects have their folders set
		$folders = array();
		for($i = 0; $i <= 16; $i++) {
			$folderName = $documents[$i]->Folder;
			$this->assertTrue(strpos($documents[$i]->getFullPath(), DIRECTORY_SEPARATOR . $folderName . DIRECTORY_SEPARATOR) !== false, "Correct folder name for the documents. Document path contains reference to folder name '$folderName'");
			$folders[] = $folderName;
		}

		//test we created 4 folder to contain the 17 files
		foreach($folders as $f) {
			$this->assertTrue(is_dir(DMS::$dmsPath . DIRECTORY_SEPARATOR . $f),"Document folder '$f' exists");
		}
	}


}