silverstripe-framework/tests/security/MemberCsvBulkLoaderTest.php

<?php
/**
 * @package sapphire
 * @subpackage tests
 */
class MemberCsvBulkLoaderTest extends SapphireTest {
	static $fixture_file = 'sapphire/tests/security/MemberCsvBulkLoaderTest.yml';
	
	function testNewImport() {
		$loader = new MemberCsvBulkLoader();
		$results = $loader->load('sapphire/tests/security/MemberCsvBulkLoaderTest.csv');
		$created = $results->Created()->toArray();
		$this->assertEquals(count($created), 2);
		$this->assertEquals($created[0]->Email, 'author1@test.com');
		$this->assertEquals($created[1]->Email, 'author2@test.com');
	}
	
	function testOverwriteExistingImport() {
		$author1 = new Member();
		$author1->FirstName = 'author1_first_old';
		$author1->Email = 'author1@test.com';
		$author1->write();
		
		$loader = new MemberCsvBulkLoader();
		$results = $loader->load('sapphire/tests/security/MemberCsvBulkLoaderTest.csv');
		$created = $results->Created()->toArray();
		$this->assertEquals(count($created), 1);
		$updated = $results->Updated()->toArray();
		$this->assertEquals(count($updated), 1);
		$this->assertEquals($created[0]->Email, 'author2@test.com');
		$this->assertEquals($updated[0]->Email, 'author1@test.com');
		$this->assertEquals($updated[0]->FirstName, 'author1_first');
	}
	
	function testAddToPredefinedGroups() {
		$existinggroup = $this->objFromFixture('Group', 'existinggroup');
		
		$loader = new MemberCsvBulkLoader();
		$loader->setGroups(array($existinggroup));
		
		$results = $loader->load('sapphire/tests/security/MemberCsvBulkLoaderTest.csv');
		
		$created = $results->Created()->toArray();
		$this->assertEquals($created[0]->Groups()->column('ID'), array($existinggroup->ID));
		$this->assertEquals($created[1]->Groups()->column('ID'), array($existinggroup->ID));
	}
	
	function testAddToCsvColumnGroupsByCode() {
		$existinggroup = $this->objFromFixture('Group', 'existinggroup');
		
		$loader = new MemberCsvBulkLoader();
		$results = $loader->load('sapphire/tests/security/MemberCsvBulkLoaderTest_withGroups.csv');
		
		$newgroup = DataObject::get_one('Group', sprintf('"Code" = \'%s\'', 'newgroup'));
		$this->assertEquals($newgroup->Title, 'newgroup');
		
		$created = $results->Created()->toArray();
		$this->assertEquals($created[0]->Groups()->column('ID'), array($existinggroup->ID));
		$this->assertEquals($created[1]->Groups()->column('ID'), array($existinggroup->ID, $newgroup->ID));
	}
}