proxysql/test/tap/tests/test_server_sess_status-t.cpp

/**
 * @file test_server_sess_status-t.cpp
 * @brief Test checking that ProxySQL 'server_status' value is properly updated for different operations.
 * @details Test should also check that unsupported status like 'SERVER_SESSION_STATE_CHANGED' are never
 *   reported by ProxySQL.
 */

#include <cstring>
#include <string>
#include <stdio.h>
#include <utility>
#include <vector>

#include "mysql.h"

#include "tap.h"
#include "command_line.h"
#include "utils.h"

using std::pair;
using std::string;
using std::vector;

int get_user_def_hg(MYSQL* admin, const string& user) {
	const string sel_q { "SELECT default_hostgroup FROM mysql_users WHERE username='" + user + "'" };
	if (mysql_query(admin, sel_q.c_str())) {
		fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(admin)); \
		return -1;
	}

	MYSQL_RES* myres = mysql_store_result(admin);
	MYSQL_ROW myrow = mysql_fetch_row(myres);

	if (myrow && myrow[0]) {
		int def_hg = std::atoi(myrow[0]);
		mysql_free_result(myres);

		return def_hg;
	} else {
		const string err_msg { "Unexpected empty result received for query: " + sel_q };
		fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, err_msg.c_str());
		return -1;
	}
}

pair<string,int> get_def_srv_host_port(MYSQL* admin, int hg) {
	const string sel_q { "SELECT hostname,port FROM mysql_servers WHERE hostgroup_id=" + std::to_string(hg) };
	int myrc = mysql_query(admin, sel_q.c_str());

	if (myrc) {
		fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(admin));
		return { "", -1 };
	} else {
		MYSQL_RES* myres = mysql_store_result(admin);
		MYSQL_ROW myrow = mysql_fetch_row(myres);

		if (myrow && myrow[0] && myrow[1]) {
			string host { myrow[0] };
			int port { std::atoi(myrow[1]) };
			mysql_free_result(myres);

			return { host, port };
		} else {
			const string err_msg { "Unexpected empty result received for query: '" + sel_q + "'"};
			fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, err_msg.c_str());
			return { "", -1 };
		}
	}
}

pair<string,int> get_def_srv_host(MYSQL* admin, const string user) {
	// Get the server from the default hostgroup
	int def_hg = get_user_def_hg(admin, user);
	if (def_hg == -1) {
		return { "", -1 };
	}

	return get_def_srv_host_port(admin, def_hg);
}

