bc4517566a
* Add fetchField(), fetchFieldValues(), fetchRow() and getSQL() to UnionQueryBuilder. * Add unit test. Change-Id: Ia5bb89caea87ab740dcc0e6d4f09c4536333c9ae
103 lines
2.6 KiB
PHP
103 lines
2.6 KiB
PHP
<?php
|
|
|
|
namespace Wikimedia\Tests\Rdbms;
|
|
|
|
use DatabaseTestHelper;
|
|
use Wikimedia\Rdbms\IResultWrapper;
|
|
use Wikimedia\Rdbms\UnionQueryBuilder;
|
|
|
|
/**
|
|
* @covers \Wikimedia\Rdbms\UnionQueryBuilder
|
|
*/
|
|
class UnionQueryBuilderTest extends \MediaWikiUnitTestCase {
|
|
/** @var DatabaseTestHelper */
|
|
private $db;
|
|
|
|
/** @var UnionQueryBuilder */
|
|
private $uqb;
|
|
|
|
protected function setUp(): void {
|
|
$this->db = new DatabaseTestHelper( __CLASS__ . '::' . $this->getName() );
|
|
$this->uqb = $this->db->newUnionQueryBuilder();
|
|
}
|
|
|
|
private function assertSql( $expected ) {
|
|
$actual = $this->uqb->getSQL();
|
|
$actual = preg_replace( '/ +/', ' ', $actual );
|
|
$actual = rtrim( $actual, " " );
|
|
$this->assertEquals( $expected, $actual );
|
|
}
|
|
|
|
private function assertLastSql( $expected ) {
|
|
$actual = $this->db->getLastSqls();
|
|
$actual = preg_replace( '/ +/', ' ', $actual );
|
|
$actual = rtrim( $actual, " " );
|
|
$this->assertEquals( $expected, $actual );
|
|
}
|
|
|
|
public function testGetSql() {
|
|
$this->uqb
|
|
->add( $this->db->newSelectQueryBuilder()
|
|
->select( 'f' )
|
|
->from( 't1' )
|
|
)
|
|
->add( $this->db->newSelectQueryBuilder()
|
|
->select( 'f' )
|
|
->from( 't2' )
|
|
)
|
|
->orderBy( 'f', UnionQueryBuilder::SORT_DESC )
|
|
->limit( 10 )
|
|
->offset( 20 )
|
|
->caller( __METHOD__ );
|
|
$this->assertSql( '(SELECT f FROM t1 ) UNION (SELECT f FROM t2 ) ' .
|
|
'ORDER BY f DESC LIMIT 20,10' );
|
|
}
|
|
|
|
public function testFetchResultSet() {
|
|
$this->uqb
|
|
->add( $this->db->newSelectQueryBuilder()
|
|
->select( 'f' )
|
|
->from( 't1' )
|
|
)
|
|
->add( $this->db->newSelectQueryBuilder()
|
|
->select( 'f' )
|
|
->from( 't2' )
|
|
)
|
|
->caller( __METHOD__ );
|
|
$res = $this->uqb->fetchResultSet();
|
|
$this->assertInstanceOf( IResultWrapper::class, $res );
|
|
$this->assertLastSql( '(SELECT f FROM t1 ) UNION (SELECT f FROM t2 )' );
|
|
}
|
|
|
|
public function testFetchField() {
|
|
$this->uqb
|
|
->add( $this->db->newSelectQueryBuilder()
|
|
->select( 'f' )
|
|
->from( 't1' )
|
|
)
|
|
->add( $this->db->newSelectQueryBuilder()
|
|
->select( 'f' )
|
|
->from( 't2' )
|
|
)
|
|
->caller( __METHOD__ );
|
|
$f = $this->uqb->fetchField();
|
|
$this->assertFalse( $f );
|
|
$this->assertLastSql( '(SELECT f FROM t1 ) UNION (SELECT f FROM t2 ) LIMIT 1' );
|
|
}
|
|
|
|
public function testFetchRow() {
|
|
$this->uqb
|
|
->add( $this->db->newSelectQueryBuilder()
|
|
->select( 'f' )
|
|
->from( 't1' )
|
|
)
|
|
->add( $this->db->newSelectQueryBuilder()
|
|
->select( 'f' )
|
|
->from( 't2' )
|
|
)
|
|
->caller( __METHOD__ );
|
|
$row = $this->uqb->fetchRow();
|
|
$this->assertFalse( $row );
|
|
$this->assertLastSql( '(SELECT f FROM t1 ) UNION (SELECT f FROM t2 ) LIMIT 1' );
|
|
}
|
|
}
|