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.");
    }

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

    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.");
    }

    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->check_log_err(0, (char *) "Length (0) is 0", false);
    Test->check_log_err(0, (char *) "Unable to parse query", false);
    Test->check_log_err(0, (char *) "query string allocation failed", false);

    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;
}