<?php namespace CodeIgniter\Database\Builder;

use CodeIgniter\Database\BaseBuilder;
use CodeIgniter\Database\MockConnection;
use CodeIgniter\Database\MockQuery;

class UpdateTest extends \CIUnitTestCase
{
	protected $db;

	//--------------------------------------------------------------------

	public function setUp()
	{
		$this->db = new MockConnection([]);
	}

	//--------------------------------------------------------------------

	public function testUpdate()
	{
	    $builder = new BaseBuilder('jobs', $this->db);

		$builder->where('id', 1)->update(['name' => 'Programmer'], null, null, true);

		$expectedSQL = "UPDATE \"jobs\" SET \"name\" = :name WHERE \"id\" = :id";
		$expectedBinds = ['id' => 1, 'name' => 'Programmer'];

		$this->assertEquals($expectedSQL, str_replace("\n", ' ', $builder->getCompiledUpdate()));
		$this->assertEquals($expectedBinds, $builder->getBinds());
	}

	//--------------------------------------------------------------------

	public function testUpdateInternalWhereAndLimit()
	{
		$builder = new BaseBuilder('jobs', $this->db);

		$builder->update(['name' => 'Programmer'], ['id' => 1], 5, true);

		$expectedSQL = "UPDATE \"jobs\" SET \"name\" = :name WHERE \"id\" = :id  LIMIT 5";
		$expectedBinds = ['id' => 1, 'name' => 'Programmer'];

		$this->assertEquals($expectedSQL, str_replace("\n", ' ', $builder->getCompiledUpdate()));
		$this->assertEquals($expectedBinds, $builder->getBinds());
	}

	//--------------------------------------------------------------------

	public function testUpdateWithSet()
	{
		$builder = new BaseBuilder('jobs', $this->db);

		$builder->set('name', 'Programmer')->where('id', 1)->update(null, null, null, true);

		$expectedSQL = "UPDATE \"jobs\" SET \"name\" = :name WHERE \"id\" = :id";
		$expectedBinds = ['id' => 1, 'name' => 'Programmer'];

		$this->assertEquals($expectedSQL, str_replace("\n", ' ', $builder->getCompiledUpdate()));
		$this->assertEquals($expectedBinds, $builder->getBinds());
	}

	//--------------------------------------------------------------------

	public function testUpdateThrowsExceptionWithNoData()
	{
		$builder = new BaseBuilder('jobs', $this->db);

		$this->expectException('CodeIgniter\DatabaseException', 'You must use the "set" method to update an entry.');

		$builder->update(null, null, null, true);
	}

	//--------------------------------------------------------------------

	public function testUpdateBatch()
	{
		$builder = new BaseBuilder('jobs', $this->db);

		$updateData = array(
			['id' => 2, 'name' => 'Comedian', 'description' => 'Theres something in your teeth'],
			['id' => 3, 'name' => 'Cab Driver', 'description' => 'Iam yellow'],
		);

		$this->db->shouldReturn('execute', 1)
		         ->shouldReturn('affectedRows', 1);

		$builder->updateBatch($updateData, 'id');

		$query = $this->db->getLastQuery();

		$this->assertTrue($query instanceof MockQuery);

		$expected = 'UPDATE "jobs" SET "name" = CASE 
WHEN "id" = :id THEN :name
WHEN "id" = :id0 THEN :name0
ELSE "name" END, "description" = CASE 
WHEN "id" = :id THEN :description
WHEN "id" = :id0 THEN :description0
ELSE "description" END
WHERE "id" IN(:id,:id0)';

		$this->assertEquals($expected, $query->getOriginalQuery() );

		$expected = 'UPDATE "jobs" SET "name" = CASE 
WHEN "id" = 2 THEN \'Comedian\'
WHEN "id" = 3 THEN \'Cab Driver\'
ELSE "name" END, "description" = CASE 
WHEN "id" = 2 THEN \'Theres something in your teeth\'
WHEN "id" = 3 THEN \'Iam yellow\'
ELSE "description" END
WHERE "id" IN(2,3)';

		$this->assertEquals($expected, $query->getQuery() );
	}

