65 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			65 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/**
 | 
						|
 * Dbfwfilter prepared statement test
 | 
						|
 *
 | 
						|
 * Checks that both text protocol and binary protocol prepared statements are
 | 
						|
 * properly handled.
 | 
						|
 */
 | 
						|
 | 
						|
#include <maxtest/testconnections.hh>
 | 
						|
#include <maxtest/fw_copy_rules.hh>
 | 
						|
 | 
						|
const char* rules = "rule test1 deny columns c on_queries select\n"
 | 
						|
                    "users %@% match any rules test1\n";
 | 
						|
 | 
						|
int main(int argc, char** argv)
 | 
						|
{
 | 
						|
    FILE* file = fopen("rules.txt", "w");
 | 
						|
    fwrite(rules, 1, strlen(rules), file);
 | 
						|
    fclose(file);
 | 
						|
 | 
						|
    TestConnections::skip_maxscale_start(true);
 | 
						|
    TestConnections test(argc, argv);
 | 
						|
 | 
						|
    copy_rules(&test, (char*) "rules.txt", (char*) ".");
 | 
						|
 | 
						|
    test.add_result(test.maxscales->restart_maxscale(0), "Restarting MaxScale failed");
 | 
						|
 | 
						|
    test.maxscales->connect_maxscale(0);
 | 
						|
    execute_query_silent(test.maxscales->conn_rwsplit[0], "DROP TABLE test.t1");
 | 
						|
 | 
						|
    test.try_query(test.maxscales->conn_rwsplit[0], "CREATE TABLE test.t1(a INT, b INT, c INT)");
 | 
						|
    test.try_query(test.maxscales->conn_rwsplit[0], "INSERT INTO test.t1 VALUES (1, 1, 1)");
 | 
						|
 | 
						|
    test.add_result(execute_query(test.maxscales->conn_rwsplit[0],
 | 
						|
                                  "PREPARE my_ps FROM 'SELECT a, b FROM test.t1'"),
 | 
						|
                    "Text protocol preparation should succeed");
 | 
						|
    test.add_result(execute_query(test.maxscales->conn_rwsplit[0], "EXECUTE my_ps"),
 | 
						|
                    "Text protocol execution should succeed");
 | 
						|
 | 
						|
    test.add_result(execute_query(test.maxscales->conn_rwsplit[0],
 | 
						|
                                  "PREPARE my_ps2 FROM 'SELECT c FROM test.t1'") == 0,
 | 
						|
                    "Text protocol preparation should fail");
 | 
						|
    test.add_result(execute_query(test.maxscales->conn_rwsplit[0], "EXECUTE my_ps2") == 0,
 | 
						|
                    "Text protocol execution should fail");
 | 
						|
 | 
						|
    MYSQL_STMT* stmt = mysql_stmt_init(test.maxscales->conn_rwsplit[0]);
 | 
						|
    const char* query = "SELECT a, b FROM test.t1";
 | 
						|
 | 
						|
    test.add_result(mysql_stmt_prepare(stmt, query, strlen(query)),
 | 
						|
                    "Binary protocol preparation should succeed");
 | 
						|
    test.add_result(mysql_stmt_execute(stmt), "Binary protocol execution should succeed");
 | 
						|
    mysql_stmt_close(stmt);
 | 
						|
 | 
						|
    stmt = mysql_stmt_init(test.maxscales->conn_rwsplit[0]);
 | 
						|
    query = "SELECT c FROM test.t1";
 | 
						|
 | 
						|
    test.add_result(!mysql_stmt_prepare(stmt, query, strlen(query)),
 | 
						|
                    "Binary protocol preparation should fail");
 | 
						|
    mysql_stmt_close(stmt);
 | 
						|
 | 
						|
    test.repl->connect();
 | 
						|
    test.try_query(test.repl->nodes[0], "DROP TABLE test.t1");
 | 
						|
 | 
						|
    return test.global_result;
 | 
						|
}
 |