silverstripe-framework/tests/forms/TableFieldTest.php

<?php

class TableFieldTest extends SapphireTest {
	static $fixture_file = 'sapphire/tests/forms/TableFieldTest.yml';
	
	
	function testTableFieldSaving() {
		$group = $this->objFromFixture('Group','a');
		
		$tableField = new TableField(
			"Permissions",
			"Permission",
			array(
			        "Code" => _t('SecurityAdmin.CODE', 'Code'),
			        "Arg" => _t('SecurityAdmin.OPTIONALID', 'Optional ID'),
			),
			array(
				"Code" => "PermissionDropdownField",
				"Arg" => "TextField",
			),
			"GroupID",
			$group->ID
		);
		$form = new Form(new TableFieldTest_Controller(), "Form", new FieldSet($tableField), new FieldSet());
		
		/* The field starts emppty.  Save some new data.  We have replicated the array structure that the specific layout of the form generates. */
		$tableField->setValue(array(
			'new' => array(
				'Code' => array(
					'CMS_ACCESS_CMSMain',
					'CMS_ACCESS_AssetAdmin',
				),
				'Arg' => array(
					'1',
					'2'
				),
			),			
		));
		$tableField->saveInto($group);

		/* Let's check that the 2 permissions entries have been saved */
		$permissions = $group->Permissions()->toDropdownMap('Arg', 'Code');
		$this->assertEquals(array(
			1 => 'CMS_ACCESS_CMSMain',
			2 => 'CMS_ACCESS_AssetAdmin',
		), $permissions);
		

		/* Now let's perform an update query */
		$value = array();
		foreach($group->Permissions() as $permission) {
			$value[$permission->ID] = array("Code" => $permission->Code, "Arg" => $permission->Arg);
		}
		$value['new'] = array(
			'Code' => array(
				'CMS_ACCESS_NewsletterAdmin',
			),
			'Arg' => array(
				'3',
			),
		);
		$tableField->setValue($value);
		$tableField->saveInto($group);

		/* Let's check that the 2 existing permissions entries, and the 1 new one, have been saved */
		$permissions = $group->Permissions()->toDropdownMap('Arg', 'Code');
		$this->assertEquals(array(
			1 => 'CMS_ACCESS_CMSMain',
			2 => 'CMS_ACCESS_AssetAdmin',
			3 => 'CMS_ACCESS_NewsletterAdmin',
		), $permissions);

	}
	
	function testAutoRelationSettingOn() {
		$tf = new TableField(
			'HasManyRelations',
			'TableFieldTest_HasManyRelation',
			array(
				'Value' => 'Value'
			),
			array(
				'Value' => 'TextField'
			)
		);
		
		// Test with auto relation setting
		$form = new Form(new TableFieldTest_Controller(), "Form", new FieldSet($tf), new FieldSet());
		$tf->setValue(array(
			'new' => array(
				'Value' => array('one','two',)
			)
		));
		$tf->setRelationAutoSetting(true);
		$o = new TableFieldTest_Object();
		$o->write();
		$form->saveInto($o);
		$this->assertEquals($o->HasManyRelations()->Count(), 2);
	}
	
	function testAutoRelationSettingOff() {
		$tf = new TableField(
			'HasManyRelations',
			'TableFieldTest_HasManyRelation',
			array(
				'Value' => 'Value'
			),
			array(
				'Value' => 'TextField'
			)
		);
		
		// Test with auto relation setting
		$form = new Form(new TableFieldTest_Controller(), "Form", new FieldSet($tf), new FieldSet());
		$tf->setValue(array(
			'new' => array(
				'Value' => array('one','two',)
			)
		));
		$tf->setRelationAutoSetting(false);
		$o = new TableFieldTest_Object();
		$o->write();
		$form->saveInto($o);
		$this->assertEquals($o->HasManyRelations()->Count(), 0);
	}
	
	function testDataValue() {
		$tf = new TableField(
			'TestTableField',
			'TestTableField',
			array(
				'Currency' => 'Currency'
			),
			array(
				'Currency' => 'CurrencyField'
			)
		);
		$form = new Form(new TableFieldTest_Controller(), "Form", new FieldSet($tf), new FieldSet());
		$tf->setValue(array(
			'new' => array(
				'Currency' => array(
					'$1,234.56',
					'1234.57',
				)
			)
		));
		$data = $form->getData();
		
		// @todo Fix getData()
		//$this->assertEquals($data['TestTableField']['new']['Currency'][0], 1234.56);
		//$this->assertEquals($data['TestTableField']['new']['Currency'][1], 1234.57);
	}
}

/**
 * Stub controller
 */
class TableFieldTest_Controller extends Controller {
	function Link() {
		return 'TableFieldTest/';
	}
}

class TableFieldTest_Object extends DataObject implements TestOnly {
	static $has_many = array(
		"HasManyRelations" => 'TableFieldTest_HasManyRelation'
	);
}

class TableFieldTest_HasManyRelation extends DataObject implements TestOnly {
	static $db = array(
		'Value' => 'Text', 
	);
	
	static $has_one = array(
		'HasOneRelation' => 'TableFieldTest_Object'
	);
}