ApiQueryTestBase.php

Go to the documentation of this file.

<?php
abstract class ApiQueryTestBase extends ApiTestCase {

    const PARAM_ASSERT = <<<STR
Each parameter must be an array of two elements,
first - an array of params to the API call,
and the second array - expected results as returned by the API
STR;

    protected function merge( /*...*/ ) {
        $request = array();
        $expected = array();
        foreach ( func_get_args() as $v ) {
            list( $req, $exp ) = $this->validateRequestExpectedPair( $v );
            $request = array_merge_recursive( $request, $req );
            $this->mergeExpected( $expected, $exp );
        }

        return array( $request, $expected );
    }

    private function validateRequestExpectedPair( $v ) {
        $this->assertType( 'array', $v, self::PARAM_ASSERT );
        $this->assertEquals( 2, count( $v ), self::PARAM_ASSERT );
        $this->assertArrayHasKey( 0, $v, self::PARAM_ASSERT );
        $this->assertArrayHasKey( 1, $v, self::PARAM_ASSERT );
        $this->assertType( 'array', $v[0], self::PARAM_ASSERT );
        $this->assertType( 'array', $v[1], self::PARAM_ASSERT );

        return $v;
    }

    private function mergeExpected( &$all, $item ) {
        foreach ( $item as $k => $v ) {
            if ( array_key_exists( $k, $all ) ) {
                if ( is_array( $all[$k] ) ) {
                    $this->mergeExpected( $all[$k], $v );
                } else {
                    $this->assertEquals( $all[$k], $v );
                }
            } else {
                $all[$k] = $v;
            }
        }
    }

    protected function check( $values ) {
        list( $req, $exp ) = $this->validateRequestExpectedPair( $values );
        if ( !array_key_exists( 'action', $req ) ) {
            $req['action'] = 'query';
        }
        foreach ( $req as &$val ) {
            if ( is_array( $val ) ) {
                $val = implode( '|', array_unique( $val ) );
            }
        }
        $result = $this->doApiRequest( $req );
        $this->assertResult( array( 'query' => $exp ), $result[0], $req );
    }

    protected function assertResult( $exp, $result, $message = '' ) {
        try {
            $exp = self::sanitizeResultArray( $exp );
            $result = self::sanitizeResultArray( $result );
            $this->assertEquals( $exp, $result );
        } catch ( PHPUnit_Framework_ExpectationFailedException $e ) {
            if ( is_array( $message ) ) {
                $message = http_build_query( $message );
            }
            throw new PHPUnit_Framework_ExpectationFailedException(
                $e->getMessage() . "\nRequest: $message",
                new PHPUnit_Framework_ComparisonFailure(
                    $exp,
                    $result,
                    print_r( $exp, true ),
                    print_r( $result, true ),
                    false,
                    $e->getComparisonFailure()->getMessage() . "\nRequest: $message"
                )
            );
        }
    }

    private static function sanitizeResultArray( $result ) {
        unset( $result['pageid'] );
        foreach ( $result as $key => $value ) {
            if ( is_array( $value ) ) {
                $result[$key] = self::sanitizeResultArray( $value );
            }
        }

        // Sort the result by keys, then take advantage of how array_merge will
        // renumber numeric keys while leaving others alone.
        ksort( $result );
        return array_merge( $result );
    }
}