silverstripe-gridfieldextensions/tests/GridFieldOrderableRowsTest.php

<?php
/**
 * Tests for the {@link GridFieldOrderableRows} component.
 */
class GridFieldOrderableRowsTest extends SapphireTest {

	protected $usesDatabase = true;

	protected static $fixture_file = 'GridFieldOrderableRowsTest.yml';

	protected $extraDataObjects = array(
		'GridFieldOrderableRowsTest_Parent',
		'GridFieldOrderableRowsTest_Ordered',
		'GridFieldOrderableRowsTest_Subclass',
	);

	public function testReorderItems() {
		$orderable = new GridFieldOrderableRows('ManyManySort');
		$reflection = new ReflectionMethod($orderable, 'executeReorder');
		$reflection->setAccessible(true);

		$parent = $this->objFromFixture('GridFieldOrderableRowsTest_Parent', 'parent');

		$config = new GridFieldConfig_RelationEditor();
		$config->addComponent($orderable);

		$grid = new GridField(
			'MyManyMany',
			'My Many Many',
			$parent->MyManyMany()->sort('ManyManySort'),
			$config
		);

		$originalOrder = $parent->MyManyMany()->sort('ManyManySort')->column('ID');
		$desiredOrder = array();

		// Make order non-contiguous, and 1-based
		foreach(array_reverse($originalOrder) as $index => $id) {
			$desiredOrder[$index * 2 + 1] = $id;
		}

		$this->assertNotEquals($originalOrder, $desiredOrder);

		$reflection->invoke($orderable, $grid, $desiredOrder);

		$newOrder = $parent->MyManyMany()->sort('ManyManySort')->map('ManyManySort', 'ID')->toArray();

		$this->assertEquals($desiredOrder, $newOrder);

	}

	/**
	 * @covers GridFieldOrderableRows::getSortTable
	 */
	public function testGetSortTable() {
		$orderable = new GridFieldOrderableRows();

		$parent = new GridFieldOrderableRowsTest_Parent();
		$parent->write();

		$this->assertEquals(
			'GridFieldOrderableRowsTest_Ordered',
			$orderable->getSortTable($parent->MyHasMany())
		);

		$this->assertEquals(
			'GridFieldOrderableRowsTest_Ordered',
			$orderable->getSortTable($parent->MyHasManySubclass())
		);

		$this->assertEquals(
			'GridFieldOrderableRowsTest_Ordered',
			$orderable->getSortTable($parent->MyManyMany())
		);

		$this->assertEquals(
			'GridFieldOrderableRowsTest_Parent_MyManyMany',
			$orderable->setSortField('ManyManySort')->getSortTable($parent->MyManyMany())
		);
	}

}

/**#@+
 * @ignore
 */

class GridFieldOrderableRowsTest_Parent extends DataObject implements TestOnly {

	private static $has_many = array(
		'MyHasMany' => 'GridFieldOrderableRowsTest_Ordered',
		'MyHasManySubclass' => 'GridFieldOrderableRowsTest_Subclass'
	);

	private static $many_many = array(
		'MyManyMany' => 'GridFieldOrderableRowsTest_Ordered'
	);

	private static $many_many_extraFields = array(
		'MyManyMany' => array('ManyManySort' => 'Int')
	);

}

class GridFieldOrderableRowsTest_Ordered extends DataObject implements TestOnly {

	private static $db = array(
		'Sort' => 'Int'
	);

	private static $has_one = array(
		'Parent' => 'GridFieldOrderableRowsTest_Parent'
	);

	private static $belongs_many_many =array(
		'MyManyMany' => 'GridFieldOrderableRowsTest_Parent',
	);

}

class GridFieldOrderableRowsTest_Subclass extends GridFieldOrderableRowsTest_Ordered implements TestOnly {
}

/**#@-*/