/**
 * @file mxs716.cpp Test for MXS-716 ("Access Denied: User without global privileges on a schema
 * but with grants only on some tables can't connect if the default schema is specified
 * in the connection string")
 *
 * - Connect using different default databases with database and table level grants.
 */


#include <iostream>
#include <unistd.h>
#include "testconnections.h"
#include "sql_t1.h"

void run_test(TestConnections* Test, const char* database)
{

    Test->set_timeout(20);
    Test->tprintf("Trying to connect using 'table_privilege'@'%%' to database '%s'", database);

    MYSQL* conn = open_conn_db(Test->maxscales->rwsplit_port[0],
                               Test->maxscales->IP[0],
                               database,
                               "table_privilege",
                               "pass",
                               Test->ssl);

    if (conn && mysql_errno(conn) == 0)
    {
        Test->set_timeout(20);
        Test->tprintf("Trying SELECT on %s.t1", database);
        Test->try_query(conn, "SELECT * FROM t1");
    }
    else
    {
        Test->add_result(1, "Failed to connect using database '%s': %s", database, mysql_error(conn));
    }

    mysql_close(conn);
}

int main(int argc, char* argv[])
{
    TestConnections* Test = new TestConnections(argc, argv);

    Test->maxscales->connect_maxscale(0);
    Test->tprintf("Preparing test");
    Test->set_timeout(180);
    execute_query(Test->maxscales->conn_rwsplit[0], "DROP DATABASE IF EXISTS db1");
    execute_query(Test->maxscales->conn_rwsplit[0], "DROP DATABASE IF EXISTS db2");
    execute_query(Test->maxscales->conn_rwsplit[0], "DROP DATABASE IF EXISTS db3");
    execute_query(Test->maxscales->conn_rwsplit[0], "DROP DATABASE IF EXISTS db4");
    execute_query(Test->maxscales->conn_rwsplit[0], "CREATE DATABASE db1");
    execute_query(Test->maxscales->conn_rwsplit[0], "CREATE DATABASE db2");
    execute_query(Test->maxscales->conn_rwsplit[0], "CREATE DATABASE db3");
    execute_query(Test->maxscales->conn_rwsplit[0], "CREATE DATABASE db4");
    execute_query(Test->maxscales->conn_rwsplit[0], "CREATE TABLE db1.t1 (id INT)");
    execute_query(Test->maxscales->conn_rwsplit[0], "CREATE TABLE db2.t1 (id INT)");
    execute_query(Test->maxscales->conn_rwsplit[0], "CREATE TABLE db3.t1 (id INT)");
    execute_query(Test->maxscales->conn_rwsplit[0], "CREATE TABLE db4.t1 (id INT)");
    execute_query(Test->maxscales->conn_rwsplit[0], "INSERT INTO db1.t1  VALUES (1)");
    execute_query(Test->maxscales->conn_rwsplit[0], "INSERT INTO db2.t1  VALUES (1)");
    execute_query(Test->maxscales->conn_rwsplit[0], "INSERT INTO db3.t1  VALUES (1)");
    execute_query(Test->maxscales->conn_rwsplit[0], "INSERT INTO db4.t1  VALUES (1)");
    execute_query(Test->maxscales->conn_rwsplit[0],
                  "CREATE USER 'table_privilege'@'%%' IDENTIFIED BY 'pass'");
    execute_query(Test->maxscales->conn_rwsplit[0], "GRANT SELECT ON db1.* TO 'table_privilege'@'%%'");
    execute_query(Test->maxscales->conn_rwsplit[0], "GRANT SELECT ON db2.* TO 'table_privilege'@'%%'");
    execute_query(Test->maxscales->conn_rwsplit[0], "GRANT SELECT ON db3.t1 TO 'table_privilege'@'%%'");
    execute_query(Test->maxscales->conn_rwsplit[0], "GRANT SELECT ON db4.t1 TO 'table_privilege'@'%%'");

    Test->repl->sync_slaves();

    run_test(Test, "db1");
    run_test(Test, "db2");
    run_test(Test, "db3");
    run_test(Test, "db4");

    Test->tprintf("Cleaning up...");
    Test->set_timeout(60);
    Test->maxscales->connect_maxscale(0);
    execute_query(Test->maxscales->conn_rwsplit[0], "DROP DATABASE db1");
    execute_query(Test->maxscales->conn_rwsplit[0], "DROP DATABASE db2");
    execute_query(Test->maxscales->conn_rwsplit[0], "DROP DATABASE db3");
    execute_query(Test->maxscales->conn_rwsplit[0], "DROP DATABASE db4");
    execute_query(Test->maxscales->conn_rwsplit[0], "DROP USER 'table_privilege'@'%%'");

    int rval = Test->global_result;
    delete Test;
    return rval;
}