mirror of https://github.com/sysown/proxysql
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
179 lines
6.6 KiB
179 lines
6.6 KiB
#include <cstdlib>
|
|
#include <cstdio>
|
|
#include <cstring>
|
|
#include <unistd.h>
|
|
|
|
#include <string>
|
|
#include <sstream>
|
|
#include "mysql.h"
|
|
|
|
#include "tap.h"
|
|
#include "command_line.h"
|
|
#include "utils.h"
|
|
|
|
/*
|
|
This TAP test validate the use of mysql-init_connect.
|
|
It uses 2 valid init_connect, and 2 invalid ones that trigger PMC-10003.
|
|
It also sets a value that causes a syntax error
|
|
This TAP test is a modified version of mysql-init_connect-1-t to use
|
|
mysql_hostgroup_attributes.init_connect instead of global variable
|
|
mysql-init_connect
|
|
We configure both hostgroup 0 and 1
|
|
*/
|
|
|
|
int main(int argc, char** argv) {
|
|
CommandLine cl;
|
|
|
|
if(cl.getEnv())
|
|
return exit_status();
|
|
|
|
plan(8);
|
|
|
|
MYSQL* mysqladmin = mysql_init(NULL);
|
|
if (!mysqladmin)
|
|
return exit_status();
|
|
|
|
if (!mysql_real_connect(mysqladmin, 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(mysqladmin));
|
|
return exit_status();
|
|
}
|
|
|
|
MYSQL* mysql = mysql_init(NULL);
|
|
if (!mysql)
|
|
return exit_status();
|
|
|
|
if (!mysql_real_connect(mysql, cl.host, cl.username, cl.password, NULL, cl.port, NULL, 0)) {
|
|
fprintf(stderr, "Failed to connect to database: Error: %s\n",
|
|
mysql_error(mysql));
|
|
return exit_status();
|
|
}
|
|
diag("Setting mysql_hostgroup_attributes.init_connect to DO 1");
|
|
MYSQL_QUERY(mysqladmin, "DELETE FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "INSERT INTO mysql_hostgroup_attributes(hostgroup_id, init_connect) VALUES (1,'DO 1')");
|
|
MYSQL_QUERY(mysqladmin, "INSERT INTO mysql_hostgroup_attributes(hostgroup_id, init_connect) SELECT 0, init_connect FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "LOAD MYSQL SERVERS TO RUNTIME");
|
|
|
|
MYSQL_RES *res;
|
|
|
|
{
|
|
const char *q = "SELECT /* create_new_connection=1 */ 100";
|
|
diag("Running query: %s", q);
|
|
MYSQL_QUERY(mysql, q);
|
|
res = mysql_store_result(mysql);
|
|
MYSQL_ROW row;
|
|
unsigned long long num_rows = mysql_num_rows(res);
|
|
ok(num_rows == 1, "mysql_num_rows() , expected: 1 , actual: %llu", num_rows);
|
|
while ((row = mysql_fetch_row(res))) {
|
|
ok(strcmp(row[0],"100")==0, "row: expected: \"100\" , actual: \"%s\"", row[0]);
|
|
}
|
|
mysql_free_result(res);
|
|
}
|
|
|
|
diag("Setting mysql_hostgroup_attributes.init_connect to SELECT 1");
|
|
MYSQL_QUERY(mysqladmin, "DELETE FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "INSERT INTO mysql_hostgroup_attributes(hostgroup_id, init_connect) VALUES (1,'SELECT 1')");
|
|
MYSQL_QUERY(mysqladmin, "INSERT INTO mysql_hostgroup_attributes(hostgroup_id, init_connect) SELECT 0, init_connect FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "LOAD MYSQL SERVERS TO RUNTIME");
|
|
|
|
{
|
|
const char *q = "SELECT /* create_new_connection=1 */ 200";
|
|
diag("Running query: %s", q);
|
|
int rc=mysql_query(mysql,q);
|
|
ok(rc!=0, "Query should fail. Error: %s", mysql_error(mysql));
|
|
if (rc==0)
|
|
return exit_status();
|
|
mysql_close(mysql);
|
|
}
|
|
|
|
diag("Setting mysql_hostgroup_attributes.init_connect to SELECT SLEEP(3)");
|
|
MYSQL_QUERY(mysqladmin, "DELETE FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "INSERT INTO mysql_hostgroup_attributes(hostgroup_id, init_connect) VALUES (1,'SELECT SLEEP(3)')");
|
|
MYSQL_QUERY(mysqladmin, "INSERT INTO mysql_hostgroup_attributes(hostgroup_id, init_connect) SELECT 0, init_connect FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "LOAD MYSQL SERVERS TO RUNTIME");
|
|
|
|
{
|
|
// reconnect
|
|
MYSQL* mysql = mysql_init(NULL);
|
|
if (!mysql)
|
|
return exit_status();
|
|
if (!mysql_real_connect(mysql, cl.host, cl.username, cl.password, NULL, cl.port, NULL, 0)) {
|
|
fprintf(stderr, "Failed to connect to database: Error: %s\n",
|
|
mysql_error(mysql));
|
|
return exit_status();
|
|
}
|
|
const char *q = "SELECT /* create_new_connection=1 */ 300";
|
|
diag("Running query: %s", q);
|
|
unsigned long long begin = monotonic_time();
|
|
int rc=mysql_query(mysql,q);
|
|
ok(rc!=0, "Query should fail. Error: %s", mysql_error(mysql));
|
|
if (rc==0)
|
|
return exit_status();
|
|
mysql_close(mysql);
|
|
unsigned long long end = monotonic_time();
|
|
unsigned long time_diff_ms = (end-begin)/1000;
|
|
ok(time_diff_ms>2900 && time_diff_ms < 3200 , "Total query execution time should be around 3 seconds. Actual : %lums", time_diff_ms);
|
|
}
|
|
|
|
diag("Setting mysql_hostgroup_attributes.init_connect to Syntax Error");
|
|
MYSQL_QUERY(mysqladmin, "DELETE FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "INSERT INTO mysql_hostgroup_attributes(hostgroup_id, init_connect) VALUES (1,'Syntax Error')");
|
|
MYSQL_QUERY(mysqladmin, "INSERT INTO mysql_hostgroup_attributes(hostgroup_id, init_connect) SELECT 0, init_connect FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "LOAD MYSQL SERVERS TO RUNTIME");
|
|
|
|
{
|
|
// reconnect
|
|
MYSQL* mysql = mysql_init(NULL);
|
|
if (!mysql)
|
|
return exit_status();
|
|
if (!mysql_real_connect(mysql, cl.host, cl.username, cl.password, NULL, cl.port, NULL, 0)) {
|
|
fprintf(stderr, "Failed to connect to database: Error: %s\n",
|
|
mysql_error(mysql));
|
|
return exit_status();
|
|
}
|
|
const char *q = "SELECT /* create_new_connection=1 */ 400";
|
|
diag("Running query: %s", q);
|
|
int rc=mysql_query(mysql,q);
|
|
ok(rc!=0, "Query should fail. Error: %s", mysql_error(mysql));
|
|
if (rc==0)
|
|
return exit_status();
|
|
mysql_close(mysql);
|
|
}
|
|
|
|
diag("Setting mysql_hostgroup_attributes.init_connect to DO 1");
|
|
MYSQL_QUERY(mysqladmin, "DELETE FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "INSERT INTO mysql_hostgroup_attributes(hostgroup_id, init_connect) VALUES (1,'DO 1')");
|
|
MYSQL_QUERY(mysqladmin, "INSERT INTO mysql_hostgroup_attributes(hostgroup_id, init_connect) SELECT 0, init_connect FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "LOAD MYSQL SERVERS TO RUNTIME");
|
|
|
|
{
|
|
// reconnect
|
|
MYSQL* mysql = mysql_init(NULL);
|
|
if (!mysql)
|
|
return exit_status();
|
|
if (!mysql_real_connect(mysql, cl.host, cl.username, cl.password, NULL, cl.port, NULL, 0)) {
|
|
fprintf(stderr, "Failed to connect to database: Error: %s\n",
|
|
mysql_error(mysql));
|
|
return exit_status();
|
|
}
|
|
const char *q = "SELECT /* create_new_connection=1 */ 500";
|
|
diag("Running query: %s", q);
|
|
MYSQL_QUERY(mysql, q);
|
|
res = mysql_store_result(mysql);
|
|
MYSQL_ROW row;
|
|
unsigned long long num_rows = mysql_num_rows(res);
|
|
ok(num_rows == 1, "mysql_num_rows() , expected: 1 , actual: %llu", num_rows);
|
|
while ((row = mysql_fetch_row(res))) {
|
|
ok(strcmp(row[0],"500")==0, "row: expected: \"500\" , actual: \"%s\"", row[0]);
|
|
}
|
|
mysql_free_result(res);
|
|
}
|
|
|
|
MYSQL_QUERY(mysqladmin, "DELETE FROM mysql_hostgroup_attributes");
|
|
MYSQL_QUERY(mysqladmin, "LOAD MYSQL SERVERS TO RUNTIME");
|
|
mysql_close(mysqladmin);
|
|
|
|
return exit_status();
|
|
}
|
|
|