mirror of
https://github.com/silverstripe/silverstripe-framework
synced 2024-08-26 20:45:47 +02:00
159 lines
4.4 KiB
PHP
159 lines
4.4 KiB
PHP
<?php
|
|
|
|
class DataObjectDuplicationTest extends SapphireTest {
|
|
|
|
protected $usesDatabase = true;
|
|
|
|
protected $extraDataObjects = array(
|
|
'DataObjectDuplicateTestClass1',
|
|
'DataObjectDuplicateTestClass2',
|
|
'DataObjectDuplicateTestClass3'
|
|
);
|
|
|
|
public function testDuplicate() {
|
|
$orig = new DataObjectDuplicateTestClass1();
|
|
$orig->text = 'foo';
|
|
$orig->write();
|
|
|
|
$duplicate = $orig->duplicate();
|
|
$this->assertInstanceOf('DataObjectDuplicateTestClass1', $duplicate,
|
|
'Creates the correct type'
|
|
);
|
|
$this->assertNotEquals($duplicate->ID, $orig->ID,
|
|
'Creates a unique record'
|
|
);
|
|
$this->assertEquals('foo', $duplicate->text,
|
|
'Copies fields'
|
|
);
|
|
$this->assertEquals(2, DataObjectDuplicateTestClass1::get()->Count(),
|
|
'Only creates a single duplicate'
|
|
);
|
|
}
|
|
|
|
public function testDuplicateHasOne() {
|
|
$relationObj = new DataObjectDuplicateTestClass1();
|
|
$relationObj->text = 'class1';
|
|
$relationObj->write();
|
|
|
|
$orig = new DataObjectDuplicateTestClass2();
|
|
$orig->text = 'class2';
|
|
$orig->oneID = $relationObj->ID;
|
|
$orig->write();
|
|
|
|
$duplicate = $orig->duplicate();
|
|
$this->assertEquals($relationObj->ID, $duplicate->oneID,
|
|
'Copies has_one relationship'
|
|
);
|
|
$this->assertEquals(2, DataObjectDuplicateTestClass2::get()->Count(),
|
|
'Only creates a single duplicate'
|
|
);
|
|
$this->assertEquals(1, DataObjectDuplicateTestClass1::get()->Count(),
|
|
'Does not create duplicate of has_one relationship'
|
|
);
|
|
}
|
|
|
|
|
|
public function testDuplicateManyManyClasses() {
|
|
//create new test classes below
|
|
$one = new DataObjectDuplicateTestClass1();
|
|
$two = new DataObjectDuplicateTestClass2();
|
|
$three = new DataObjectDuplicateTestClass3();
|
|
|
|
//set some simple fields
|
|
$text1 = "Test Text 1";
|
|
$text2 = "Test Text 2";
|
|
$text3 = "Test Text 3";
|
|
$one->text = $text1;
|
|
$two->text = $text2;
|
|
$three->text = $text3;
|
|
|
|
//write the to DB
|
|
$one->write();
|
|
$two->write();
|
|
$three->write();
|
|
|
|
//create relations
|
|
$one->twos()->add($two);
|
|
$one->threes()->add($three);
|
|
|
|
$one = DataObject::get_by_id("DataObjectDuplicateTestClass1", $one->ID);
|
|
$two = DataObject::get_by_id("DataObjectDuplicateTestClass2", $two->ID);
|
|
$three = DataObject::get_by_id("DataObjectDuplicateTestClass3", $three->ID);
|
|
|
|
//test duplication
|
|
$oneCopy = $one->duplicate();
|
|
$twoCopy = $two->duplicate();
|
|
$threeCopy = $three->duplicate();
|
|
|
|
$oneCopy = DataObject::get_by_id("DataObjectDuplicateTestClass1", $oneCopy->ID);
|
|
$twoCopy = DataObject::get_by_id("DataObjectDuplicateTestClass2", $twoCopy->ID);
|
|
$threeCopy = DataObject::get_by_id("DataObjectDuplicateTestClass3", $threeCopy->ID);
|
|
|
|
$this->assertNotNull($oneCopy, "Copy of 1 exists");
|
|
$this->assertNotNull($twoCopy, "Copy of 2 exists");
|
|
$this->assertNotNull($threeCopy, "Copy of 3 exists");
|
|
|
|
$this->assertEquals($text1, $oneCopy->text);
|
|
$this->assertEquals($text2, $twoCopy->text);
|
|
$this->assertEquals($text3, $threeCopy->text);
|
|
|
|
$this->assertNotEquals($one->twos()->Count(), $oneCopy->twos()->Count(),
|
|
"Many-to-one relation not copied (has_many)");
|
|
$this->assertEquals($one->threes()->Count(), $oneCopy->threes()->Count(),
|
|
"Object has the correct number of relations");
|
|
$this->assertEquals($three->ones()->Count(), $threeCopy->ones()->Count(),
|
|
"Object has the correct number of relations");
|
|
|
|
$this->assertEquals($one->ID, $twoCopy->one()->ID,
|
|
"Match between relation of copy and the original");
|
|
$this->assertEquals(0, $oneCopy->twos()->Count(),
|
|
"Many-to-one relation not copied (has_many)");
|
|
$this->assertEquals($three->ID, $oneCopy->threes()->First()->ID,
|
|
"Match between relation of copy and the original");
|
|
$this->assertEquals($one->ID, $threeCopy->ones()->First()->ID,
|
|
"Match between relation of copy and the original");
|
|
}
|
|
|
|
}
|
|
|
|
|
|
class DataObjectDuplicateTestClass1 extends DataObject implements TestOnly {
|
|
|
|
private static $db = array(
|
|
'text' => 'Varchar'
|
|
);
|
|
|
|
private static $has_many = array(
|
|
'twos' => 'DataObjectDuplicateTestClass2'
|
|
);
|
|
|
|
private static $many_many = array(
|
|
'threes' => 'DataObjectDuplicateTestClass3'
|
|
);
|
|
}
|
|
|
|
class DataObjectDuplicateTestClass2 extends DataObject implements TestOnly {
|
|
|
|
private static $db = array(
|
|
'text' => 'Varchar'
|
|
);
|
|
|
|
private static $has_one = array(
|
|
'one' => 'DataObjectDuplicateTestClass1'
|
|
);
|
|
|
|
}
|
|
|
|
class DataObjectDuplicateTestClass3 extends DataObject implements TestOnly {
|
|
|
|
private static $db = array(
|
|
'text' => 'Varchar'
|
|
);
|
|
|
|
private static $belongs_many_many = array(
|
|
'ones' => 'DataObjectDuplicateTestClass1'
|
|
);
|
|
}
|
|
|
|
|