int main(int argc, char** argv) {
	CommandLine cl;

	// TODO: Harcoded for now, this is an initial version of the test.
	plan(6);

	if (cl.getEnv()) {
		diag("Failed to get the required environmental variables.");
		return EXIT_FAILURE;
	}

	MYSQL* proxy = mysql_init(NULL);
	MYSQL* mysql = mysql_init(NULL);
	MYSQL* admin = mysql_init(NULL);

	if (!mysql_real_connect(proxy, cl.host, cl.username, cl.password, NULL, cl.port, NULL, 0)) {
		fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(proxy));
		return EXIT_FAILURE;
	}

	if (!mysql_real_connect(admin, cl.host, cl.admin_username, cl.admin_password, NULL, cl.admin_port, NULL, 0)) {
		fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(admin));
		return EXIT_FAILURE;
	}

	// Create required extra users; also used in 'test_com_reset_connection_com_change_user-t.cpp'
	{
		MYSQL* mysql_server = mysql_init(NULL);

		if (!mysql_real_connect(mysql_server, cl.mysql_host, cl.mysql_username, cl.mysql_password, NULL, cl.mysql_port, NULL, 0)) {
			fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(mysql_server));
			return EXIT_FAILURE;
		}

		int err_code = create_extra_users(admin, mysql_server,
			vector<user_config> {
				std::make_tuple(
					"sbtest_reset_conn_1",
					"sbtest_reset_conn_1",
					"{\"default-transaction_isolation\":\"REPEATABLE READ\"}"
				)
			}
		);
		if (err_code) {
			diag("'create_extra_users' failed at ('%s':'%d') with error '%d'", __FILE__, __LINE__, err_code);
			return EXIT_FAILURE;
		}

		mysql_close(mysql_server);
	}

	const pair<string,int> srv_host { get_def_srv_host(admin, cl.username) };
	if (srv_host.first.empty()) {
		diag("Failed to obtain the target server hostname/port. Aborting further testing");
		goto cleanup;
	}

	{
		if (!mysql_real_connect(mysql, srv_host.first.c_str(), cl.username, cl.password, NULL, srv_host.second, NULL, 0)) {
			fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(proxy));
			goto cleanup;
		}

		int exp_mysql_srv_st = SERVER_STATUS_AUTOCOMMIT;

		ok(
			exp_mysql_srv_st == mysql->server_status,
			"MySQL init server status should match expected - exp: '%d', act:'%d'",
			exp_mysql_srv_st, mysql->server_status
		);

		mysql_query(mysql, "SET SESSION session_track_transaction_info=\"CHARACTERISTICS\"");
		mysql_query(mysql, "START TRANSACTION");

		exp_mysql_srv_st = SERVER_STATUS_AUTOCOMMIT | SERVER_STATUS_IN_TRANS | SERVER_SESSION_STATE_CHANGED;

		ok(
			exp_mysql_srv_st == mysql->server_status,
			"MySQL new server status should match expected - exp: '%d', act:'%d'",
			exp_mysql_srv_st, mysql->server_status
		);

		uint32_t exp_proxy_srv_st = SERVER_STATUS_AUTOCOMMIT;

		ok(
			exp_proxy_srv_st == proxy->server_status,
			"ProxySQL init server status should match expected - exp: '%d', act:'%d'",
			exp_proxy_srv_st, proxy->server_status
		);

		int err_code = mysql_reset_connection(proxy);
		if (err_code != EXIT_SUCCESS) {
			diag(
				"'mysql_reset_connection' failed with error: (%d,'%s') at ('%s':'%d')",
				mysql_errno(proxy), mysql_error(proxy), __FILE__, __LINE__
			);
			return EXIT_FAILURE;
		}

		ok(
			exp_proxy_srv_st == proxy->server_status,
			"ProxySQL server status after reset connection should match expected - exp: '%d', act:'%d'",
			exp_proxy_srv_st, proxy->server_status
		);

		const string username = { "sbtest_reset_conn_1" };
		const string password = { "sbtest_reset_conn_1" };
		const string attributes = { "" };
		err_code = create_proxysql_user(admin, username, password, attributes);
		if (err_code) {
			diag("'create_proxysql_user' failed at ('%s':'%d') with error '%d'", __FILE__, __LINE__, err_code);
			return EXIT_FAILURE;
		}

		MYSQL_QUERY(admin, "LOAD MYSQL USERS TO RUNTIME");

		err_code = mysql_change_user(proxy, username.c_str(), password.c_str(), NULL);
		if (err_code != EXIT_SUCCESS) {
			diag(
				"'mysql_change_user' failed with error: (%d,'%s') at ('%s':'%d')",
				mysql_errno(proxy), mysql_error(proxy), __FILE__, __LINE__
			);
			return EXIT_FAILURE;
		}

		ok(
			exp_proxy_srv_st == proxy->server_status,
			"ProxySQL server status after change user should match expected - exp: '%d', act:'%d'",
			exp_proxy_srv_st, proxy->server_status
		);

		mysql_query(proxy, "SET SESSION session_track_transaction_info=\"CHARACTERISTICS\"");
		mysql_query(proxy, "START TRANSACTION");

		exp_proxy_srv_st = SERVER_STATUS_AUTOCOMMIT | SERVER_STATUS_IN_TRANS;

		ok(
			exp_proxy_srv_st == proxy->server_status,
			"ProxySQL new server status should match expected - exp: '%d', act:'%d'",
			exp_proxy_srv_st, proxy->server_status
		);
	}

cleanup:
	mysql_close(proxy);
	mysql_close(mysql);
	mysql_close(admin);

	return exit_status();
}