<?php
class ArrayColumnTest extends \PHPUnit_Framework_TestCase
{
    protected $recordSet;
    protected $multiDataTypes;
    protected $numericColumns;
    protected $mismatchedColumns;

    protected function setUp()
    {
        $this->recordSet = array(
            array(
                'id' => 1,
                'first_name' => 'John',
                'last_name' => 'Doe'
            ),
            array(
                'id' => 2,
                'first_name' => 'Sally',
                'last_name' => 'Smith'
            ),
            array(
                'id' => 3,
                'first_name' => 'Jane',
                'last_name' => 'Jones'
            ),
        );

        $fh = fopen(__FILE__, 'r', true);

        $this->multiDataTypes = array(
            array(
                'id' => 1,
                'value' => new stdClass
            ),
            array(
                'id' => 2,
                'value' => 34.2345
            ),
            array(
                'id' => 3,
                'value' => true
            ),
            array(
                'id' => 4,
                'value' => false
            ),
            array(
                'id' => 5,
                'value' => null
            ),
            array(
                'id' => 6,
                'value' => 1234
            ),
            array(
                'id' => 7,
                'value' => 'Foo'
            ),
            array(
                'id' => 8,
                'value' => $fh
            ),
        );

        $this->numericColumns = array(
            array('aaa', '111'),
            array('bbb', '222'),
            array('ccc', '333', -1 => 'ddd'),
        );

        $this->mismatchedColumns = array(
            array('a' => 'foo', 'b' => 'bar', 'e' => 'bbb'),
            array('a' => 'baz', 'c' => 'qux', 'd' => 'aaa'),
            array('a' => 'eee', 'b' => 'fff', 'e' => 'ggg'),
        );
    }

    public function testFirstNameColumnFromRecordset()
    {
        $expected = array('John', 'Sally', 'Jane');
        $this->assertEquals($expected, array_column($this->recordSet, 'first_name'));
    }

    public function testIdColumnFromRecordset()
    {
        $expected = array(1, 2, 3);
        $this->assertEquals($expected, array_column($this->recordSet, 'id'));
    }

    public function testLastNameColumnKeyedByIdColumnFromRecordset()
    {
        $expected = array(1 => 'Doe', 2 => 'Smith', 3 => 'Jones');
        $this->assertEquals($expected, array_column($this->recordSet, 'last_name', 'id'));
    }

    public function testLastNameColumnKeyedByFirstNameColumnFromRecordset()
    {
        $expected = array('John' => 'Doe', 'Sally' => 'Smith', 'Jane' => 'Jones');
        $this->assertEquals($expected, array_column($this->recordSet, 'last_name', 'first_name'));
    }

    public function testValueColumnWithMultipleDataTypes()
    {
        $expected = array();
        foreach ($this->multiDataTypes as $row) {
            $expected[] = $row['value'];
        }
        $this->assertEquals($expected, array_column($this->multiDataTypes, 'value'));
    }

    public function testValueColumnKeyedByIdWithMultipleDataTypes()
    {
        $expected = array();
        foreach ($this->multiDataTypes as $row) {
            $expected[$row['id']] = $row['value'];
        }
        $this->assertEquals($expected, array_column($this->multiDataTypes, 'value', 'id'));
    }

    public function testNumericColumnKeys1()
    {
        $expected = array('111', '222', '333');
        $this->assertEquals($expected, array_column($this->numericColumns, 1));
    }

    public function testNumericColumnKeys2()
    {
        $expected = array('aaa' => '111', 'bbb' => '222', 'ccc' => '333');
        $this->assertEquals($expected, array_column($this->numericColumns, 1, 0));
    }

    public function testNumericColumnKeys3()
    {
        $expected = array('aaa' => '111', 'bbb' => '222', 'ccc' => '333');
        $this->assertEquals($expected, array_column($this->numericColumns, 1, 0.123));
    }

    public function testNumericColumnKeys4()
    {
        $expected = array(0 => '111', 1 => '222', 'ddd' => '333');
        $this->assertEquals($expected, array_column($this->numericColumns, 1, -1));
    }

    public function testFailureToFindColumn1()
    {
        $this->assertEquals(array(), array_column($this->numericColumns, 2));
    }

    public function testFailureToFindColumn2()
    {
        $this->assertEquals(array(), array_column($this->numericColumns, 'foo'));
    }

    public function testFailureToFindColumn3()
    {
        $expected = array('aaa', 'bbb', 'ccc');
        $this->assertEquals($expected, array_column($this->numericColumns, 0, 'foo'));
    }

    public function testFailureToFindColumn4()
    {
        $this->assertEquals(array(), array_column($this->numericColumns, 3.14));
    }

    public function testSingleDimensionalArray()
    {
        $singleDimension = array('foo', 'bar', 'baz');
        $this->assertEquals(array(), array_column($singleDimension, 1));
    }

    public function testMismatchedColumns1()
    {
        $expected = array('qux');
        $this->assertEquals($expected, array_column($this->mismatchedColumns, 'c'));
    }

    public function testMismatchedColumns2()
    {
        $expected = array('baz' => 'qux');
        $this->assertEquals($expected, array_column($this->mismatchedColumns, 'c', 'a'));
    }

    public function testMismatchedColumns3()
    {
        $expected = array('foo', 'aaa' => 'baz', 'eee');
        $this->assertEquals($expected, array_column($this->mismatchedColumns, 'a', 'd'));
    }

    public function testMismatchedColumns4()
    {
        $expected = array('bbb' => 'foo', 'baz', 'ggg' => 'eee');
        $this->assertEquals($expected, array_column($this->mismatchedColumns, 'a', 'e'));
    }

