130 lines
3.2 KiB
PHP
130 lines
3.2 KiB
PHP
<?php
|
|
|
|
use SilverStripe\ORM\DB;
|
|
use SilverStripe\ORM\Connect\MySQLiConnector;
|
|
use SilverStripe\ORM\Queries\SQLUpdate;
|
|
use SilverStripe\ORM\DataObject;
|
|
/**
|
|
* @package framework
|
|
* @subpackage testing
|
|
*/
|
|
|
|
class MySQLDatabaseTest extends SapphireTest {
|
|
|
|
protected static $fixture_file = 'MySQLDatabaseTest.yml';
|
|
|
|
protected $extraDataObjects = array(
|
|
'MySQLDatabaseTest_Data'
|
|
);
|
|
|
|
public function testPreparedStatements() {
|
|
if(!(DB::get_connector() instanceof MySQLiConnector)) {
|
|
$this->markTestSkipped('This test requires the current DB connector is MySQLi');
|
|
}
|
|
|
|
// Test preparation of equivalent statemetns
|
|
$result1 = DB::get_connector()->preparedQuery(
|
|
'SELECT "Sort", "Title" FROM "MySQLDatabaseTest_Data" WHERE "Sort" > ? ORDER BY "Sort"',
|
|
array(0)
|
|
);
|
|
|
|
$result2 = DB::get_connector()->preparedQuery(
|
|
'SELECT "Sort", "Title" FROM "MySQLDatabaseTest_Data" WHERE "Sort" > ? ORDER BY "Sort"',
|
|
array(2)
|
|
);
|
|
$this->assertInstanceOf('SilverStripe\\ORM\\Connect\\MySQLStatement', $result1);
|
|
$this->assertInstanceOf('SilverStripe\\ORM\\Connect\\MySQLStatement', $result2);
|
|
|
|
// Also select non-prepared statement
|
|
$result3 = DB::get_connector()->query('SELECT "Sort", "Title" FROM "MySQLDatabaseTest_Data" ORDER BY "Sort"');
|
|
$this->assertInstanceOf('SilverStripe\\ORM\\Connect\\MySQLQuery', $result3);
|
|
|
|
// Iterating one level should not buffer, but return the right result
|
|
$this->assertEquals(
|
|
array(
|
|
'Sort' => 1,
|
|
'Title' => 'First Item'
|
|
),
|
|
$result1->next()
|
|
);
|
|
$this->assertEquals(
|
|
array(
|
|
'Sort' => 2,
|
|
'Title' => 'Second Item'
|
|
),
|
|
$result1->next()
|
|
);
|
|
|
|
// Test first
|
|
$this->assertEquals(
|
|
array(
|
|
'Sort' => 1,
|
|
'Title' => 'First Item'
|
|
),
|
|
$result1->first()
|
|
);
|
|
|
|
// Test seek
|
|
$this->assertEquals(
|
|
array(
|
|
'Sort' => 2,
|
|
'Title' => 'Second Item'
|
|
),
|
|
$result1->seek(1)
|
|
);
|
|
|
|
// Test count
|
|
$this->assertEquals(4, $result1->numRecords());
|
|
|
|
// Test second statement
|
|
$this->assertEquals(
|
|
array(
|
|
'Sort' => 3,
|
|
'Title' => 'Third Item'
|
|
),
|
|
$result2->next()
|
|
);
|
|
|
|
// Test non-prepared query
|
|
$this->assertEquals(
|
|
array(
|
|
'Sort' => 1,
|
|
'Title' => 'First Item'
|
|
),
|
|
$result3->next()
|
|
);
|
|
}
|
|
|
|
public function testAffectedRows() {
|
|
if(!(DB::get_connector() instanceof MySQLiConnector)) {
|
|
$this->markTestSkipped('This test requires the current DB connector is MySQLi');
|
|
}
|
|
|
|
$query = new SQLUpdate('MySQLDatabaseTest_Data');
|
|
$query->setAssignments(array('Title' => 'New Title'));
|
|
|
|
// Test update which affects no rows
|
|
$query->setWhere(array('Title' => 'Bob'));
|
|
$result = $query->execute();
|
|
$this->assertInstanceOf('SilverStripe\\ORM\\Connect\\MySQLQuery', $result);
|
|
$this->assertEquals(0, DB::affected_rows());
|
|
|
|
// Test update which affects some rows
|
|
$query->setWhere(array('Title' => 'First Item'));
|
|
$result = $query->execute();
|
|
$this->assertInstanceOf('SilverStripe\\ORM\\Connect\\MySQLQuery', $result);
|
|
$this->assertEquals(1, DB::affected_rows());
|
|
}
|
|
}
|
|
|
|
class MySQLDatabaseTest_Data extends DataObject implements TestOnly {
|
|
private static $db = array(
|
|
'Title' => 'Varchar',
|
|
'Description' => 'Text',
|
|
'Enabled' => 'Boolean',
|
|
'Sort' => 'Int'
|
|
);
|
|
|
|
private static $default_sort = '"Sort" ASC';
|
|
}
|