ComponentSetTest.php
1.72 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
<?php
/**
 * @package framework
 * @subpackage tests
 */
class ComponentSetTest extends SapphireTest {
	
	protected static $fixture_file = 'ComponentSetTest.yml';
	
	protected $extraDataObjects = array(
		'ComponentSetTest_Player',
		'ComponentSetTest_Team'
	);
	
	public function testSetByIDListManyMany() {
		$team1 = $this->objFromFixture('ComponentSetTest_Team', 'team1');
		$player1_team1 = $this->objFromFixture('ComponentSetTest_Player', 'player1_team1');
		$player2 = $this->objFromFixture('ComponentSetTest_Player', 'player2');
		
		$team1->Players()->setByIdList(array(
			$player1_team1->ID,
			$player2->ID
		));
		$team1->flushCache();
		$this->assertContains(
			$player2->ID,
			$team1->Players()->column('ID'),
			'Can add new entry'
		);
		$this->assertContains(
			$player1_team1->ID,
			$team1->Players()->column('ID'),
			'Can retain existing entry'
		);
		
		$team1->Players()->setByIdList(array(
			$player1_team1->ID
		));
		$team1->flushCache();
		$this->assertNotContains(
			$player2->ID,
			$team1->Players()->column('ID'),
			'Can remove existing entry'
		);
		$this->assertContains(
			$player1_team1->ID,
			$team1->Players()->column('ID'),
			'Can retain existing entry'
		);
		
		$team1->Players()->setByIdList(array());
		$team1->flushCache();
		$this->assertEquals(0, $team1->Players()->Count(),
			'Can remove all entries by passing an empty array'
		);
	}
}
class ComponentSetTest_Player extends Member implements TestOnly {
	private static $belongs_many_many = array(
		'Teams' => 'ComponentSetTest_Team'
	);
}
class ComponentSetTest_Team extends DataObject implements TestOnly {
	private static $db = array(
		'Title' => 'Varchar', 
	);
	private static $many_many = array(
		'Players' => 'ComponentSetTest_Player'
	);
}