    public function testMismatchedColumns5()
    {
        $expected = array('bar', 'fff');
        $this->assertEquals($expected, array_column($this->mismatchedColumns, 'b'));
    }

    public function testMismatchedColumns6()
    {
        $expected = array('foo' => 'bar', 'eee' => 'fff');
        $this->assertEquals($expected, array_column($this->mismatchedColumns, 'b', 'a'));
    }

    public function testObjectConvertedToString()
    {
        $f = new Foo();
        $b = new Bar();
        $this->assertEquals(array('Doe', 'Smith', 'Jones'), array_column($this->recordSet, $f));
        $this->assertEquals(array('John' => 'Doe', 'Sally' => 'Smith', 'Jane' => 'Jones'), array_column($this->recordSet, $f, $b));
    }

    /**
     * @expectedException PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessage array_column() expects at least 2 parameters, 0 given
     */
    public function testFunctionWithZeroArgs()
    {
        $foo = array_column();
    }

    public function testFunctionWithZeroArgsReturnValue()
    {
        $foo = @array_column();
        $this->assertNull($foo);
    }

    /**
     * @expectedException PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessage array_column() expects at least 2 parameters, 1 given
     */
    public function testFunctionWithOneArg()
    {
        $foo = array_column(array());
    }

    public function testFunctionWithOneArgReturnValue()
    {
        $foo = @array_column(array());
        $this->assertNull($foo);
    }

    /**
     * @expectedException PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessage array_column() expects parameter 1 to be array, string given
     */
    public function testFunctionWithStringAsFirstArg()
    {
        $foo = array_column('foo', 0);
    }

    public function testFunctionWithStringAsFirstArgReturnValue()
    {
        $foo = @array_column('foo', 0);
        $this->assertNull($foo);
    }

    /**
     * @expectedException PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessage array_column() expects parameter 1 to be array, integer given
     */
    public function testFunctionWithIntAsFirstArg()
    {
        $foo = array_column(1, 'foo');
    }

    public function testFunctionWithIntAsFirstArgReturnValue()
    {
        $foo = @array_column(1, 'foo');
        $this->assertNull($foo);
    }

    /**
     * @expectedException PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessage array_column(): The column key should be either a string or an integer
     */
    public function testFunctionWithColumnKeyAsBool()
    {
        $foo = array_column(array(), true);
    }

    public function testFunctionWithColumnKeyAsBoolReturnValue()
    {
        $foo = @array_column(array(), true);
        $this->assertFalse($foo);
    }

    /**
     * @expectedException PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessage array_column(): The column key should be either a string or an integer
     */
    public function testFunctionWithColumnKeyAsArray()
    {
        $foo = array_column(array(), array());
    }

    public function testFunctionWithColumnKeyAsArrayReturnValue()
    {
        $foo = @array_column(array(), array());
        $this->assertFalse($foo);
    }

    /**
     * @expectedException PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessage array_column(): The index key should be either a string or an integer
     */
    public function testFunctionWithIndexKeyAsBool()
    {
        $foo = array_column(array(), 'foo', true);
    }

    public function testFunctionWithIndexKeyAsBoolReturnValue()
    {
        $foo = @array_column(array(), 'foo', true);
        $this->assertFalse($foo);
    }

    /**
     * @expectedException PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessage array_column(): The index key should be either a string or an integer
     */
    public function testFunctionWithIndexKeyAsArray()
    {
        $foo = array_column(array(), 'foo', array());
    }

    public function testFunctionWithIndexKeyAsArrayReturnValue()
    {
        $foo = @array_column(array(), 'foo', array());
        $this->assertFalse($foo);
    }

    /**
     * @link https://bugs.php.net/bug.php?id=64493
     */
    public function testBugRequest64493()
    {
        // Array from Bug Request #64493 test script
        $rows = array(
            456 => array('id' => '3', 'title' => 'Foo', 'date' => '2013-03-25'),
            457 => array('id' => '5', 'title' => 'Bar', 'date' => '2012-05-20'),
        );

        // pass null as second parameter to get back all columns indexed by third parameter
        $expected1 = array(
            3 => array('id' => '3', 'title' => 'Foo', 'date' => '2013-03-25'),
            5 => array('id' => '5', 'title' => 'Bar', 'date' => '2012-05-20'),
        );
        $this->assertEquals($expected1, array_column($rows, null, 'id'));

        // pass null as second parameter and bogus third param to get back zero-indexed array of all columns
        $expected2 = array(
            array('id' => '3', 'title' => 'Foo', 'date' => '2013-03-25'),
            array('id' => '5', 'title' => 'Bar', 'date' => '2012-05-20'),
        );
        $this->assertEquals($expected2, array_column($rows, null, 'foo'));

        // pass null as second parameter and no third param to get back array_values(input) (same as $expected2)
        $this->assertEquals($expected2, array_column($rows, null));
    }

    public function testObjectCast()
    {
        $columnKey = new ColumnKeyClass();
        $indexKey = new IndexKeyClass();
        $value = new ValueClass();

        $records = array(
            array(
                'id' => $value,
                'first_name' => 'John',
                'last_name' => 'XXX'
            ),
            array(
                'id' => 3245,
                'first_name' => 'Sally',
                'last_name' => 'Smith'
            ),
        );

        $expected = array(
            2135 => 'John',
            3245 => 'Sally',
        );

        $this->assertEquals($expected, array_column($records, $columnKey, $indexKey));
    }
}

class Foo
{
    public function __toString()
    {
        return 'last_name';
    }
}

class Bar
{
    public function __toString()
    {
        return 'first_name';
    }
}

class ColumnKeyClass
{
    function __toString() { return 'first_name'; }
}

class IndexKeyClass
{
    function __toString() { return 'id'; }
}

class ValueClass
{
    function __toString() { return '2135'; }
}