MaxScale/maxscale-system-test/sharding.cpp

/**
 * @file sharding.cpp - Schema router test and regression test for MXS-78, MXS-79
 *
 * @verbatim
 *  [MySQL Monitor]
 *  type=monitor
 *  module=mysqlmon
 *  servers= server1, server2,server3  ,server4
 *  user=skysql
 *  passwd= skysql
 *
 *  [Sharding router]
 *  type=service
 *  router=schemarouter
 *  servers=server1,     server2,              server3,server4
 *  user=skysql
 *  passwd=skysql
 *  auth_all_servers=1
 *  filters=QLA
 *
 *  @endverbatim
 * - stop all slaves in Master/Slave setup
 * - restrt Maxscale
 * - using direct connection to backend nodes
 *    - create user0...userN users on all nodes
 *    - create sharddb on all nodes
 *    - create database 'shard_db%d" on node %d (% from 0 to N)
 *    - GRANT SELECT,USAGE,CREATE ON shard_db.* TO 'user%d'@'%%' only on node %d
 * - for every user%d
 *   - open connection to schemarouter using user%d
 * - CREATE TABLE table%d (x1 int, fl int)
 * - check if Maxscale alive
 */


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

int main(int argc, char* argv[])
{

    TestConnections* Test = new TestConnections(argc, argv);
    Test->set_timeout(30);
    int i, j;
    char str[256];
    char str1[256];
    char user_str[256];
    char pass_str[256];

    Test->repl->execute_query_all_nodes("STOP SLAVE");
    Test->repl->connect();

    for (i = 0; i < Test->repl->N; i++)     // nodes
    {
        for (j = 0; j < Test->repl->N; j++)     // users
        {
            Test->set_timeout(30);
            execute_query(Test->repl->nodes[i], "DROP USER 'user%d'@'%%';", j);
            execute_query(Test->repl->nodes[i], "CREATE USER 'user%d'@'%%' IDENTIFIED BY 'pass%d';", j, j);
            execute_query(Test->repl->nodes[i], "DROP DATABASE IF EXISTS shard_db");
        }

        execute_query(Test->repl->nodes[i], "DROP DATABASE IF EXISTS shard_db%d", i);
        execute_query(Test->repl->nodes[i], "CREATE DATABASE shard_db%d", i);
    }
    Test->stop_timeout();

    for (i = 0; i < Test->repl->N; i++)     // nodes
    {
        Test->set_timeout(30);
        Test->tprintf("Node %d\t", i);
        Test->tprintf("Creating shard_db\t");
        execute_query(Test->repl->nodes[i], "CREATE DATABASE shard_db");
        Test->add_result(execute_query(Test->repl->nodes[i],
                                       "GRANT SELECT,USAGE,CREATE ON shard_db.* TO 'user%d'@'%%'",
                                       i),
                         "Query should succeed.");
        execute_query(Test->repl->nodes[i], "FLUSH PRIVILEGES");
    }

    Test->repl->close_connections();
    Test->stop_timeout();

    MYSQL* conn;
    for (i = 0; i < Test->repl->N; i++)
    {
        Test->set_timeout(30);
        sprintf(user_str, "user%d", i);
        sprintf(pass_str, "pass%d", i);
        Test->tprintf("Open connection to Sharding router using %s %s\n", user_str, pass_str);
        conn = open_conn_db(Test->maxscales->rwsplit_port[0],
                            Test->maxscales->IP[0],
                            (char*) "shard_db",
                            user_str,
                            pass_str,
                            Test->ssl);
        Test->add_result(execute_query(conn, "CREATE TABLE table%d (x1 int, fl int);", i),
                         "Query should succeed.");
    }
    sleep(6); // The router is configured to refresh the shard map if older than 5 seconds.

    for (i = 0; i < Test->repl->N; i++)
    {
        Test->set_timeout(30);
        sprintf(user_str, "user%d", i);
        sprintf(pass_str, "pass%d", i);
        Test->tprintf("Open connection to Sharding router using %s %s\n", user_str, pass_str);
        conn = open_conn_db(Test->maxscales->rwsplit_port[0],
                            Test->maxscales->IP[0],
                            (char*) "shard_db",
                            user_str,
                            pass_str,
                            Test->ssl);

        sprintf(str, "SHOW TABLES;");
        Test->tprintf("%s\n", str);
        sprintf(str1, "table%d", i);
        Test->tprintf("Table should be %s\n", str1);
        Test->add_result(execute_query_check_one(conn, str, str1), "check failed\n");
        mysql_ping(conn);
        mysql_close(conn);
    }

    Test->maxscales->connect_rwsplit(0);

    Test->tprintf("Trying USE shard_db\n");
    execute_query(Test->maxscales->conn_rwsplit[0], "USE shard_db");

    for (i = 0; i < Test->repl->N; i++)
    {
        Test->add_result(execute_query(Test->maxscales->conn_rwsplit[0], "USE shard_db%d", i),
                         "Query should succeed.");
    }

    mysql_close(Test->maxscales->conn_rwsplit[0]);

    Test->tprintf("Trying to connect with empty database name\n");
    conn = open_conn_db(Test->maxscales->rwsplit_port[0],
                        Test->maxscales->IP[0],
                        (char*) "",
                        user_str,
                        pass_str,
                        Test->ssl);
    mysql_close(conn);

    Test->stop_timeout();
    Test->log_excludes(0, "Length (0) is 0");
    Test->log_excludes(0, "Unable to parse query");
    Test->log_excludes(0, "query string allocation failed");

    Test->repl->connect();
    /** Cleanup */
    for (i = 0; i < Test->repl->N; i++)
    {
        for (j = 0; j < Test->repl->N; j++)
        {
            Test->set_timeout(30);
            execute_query(Test->repl->nodes[i], "DROP USER 'user%d'@'%%';", j);
            execute_query(Test->repl->nodes[i], "DROP DATABASE IF EXISTS shard_db");
        }

        execute_query(Test->repl->nodes[i], "DROP DATABASE IF EXISTS shard_db%d", i);
    }

    Test->repl->execute_query_all_nodes("START SLAVE");
    sleep(1);
    Test->repl->fix_replication();
    int rval = Test->global_result;
    delete Test;
    return rval;
}