<?php
/**
 * php7-mysql-shim
 *
 * @author Davey Shafik <dshafik@akamai.com>
 * @copyright Copyright 2015 Akamai Technologies, Inc. All rights reserved.
 * @license Apache 2.0
 * @link https://github.com/akamai-open/php7-mysql-shim
 * @link https://developer.akamai.com
 */

namespace Dshafik\MySQL\Tests;

class MySqlShimTest extends \PHPUnit_Framework_TestCase
{
    static $host;
    static $container;
    static $bin = [];

    public function test_mysql_connect()
    {
        $mysql = \mysql_connect(static::$host, 'root');
        $this->assertConnection($mysql);
    }

    /**
     * @expectedException \PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessageRegExp /^mysql(i?)_connect\(\): (\(HY000\/1045\): )?Access denied for user 'baduser'@'(.*?)' \(using password: YES\)$/
     */
    public function test_mysql_connect_fail()
    {
        $mysql = \mysql_connect(static::$host, "baduser", "badpass");
    }

    /**
     * @expectedException \PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessage Argument $new is no longer supported in PHP > 7
     * @requires PHP 7.0.0
     */
    public function test_mysql_connect_new()
    {
        $mysql = \mysql_connect(static::$host, 'root', null, true);
    }

    public function test_mysql_connect_options()
    {
        $mysql = \mysql_connect(static::$host, 'root', null, false, MYSQL_CLIENT_COMPRESS);
        $this->assertConnection($mysql);
    }

    /**
     * @expectedException \PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessageRegExp /^mysql((i_real)?)_connect\(\): (\(HY000\/1045\): )?Access denied for user 'baduser'@'(.*?)' \(using password: YES\)$/
     */
    public function test_mysql_connect_options_fail()
    {
        \mysql_connect(static::$host, "baduser", "badpass", false, MYSQL_CLIENT_COMPRESS);
    }

    public function test_mysql_connect_multi()
    {
        $conn = \mysql_connect(static::$host, 'root');
        $conn2 = \mysql_connect(static::$host, 'root');

        $this->assertEquals($conn, $conn2);

        $result = \mysql_query("SELECT CONNECTION_ID()", $conn);
        $row = \mysql_fetch_row($result);
        $id = $row[0];

        $result = \mysql_query("SELECT CONNECTION_ID()", $conn2);
        $row = \mysql_fetch_row($result);
        $id2 = $row[0];

        $this->assertEquals($id, $id2);
    }

    public function test_mysql_pconnect()
    {
        $conn = \mysql_pconnect(static::$host, 'root');

        $result = \mysql_query("SELECT 'persistent'", $conn);
        $row = \mysql_fetch_row($result);
        $this->assertEquals('persistent', $row[0]);
    }