	//--------------------------------------------------------------------

	public function testUpdateBatchThrowsExceptionWithNoData()
	{
		$builder = new BaseBuilder('jobs', $this->db);

		$this->expectException('CodeIgniter\DatabaseException');
		$this->expectExceptionMessage('You must use the "set" method to update an entry.');

		$builder->updateBatch(null, 'id');
	}

	//--------------------------------------------------------------------

	public function testUpdateBatchThrowsExceptionWithNoID()
	{
		$builder = new BaseBuilder('jobs', $this->db);

		$this->expectException('CodeIgniter\DatabaseException');
		$this->expectExceptionMessage('You must specify an index to match on for batch updates.');

		$builder->updateBatch([]);
	}

	//--------------------------------------------------------------------

	public function testUpdateBatchThrowsExceptionWithEmptySetArray()
	{
		$builder = new BaseBuilder('jobs', $this->db);

		$this->expectException('CodeIgniter\DatabaseException');
		$this->expectExceptionMessage('updateBatch() called with no data');

		$builder->updateBatch([], 'id');
	}

	//--------------------------------------------------------------------

	public function testUpdateWithWhereSameColumn()
	{
		$builder = new BaseBuilder('jobs', $this->db);

		$builder->update(['name' => 'foobar'], ['name' => 'Programmer'], null, true);

		$expectedSQL = 'UPDATE "jobs" SET "name" = :name WHERE "name" = :name0';
		$expectedBinds = ['name' => 'foobar', 'name0' => 'Programmer'];

		$this->assertEquals($expectedSQL, str_replace("\n", ' ', $builder->getCompiledUpdate()));
		$this->assertEquals($expectedBinds, $builder->getBinds());
	}

	//--------------------------------------------------------------------

	public function testUpdateWithWhereSameColumn2()
	{
		// calling order: set() -> where()
		$builder = new BaseBuilder('jobs', $this->db);

		$builder->set('name', 'foobar')
			->where('name', 'Programmer')
			->update(null, null, null, true);

		$expectedSQL = 'UPDATE "jobs" SET "name" = :name WHERE "name" = :name0';
		$expectedBinds = ['name' => 'foobar', 'name0' => 'Programmer'];

		$this->assertEquals($expectedSQL, str_replace("\n", ' ', $builder->getCompiledUpdate()));
		$this->assertEquals($expectedBinds, $builder->getBinds());
	}

	//--------------------------------------------------------------------

	public function testUpdateWithWhereSameColumn3()
	{
		// calling order: where() -> set() in update()
		$builder = new BaseBuilder('jobs', $this->db);

		$builder->where('name', 'Programmer')
			->update(['name' => 'foobar'], null, null, true);

		$expectedSQL = 'UPDATE "jobs" SET "name" = :name0 WHERE "name" = :name';
		$expectedBinds = ['name' => 'Programmer', 'name0' => 'foobar'];

		$this->assertEquals($expectedSQL, str_replace("\n", ' ', $builder->getCompiledUpdate()));
		$this->assertEquals($expectedBinds, $builder->getBinds());
	}

	//--------------------------------------------------------------------

	// @see https://bcit-ci.github.io/CodeIgniter4/database/query_builder.html#updating-data
	public function testSetWithoutEscape()
	{
		$builder = new BaseBuilder('mytable', $this->db);

		$builder->set('field', 'field+1', false)
			->where('id', 2)
			->update(null, null, null, true);

		$expectedSQL = 'UPDATE "mytable" SET field = field+1 WHERE "id" = :id';
		$expectedBinds = ['id' => 2];

		$this->assertEquals($expectedSQL, str_replace("\n", ' ', $builder->getCompiledUpdate()));
		$this->assertEquals($expectedBinds, $builder->getBinds());
	}
}
