/**
 * @file routing_hints.cpp - Test routing hints
 * - execute a number of 'select @@server_id' with different hints and check that
 * query goes to backend according hint
 */


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

#define SERVER1 0
#define SERVER2 1
#define SERVER3 2
#define SERVER4 3

static struct result
{
    const char* query;
    int reply;
} queries[] =
{
    {"select @@server_id; -- maxscale begin route to master", SERVER1},
    {"select @@server_id;", SERVER1},
    {"select @@server_id; -- maxscale route to server server3", SERVER3},
    {"select @@server_id;", SERVER1},
    {"select @@server_id; -- maxscale end", SERVER2},
    {"select @@server_id; -- maxscale named1 prepare route to master", SERVER2},
    {"select @@server_id; -- maxscale named1 begin", SERVER1},
    {"select @@server_id;", SERVER1},
    {"select @@server_id; -- maxscale route to server server3", SERVER3},
    {"select @@server_id;", SERVER1},
    {"select @@server_id; -- maxscale end", SERVER2},
    {"select @@server_id; -- maxscale shorthand1 begin route to server server2", SERVER2},
    {"select @@server_id;", SERVER2},
    {"select @@server_id; -- maxscale route to server server3", SERVER3},
    {"select @@server_id;", SERVER2},
    {"select @@server_id; -- maxscale end", SERVER2},
    {"select @@server_id; # maxscale begin route to master", SERVER1},
    {"select @@server_id;", SERVER1},
    {"select @@server_id; # maxscale route to server server3", SERVER3},
    {"select @@server_id;", SERVER1},
    {"select @@server_id; # maxscale end", SERVER2},
    {"select @@server_id; # maxscale named2 prepare route to master", SERVER2},
    {"select @@server_id; # maxscale named2 begin", SERVER1},
    {"select @@server_id;", SERVER1},
    {"select @@server_id; # maxscale route to server server3", SERVER3},
    {"select @@server_id;", SERVER1},
    {"select @@server_id; # maxscale end", SERVER2},
    {"select @@server_id; # maxscale shorthand2 begin route to server server2", SERVER2},
    {"select @@server_id;", SERVER2},
    {"select @@server_id; # maxscale route to server server3", SERVER3},
    {"select @@server_id;", SERVER2},
    {"select @@server_id; # maxscale end", SERVER2},
    {"select @@server_id/* maxscale begin route to master */;", SERVER1},
    {"select @@server_id;", SERVER1},
    {"select @@server_id/* maxscale route to server server3 */;", SERVER3},
    {"select @@server_id;", SERVER1},
    {"select @@server_id/* maxscale end */;", SERVER2},
    {"select @@server_id/* maxscale named3 prepare route to master */;", SERVER2},
    {"select @@server_id/* maxscale named3 begin */;", SERVER1},
    {"select @@server_id;", SERVER1},
    {"select @@server_id/* maxscale route to server server3 */;", SERVER3},
    {"select @@server_id;", SERVER1},
    {"select @@server_id/* maxscale end */;", SERVER2},
    {"select @@server_id/* maxscale shorthand3 begin route to server server2 */; ", SERVER2},
    {"select @@server_id;", SERVER2},
    {"select @@server_id/* maxscale route to server server3 */;", SERVER3},
    {"select @@server_id;", SERVER2},
    {"select @@server_id/* maxscale end */;", SERVER2},
    {NULL, SERVER1}
};

int main(int argc, char** argv)
{
    TestConnections* test = new TestConnections(argc, argv);
    test->repl->connect();
    test->maxscales->connect_maxscale(0);

    char server_id[test->repl->N][1024];

    /** Get server_id for each node */
    for (int i = 0; i < test->repl->N; i++)
    {
        sprintf(server_id[i], "%d", test->repl->get_server_id(i));
    }

    for (int i = 0; queries[i].query; i++)
    {
        char str[1024];
        find_field(test->maxscales->conn_rwsplit[0], queries[i].query, "@@server_id", str);
        if (strcmp(server_id[queries[i].reply], str) != 0)
        {
            test->add_result(1, "%s: Expected %s but got %s.\n",
                             queries[i].query, server_id[queries[i].reply], str);
        }
    }
    int rval = test->global_result;
    delete test;
    return rval;
}