mirror of https://github.com/sysown/proxysql
Merge pull request #4819 from sysown/v3.0-ff_conns_eof_caps_match
Enforce `CLIENT_DEPRECATE_EOF` capability match for `fast-forward` sessionspull/4945/head v3.0.1
René Cannaò
11 months ago
committed by
GitHub
commit
2c26a42897
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,113 @@
|
||||
/**
|
||||
* @file test_sqlite3_special_queries.cpp
|
||||
* @brief Execute all the special queries intercepted by SQLite3 sessions.
|
||||
* @details The test should be compiled against 'libmariadb' and 'libmysql' to ensure compatibility in the
|
||||
* response of the intercepted queries.
|
||||
*/
|
||||
|
||||
#include <vector>
|
||||
#include <string>
|
||||
|
||||
#include "mysql.h"
|
||||
|
||||
#include "command_line.h"
|
||||
#include "utils.h"
|
||||
#include "tap.h"
|
||||
|
||||
using std::string;
|
||||
using std::vector;
|
||||
|
||||
struct test_opts_t {
|
||||
unsigned long cflags;
|
||||
enum enum_mysql_set_option set_opt;
|
||||
};
|
||||
|
||||
string to_string(const test_opts_t& opts) {
|
||||
return string { "{" }
|
||||
+ "cflags:" + std::to_string(opts.cflags) + ", "
|
||||
+ "set_opt:" + std::to_string(opts.set_opt)
|
||||
+ "}";
|
||||
}
|
||||
|
||||
test_opts_t get_opt(vector<bool> bin_vec) {
|
||||
return test_opts_t {
|
||||
bin_vec[0] ? CLIENT_DEPRECATE_EOF : 0,
|
||||
bin_vec[1] ? MYSQL_OPTION_MULTI_STATEMENTS_ON : MYSQL_OPTION_MULTI_STATEMENTS_OFF
|
||||
};
|
||||
}
|
||||
|
||||
vector<test_opts_t> gen_tests() {
|
||||
vector<test_opts_t> tests {};
|
||||
const auto opts { get_all_bin_vec(2) };
|
||||
|
||||
std::transform(opts.begin(), opts.end(), std::back_inserter(tests), get_opt);
|
||||
|
||||
return tests;
|
||||
}
|
||||
|
||||
const vector<string> set_queries {
|
||||
"SET character_set_results='latin1'",
|
||||
"SET SQL_AUTO_IS_NULL=1",
|
||||
"SET NAMES 'utf8'",
|
||||
"/*!40100 SET @@SQL_MODE='' */",
|
||||
"/*!40103 SET TIME_ZONE='UTC' */",
|
||||
"/*!80000 SET SESSION transaction_isolation = 'READ-COMMITTED' */",
|
||||
"SET SESSION transaction_isolation = 'READ-COMMITTED'",
|
||||
"SET wait_timeout=86400"
|
||||
};
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
CommandLine cl;
|
||||
|
||||
if (cl.getEnv()) {
|
||||
diag("Failed to get the required environmental variables.");
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
const vector<test_opts_t> tests { gen_tests() };
|
||||
plan(tests.size()*(3 + set_queries.size()));
|
||||
|
||||
for (const test_opts_t& opts : tests) {
|
||||
diag("Executing test test_opts=%s", to_string(opts).c_str());
|
||||
MYSQL* proxy = mysql_init(NULL);
|
||||
mysql_options(proxy, MYSQL_DEFAULT_AUTH, "mysql_native_password");
|
||||
|
||||
int cflags = opts.cflags;
|
||||
|
||||
#ifdef LIBMYSQL_HELPER8
|
||||
{
|
||||
enum mysql_ssl_mode ssl_mode = SSL_MODE_DISABLED;
|
||||
mysql_options(proxy, MYSQL_OPT_SSL_MODE, &ssl_mode);
|
||||
}
|
||||
#endif
|
||||
|
||||
if (!mysql_real_connect(proxy, cl.host, cl.username, cl.password, NULL, cl.port, NULL, cflags)) {
|
||||
fprintf(stderr, "File %s, line %d, Error: %s\n", __FILE__, __LINE__, mysql_error(proxy));
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
int set_rc = mysql_set_server_option(proxy, opts.set_opt);
|
||||
|
||||
ok(
|
||||
set_rc == 0,
|
||||
"Setting server option should succeed rc=%d test_opts=%s",
|
||||
set_rc, to_string(opts).c_str()
|
||||
);
|
||||
|
||||
int ping_rc = mysql_ping(proxy);
|
||||
ok(ping_rc == 0, "Pinging the server succeed rc=%d", ping_rc);
|
||||
|
||||
int initdb_rc = mysql_select_db(proxy, "information_schema");
|
||||
ok(initdb_rc == 0, "COM_INIT_DB should succeed rc=%d", initdb_rc);
|
||||
|
||||
for (const auto& q : set_queries) {
|
||||
diag("Executing 'special SET' query q='%s'", q.c_str());
|
||||
int rc = mysql_query(proxy, q.c_str());
|
||||
ok(rc == 0, "Query should execute without error q='%s'", q.c_str());
|
||||
}
|
||||
|
||||
mysql_close(proxy);
|
||||
}
|
||||
|
||||
return exit_status();
|
||||
}
|
||||
Loading…
Reference in new issue