    public function test_mysql_query_ddl()
    {
        $conn = \mysql_connect(static::$host, 'root');
        $result = \mysql_query("CREATE DATABASE shim_test CHARACTER SET latin1;");
        $this->assertTrue($result);
        $result = \mysql_select_db('shim_test');
        $this->assertTrue($result);
        $result = \mysql_query(
            "CREATE TABLE testing (
                id int AUTO_INCREMENT,
                one varchar(255),
                two varchar(255),
                three varchar(255),
                four varchar(255),
                five varchar(255),
                six varchar(255),
                seven varchar(255),
                eight varchar(255),
                nine ENUM('one', 'two', '\'three'),
                ten SET('one', 'two', '\'three'),
                eleven MEDIUMTEXT,
                INDEX one_idx (one),
                UNIQUE INDEX two_unq (two),
                INDEX three_four_idx (three, four),
                UNIQUE INDEX four_five_unq (four, five),
                INDEX seven_eight_idx (seven, eight),
                UNIQUE INDEX seven_eight_unq (seven, eight),
                PRIMARY KEY (id)
            ) CHARACTER SET latin1;"
        );
        $this->assertTrue($result, \mysql_error());
    }

    public function test_mysql_query_insert()
    {
        $this->getConnection("shim_test");
        $result = \mysql_query(
            "INSERT INTO
                testing (one, two, three, four, five, six, seven, eight)
             VALUES
                ('1', '1', '1', '1', '1', '1', '1', '1'),
                ('2', '2', '2', '2', '2', '2', '2', '2'),
                ('3', '3', '3', '3', '3', '3', '3', '3'),
                ('4', '4', '4', '4', '4', '4', '4', '4')"
        );

        $this->assertTrue($result, \mysql_error());
    }

    public function test_mysql_query()
    {
        $this->getConnection("shim_test");
        $result = \mysql_query("SELECT VERSION()");

        $this->assertResult($result);
    }

    public function test_mysql_query_nodata()
    {
        $this->getConnection("shim_test");
        $result = \mysql_query("SET @test = 'foo'");

        $this->assertTrue($result);
    }

    public function test_mysql_query_fail()
    {
        $this->getConnection("shim_test");
        $result = \mysql_query("SELECT VERSION(");

        $this->assertFalse($result);
    }

    public function test_mysql_unbuffered_query()
    {
        $this->getConnection("shim_test");

        $result = \mysql_unbuffered_query("SELECT one, two FROM testing LIMIT 4");
        $this->assertResult($result);
        $i = 0;
        while ($row = \mysql_fetch_assoc($result)) {
            $i++;
        }
        $this->assertEquals(4, $i);

        $result = \mysql_query("SELECT one, two FROM testing LIMIT 4");
        $this->assertResult($result);
    }

    public function test_mysql_unbuffered_query_fail()
    {
        $this->getConnection();

        $result = \mysql_unbuffered_query("SELECT VERSION(");
        $this->assertFalse($result);
    }

    public function test_mysql_unbuffered_query_num_rows()
    {
        $this->getConnection("shim_test");

        $result = \mysql_unbuffered_query("SELECT one, two FROM testing");
        $this->assertResult($result);
        $this->assertEquals(0, \mysql_num_rows($result));
        \mysql_free_result($result);
    }

    public function test_mysql_unbuffered_query_close_legacy()
    {
        if (!version_compare(PHP_VERSION, '7.0.0', '<')) {
            $this->markTestSkipped("PHP < 7.0.0 is required");
        }

        $conn = $this->getConnection("shim_test");

        $result = \mysql_unbuffered_query("SELECT one, two FROM testing");
        $this->assertResult($result);
        try {
            \mysql_close($conn);
        } catch (\PHPUnit_Framework_Error_Notice $e) {
            $this->assertEquals(
                "mysql_close(): Function called without first fetching all rows from a previous unbuffered query",
                $e->getMessage()
            );
        }
    }

    /**
     * @requires PHP 7.0.0
     */
    public function test_mysql_unbuffered_query_close()
    {
        $conn = $this->getConnection("shim_test");
        $result = \mysql_unbuffered_query("SELECT one, two FROM testing");
        $this->assertResult($result);
        \mysql_close($conn);
    }

    public function test_mysql_db_query()
    {
        $this->getConnection();
        $result = \mysql_db_query("shim_test", "SELECT DATABASE()");
        $this->assertResult($result);
        $this->assertEquals("shim_test", \mysql_fetch_row($result)[0]);
        $result = \mysql_db_query("mysql", "SELECT DATABASE()");
        $this->assertResult($result);
        $this->assertEquals("mysql", \mysql_fetch_row($result)[0]);
    }

    public function test_mysql_db_query_fail()
    {
        $this->getConnection();
        $result = \mysql_db_query("nonexistent", "SELECT DATABASE()");
        $this->assertFalse($result);
    }

    public function test_mysql_list_dbs()
    {
        $this->getConnection();
        $result = \mysql_list_dbs();
        $this->assertResult($result);
        while ($row = mysql_fetch_assoc($result)) {
            $this->assertArrayHasKey("Database", $row);
        }
    }

    public function test_mysql_list_tables()
    {
        $this->getConnection();
        $result = \mysql_list_tables("mysql");
        $this->assertResult($result);
        while ($row = mysql_fetch_assoc($result)) {
            $this->assertArrayHasKey("Tables_in_mysql", $row);
        }
    }

    public function test_mysql_list_tables_fail()
    {
        $this->getConnection();
        $result = \mysql_list_tables("nonexistent");
        $this->assertFalse($result);
    }

    public function test_mysql_list_fields()
    {
        $this->getConnection();
        $result = \mysql_list_fields("shim_test", "testing");
        $this->assertResult($result);

        $this->assertEquals("testing", \mysql_field_table($result, 0));
        $this->assertEquals("id", \mysql_field_name($result, 0));
        $this->assertEquals("int", \mysql_field_type($result, 0));
        $this->assertEquals(11, \mysql_field_len($result, 0));
        $this->assertEquals("not_null primary_key auto_increment", \mysql_field_flags($result, 0));

        $this->assertEquals("testing", \mysql_field_table($result, 1));
        $this->assertEquals("one", \mysql_field_name($result, 1));
        $this->assertEquals("string", \mysql_field_type($result, 1));
        $this->assertEquals(255, \mysql_field_len($result, 1));
        $this->assertEquals("multiple_key", \mysql_field_flags($result, 1));

        $this->assertEquals("testing", \mysql_field_table($result, 2));
        $this->assertEquals("two", \mysql_field_name($result, 2));
        $this->assertEquals("string", \mysql_field_type($result, 2));
        $this->assertEquals(255, \mysql_field_len($result, 2));
        $this->assertEquals("unique_key", \mysql_field_flags($result, 2));

        $this->assertEquals("testing", \mysql_field_table($result, 3));
        $this->assertEquals("three", \mysql_field_name($result, 3));
        $this->assertEquals("string", \mysql_field_type($result, 3));
        $this->assertEquals(255, \mysql_field_len($result, 3));
        $this->assertEquals("multiple_key", \mysql_field_flags($result, 3));

        $this->assertEquals("testing", \mysql_field_table($result, 4));
        $this->assertEquals("four", \mysql_field_name($result, 4));
        $this->assertEquals("string", \mysql_field_type($result, 4));
        $this->assertEquals(255, \mysql_field_len($result, 4));
        $this->assertEquals("multiple_key", \mysql_field_flags($result, 4));

        $this->assertEquals("testing", \mysql_field_table($result, 5));
        $this->assertEquals("five", \mysql_field_name($result, 5));
        $this->assertEquals("string", \mysql_field_type($result, 5));
        $this->assertEquals(255, \mysql_field_len($result, 5));
        $this->assertEmpty(\mysql_field_flags($result, 5));

        $this->assertEquals("testing", \mysql_field_table($result, 6));
        $this->assertEquals("six", \mysql_field_name($result, 6));
        $this->assertEquals("string", \mysql_field_type($result, 6));
        $this->assertEquals(255, \mysql_field_len($result, 6));
        $this->assertEmpty(\mysql_field_flags($result, 6));

        $this->assertEquals("testing", \mysql_field_table($result, 7));
        $this->assertEquals("seven", \mysql_field_name($result, 7));
        $this->assertEquals("string", \mysql_field_type($result, 7));
        $this->assertEquals(255, \mysql_field_len($result, 7));
        $this->assertEquals("multiple_key", \mysql_field_flags($result, 7));

        $this->assertEquals("testing", \mysql_field_table($result, 8));
        $this->assertEquals("eight", \mysql_field_name($result, 8));
        $this->assertEquals("string", \mysql_field_type($result, 8));
        $this->assertEquals(255, \mysql_field_len($result, 8));
        $this->assertEmpty(\mysql_field_flags($result, 8));

        $this->assertEquals("testing", \mysql_field_table($result, 9));
        $this->assertEquals("nine", \mysql_field_name($result, 9));
        $this->assertEquals("string", \mysql_field_type($result, 9));
        $this->assertEquals(6, \mysql_field_len($result, 9));
        $this->assertEquals("enum", \mysql_field_flags($result, 9));

        $this->assertEquals("testing", \mysql_field_table($result, 11));
        $this->assertEquals("eleven", \mysql_field_name($result, 11));
        $this->assertEquals("blob", \mysql_field_type($result, 11));
        $this->assertEquals(16777215, \mysql_field_len($result, 11));
        $this->assertEquals("blob", \mysql_field_flags($result, 11));
    }

    /**
     * @expectedException \PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessageRegExp /^mysql_field_name\(\): Field 999 is invalid for MySQL result index .*$/
     */
    public function test_mysql_list_fields_fail()
    {
        $this->getConnection();
        $result = \mysql_list_fields("shim_test", "testing");
        $this->assertResult($result);

        \mysql_field_name($result, 999);
    }

    public function test_mysql_num_fields()
    {
        $this->getConnection('shim_test');
        $result = \mysql_query("SELECT one, two FROM testing LIMIT 1");
        $this->assertResult($result);

        $this->assertEquals(2, \mysql_num_fields($result));
    }

    /**
     * @expectedException \PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessageRegExp /^mysql_num_fields\(\) expects parameter 1 to be resource, boolean given$/
     */
    public function test_mysql_num_fields_fail()
    {
        $this->getConnection('shim_test');
        $result = \mysql_query("SELECT one, two FROM nonexistent");

        \mysql_num_fields($result);
    }

    /**
     * @dataProvider mysql_fetch_DataProvider
     */
    public function test_mysql_fetch($function, $results)
    {
        $this->getConnection("shim_test");

        $result = \mysql_query("SELECT one, two FROM testing");
        $this->assertResult($result);

        $this->assertEquals(sizeof($results), \mysql_num_rows($result));

        $i = 0;
        while ($row = $function($result)) {
            $this->assertEquals($results[$i], $row);
            $i++;
        }

        $this->assertEquals(sizeof($results), $i);
    }

    public function test_mysql_num_rows()
    {
        $this->getConnection("shim_test");

        $result = \mysql_query("SELECT * FROM testing");
        $this->assertResult($result);
        $this->assertEquals(4, \mysql_num_rows($result));
    }

    public function test_mysql_affected_rows()
    {
        $this->getConnection("shim_test");

        $result = \mysql_query("UPDATE testing SET one = one + 1000, two = two + 1000 ORDER BY one DESC LIMIT 4");
        $this->assertTrue($result);
        $this->assertEquals(4, \mysql_affected_rows());
        $result = \mysql_query("UPDATE testing SET one = one - 1000, two = two - 1000 ORDER BY one DESC LIMIT 4");
        $this->assertTrue($result, \mysql_error());
        $this->assertEquals(4, \mysql_affected_rows());
    }

    public function test_mysql_close()
    {
        \mysql_connect(static::$host, 'root');
        $this->assertTrue(\mysql_close());
    }

    /**
     * @expectedException \PHPUnit_Framework_Error_Warning
     * @expectedExceptionMessage mysql_close(): no MySQL-Link resource supplied
     */
    public function test_mysql_close_fail()
    {
        \mysql_close();
    }

    public function tearDown()
    {
        @\mysql_close();
    }

    public static function setUpBeforeClass()
    {
        error_reporting(E_ALL & ~E_DEPRECATED);
        if (getenv('TRAVIS') === false) {
            fwrite(STDERR, "=> Finding binaries\n");
            static::$bin['dm'] = $dm = exec('/usr/bin/env which docker-machine');
            static::$bin['docker'] = $docker = exec('/usr/bin/env which docker');
            if (empty($dm) && empty($docker)) {
                static::markTestSkipped('Docker is required to run these tests');
            }

            if (!empty($dm)) {
                fwrite(STDERR, "=> Starting Docker Machine\n");
                exec($dm . ' create -d virtualbox mysql-shim');
                exec($dm . ' start mysql-shim');

                $env = '';
                exec($dm . ' env mysql-shim', $env);
                foreach ($env as $line) {
                    if ($line{0} !== '#') {
                        putenv(str_replace(["export ", '"'], "", $line));
                    }
                }
            }

            fwrite(STDERR, "=> Running Docker Container: ");
            static::$container = exec($docker . ' run -e MYSQL_ALLOW_EMPTY_PASSWORD=1 -P -d  mysql/mysql-server:5.7');

            if (empty(static::$container)) {
                static::markTestSkipped("Unable to start docker container");
            }

            fwrite(STDERR, static::$container . "\n");

            fwrite(STDERR, "=> Finding MySQL Host: ");
            static::$host = exec($docker . ' port ' . self::$container . ' 3306');
            fwrite(STDERR, static::$host . "\n");

            if (!empty($dm)) {
                fwrite(STDERR, "=> Using Docker Machine IP: ");
                $info = explode(':', static::$host);
                $port = array_pop($info);
                static::$host = exec($dm . ' ip mysql-shim') . ':' . $port;
                fwrite(STDERR, static::$host . "\n");
            }

            fwrite(STDERR, "=> Waiting on mysqld to start:");
            $out = '';
            while (trim($out) != 'mysqld') {
                $out = exec(static::$bin['docker'] . ' exec ' . static::$container . ' ps ax | awk \'/mysqld/ {print $NF}\'');
            }
            fwrite(STDERR, " started\n");
            sleep(3);

            fwrite(STDERR, "=> Docker Container Running\n\n");

            return;
        }

        static::$host = 'localhost';
    }

    public static function tearDownAfterClass()
    {
        if (getenv('TRAVIS') === false) {
            fwrite(STDERR, "\n\nStopping Docker Container: ");
            $output = exec(static::$bin['docker'] . ' stop ' . static::$container);
            if (trim($output) !== static::$container) {
                fwrite(STDERR, " Failed to stop container!\n");
                return;
            }

            $output = exec(static::$bin['docker'] . ' rm ' . static::$container);
            if (trim($output) !== static::$container) {
                fwrite(STDERR, " Failed to remove container!\n");
                return;
            }
            fwrite(STDERR, "Done\n");

            return;
        }

        \mysql_connect(static::$host, "root");
        \mysql_query("DROP DATABASE shim_test");
    }

    public function mysql_fetch_DataProvider()
    {
        $numeric = [
            ['1', '1'],
            ['2', '2'],
            ['3', '3'],
            ['4', '4'],
        ];

        $assoc = [
            ['one' => '1', 'two' => '1'],
            ['one' => '2', 'two' => '2'],
            ['one' => '3', 'two' => '3'],
            ['one' => '4', 'two' => '4'],
        ];

        $array = [
            ['1', '1', 'one' => '1', 'two' => '1'],
            ['2', '2', 'one' => '2', 'two' => '2'],
            ['3', '3', 'one' => '3', 'two' => '3'],
            ['4', '4', 'one' => '4', 'two' => '4'],
        ];

        $object = [
            (object) ['one' => '1', 'two' => '1'],
            (object) ['one' => '2', 'two' => '2'],
            (object) ['one' => '3', 'two' => '3'],
            (object) ['one' => '4', 'two' => '4'],
        ];

        return [
            [
                'function' => 'mysql_fetch_array',
                'results' => $array
            ],
            [
                'function' => 'mysql_fetch_assoc',
                'results' => $assoc
            ],
            [
                'function' => 'mysql_fetch_row',
                'results' => $numeric
            ],
            [
                'function' => 'mysql_fetch_object',
                'results' => $object,
            ]
        ];
    }

    /**
     * @param $result
     */
    protected function assertResult($result)
    {
        $this->assertTrue(
            is_resource($result) && get_resource_type($result) == 'mysql result'
            || $result instanceof \mysqli_result,
            \mysql_error()
        );
    }

    protected function getConnection($db = null)
    {
        $mysql = \mysql_connect(static::$host, 'root');
        $this->assertConnection($mysql);

        \mysql_query("SET NAMES latin1");

        if ($db !== null) {
            $this->assertTrue(\mysql_select_db($db));
        }

        return $mysql;
    }

    /**
     * @param $mysql
     */
    protected function assertConnection($mysql)
    {
        $this->assertTrue(
            is_resource($mysql) && get_resource_type($mysql) == 'mysql link'
            ||
            $mysql instanceof \mysqli,
            "Not a valid MySQL connection"
        );
    }
}