silverstripe-framework/tests/php/ORM/SQLUpdateTest.php

<?php

namespace SilverStripe\ORM\Tests;

use SilverStripe\ORM\Queries\SQLUpdate;
use SilverStripe\ORM\DB;
use SilverStripe\ORM\DataObject;
use SilverStripe\Dev\SapphireTest;

/**
 * Tests for {@see SQLUpdate}
 */
class SQLUpdateTest extends SapphireTest {

	public static $fixture_file = 'SQLUpdateTest.yml';

	protected $extraDataObjects = array(
		SQLUpdateTest\TestBase::class,
		SQLUpdateTest\TestChild::class
	);

	public function testEmptyQueryReturnsNothing() {
		$query = new SQLUpdate();
		$this->assertSQLEquals('', $query->sql($parameters));
	}

	public function testBasicUpdate() {
		$query = SQLUpdate::create()
				->setTable('"SQLUpdateTestBase"')
				->assign('"Description"', 'Description 1a')
				->addWhere(array('"Title" = ?' => 'Object 1'));
		$sql = $query->sql($parameters);

		// Check SQL
		$this->assertSQLEquals('UPDATE "SQLUpdateTestBase" SET "Description" = ? WHERE ("Title" = ?)', $sql);
		$this->assertEquals(array('Description 1a', 'Object 1'), $parameters);

		// Check affected rows
		$query->execute();
		$this->assertEquals(1, DB::affected_rows());

		// Check item updated
		$item = DataObject::get_one(SQLUpdateTest\TestBase::class, array('"Title"' => 'Object 1'));
		$this->assertEquals('Description 1a', $item->Description);
	}
}