Yiisoft\Db\TestSupport\TestBatchQueryResultTrait::testQuery()

Complexity

Conditions	7
Paths	64

Size

Total Lines	112
Code Lines	62

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

<?php
    public function testQuery(): void
    {
        /* initialize property test */
        $db = $this->getConnection(true);

It seems like getConnection() must be provided by classes using this trait. How about adding it as abstract method to this trait?

        $query = new Query($db);

        $query->from('customer')->orderBy('id');

        $result = $query->batch(2);

        $this->assertInstanceOf(BatchQueryResultInterface::class, $result);

It seems like assertInstanceOf() must be provided by classes using this trait. How about adding it as abstract method to this trait?

        $this->assertEquals(2, $result->getBatchSize());

It seems like assertEquals() must be provided by classes using this trait. How about adding it as abstract method to this trait?

        $this->assertSame($result->getQuery(), $query);

It seems like assertSame() must be provided by classes using this trait. How about adding it as abstract method to this trait?

        /* normal query */
        $query = new Query($db);

        $query->from('customer')->orderBy('id');

        $allRows = [];

        $batch = $query->batch(2);
        $step = 0;
        foreach ($batch as $rows) {
            $allRows = array_merge($allRows, $rows);
            $step++;
        }

        $this->assertCount(3, $allRows);

It seems like assertCount() must be provided by classes using this trait. How about adding it as abstract method to this trait?

        $this->assertEquals(2, $step);
        $this->assertEquals('user1', $allRows[0]['name']);
        $this->assertEquals('user2', $allRows[1]['name']);
        $this->assertEquals('user3', $allRows[2]['name']);

        /* rewind */
        $allRows = [];

        $step = 0;
        foreach ($batch as $rows) {
            $allRows = array_merge($allRows, $rows);
            $step++;
        }

        $this->assertCount(3, $allRows);
        $this->assertEquals(2, $step);

        /* reset */
        $batch->reset();

        /* empty query */
        $query = new Query($db);

        $query->from('customer')->where(['id' => 100]);

        $allRows = [];

        $batch = $query->batch(2);

        foreach ($batch as $rows) {
            $allRows = array_merge($allRows, $rows);
        }

        $this->assertCount(0, $allRows);

        /* query with index */
        $query = new Query($db);

        $query->from('customer')->indexBy('name');

        $allRows = [];

        foreach ($query->batch(2) as $rows) {
            $allRows = array_merge($allRows, $rows);
        }

        $this->assertCount(3, $allRows);
        $this->assertEquals('address1', $allRows['user1']['address']);
        $this->assertEquals('address2', $allRows['user2']['address']);
        $this->assertEquals('address3', $allRows['user3']['address']);

        /* each */
        $query = new Query($db);

        $query->from('customer')->orderBy('id');
        $allRows = [];

        foreach ($query->each(2) as $index => $row) {
            /** @psalm-suppress PossiblyNullArrayOffset */
            $allRows[$index] = $row;
        }
        $this->assertCount(3, $allRows);
        $this->assertEquals('user1', $allRows[0]['name']);
        $this->assertEquals('user2', $allRows[1]['name']);
        $this->assertEquals('user3', $allRows[2]['name']);

        /* each with key */
        $query = new Query($db);

        $query->from('customer')->orderBy('id')->indexBy('name');

        $allRows = [];

        foreach ($query->each(100) as $key => $row) {
            /** @psalm-suppress PossiblyNullArrayOffset */
            $allRows[$key] = $row;
        }

        $this->assertCount(3, $allRows);
        $this->assertEquals('address1', $allRows['user1']['address']);
        $this->assertEquals('address2', $allRows['user2']['address']);
        $this->assertEquals('address3', $allRows['user3']['address']);
    }