silverstripe-multiform/tests/MultiFormObjectDecoratorTest.php

<?php
class MultiFormObjectDecoratorTest extends SapphireTest {

	protected static $fixture_file = 'MultiFormObjectDecoratorTest.yml';

	protected $requiredExtensions = array(
		'MultiFormObjectDecorator_DataObject' => array('MultiFormObjectDecorator')
	);

	protected $extraDataObjects = array(
		'MultiFormObjectDecorator_DataObject'
	);

	public function testTemporaryDataFilteredQuery() {
		$records = MultiFormObjectDecorator_DataObject::get()
			->map('Name')
			->toArray();

		$this->assertContains('Test 1', $records);
		$this->assertContains('Test 2', $records);
		$this->assertNotContains('Test 3', $records);

	}

	public function testTemporaryDataQuery() {
		$records = MultiFormObjectDecorator_DataObject::get()
			->filter(array('MultiFormIsTemporary' => 1))
			->map('Name')
			->toArray();

		$this->assertNotContains('Test 1', $records);
		$this->assertNotContains('Test 2', $records);
		$this->assertContains('Test 3', $records);
	}

}

class MultiFormObjectDecorator_DataObject extends DataObject {

	private static $db = array(
		'Name' => 'Varchar'
	);

}