61 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			61 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /**
 | |
|  * MXS-872: MaxScale doesn't understand roles
 | |
|  *
 | |
|  * https://jira.mariadb.org/browse/MXS-872
 | |
|  */
 | |
| 
 | |
| #include "testconnections.h"
 | |
| #include <vector>
 | |
| 
 | |
| using namespace std;
 | |
| 
 | |
| int main(int argc, char** argv)
 | |
| {
 | |
|     TestConnections test(argc, argv);
 | |
| 
 | |
|     test.repl->connect();
 | |
|     for (auto a : vector<string>({"DROP DATABASE IF EXISTS my_db",
 | |
|                                   "CREATE DATABASE my_db",
 | |
|                                   "DROP ROLE IF EXISTS dba",
 | |
|                                   "CREATE ROLE dba",
 | |
|                                   "GRANT SELECT ON my_db.* TO dba",
 | |
|                                   "DROP USER IF EXISTS 'test'@'%'",
 | |
|                                   "DROP USER IF EXISTS 'test2'@'%'",
 | |
|                                   "CREATE USER 'test'@'%' IDENTIFIED BY 'test'",
 | |
|                                   "CREATE USER 'test2'@'%' IDENTIFIED BY 'test2'",
 | |
|                                   "GRANT dba TO 'test'@'%'",
 | |
|                                   "GRANT dba TO 'test2'@'%'",
 | |
|                                   "SET DEFAULT ROLE dba FOR 'test'@'%'"}))
 | |
|     {
 | |
|         test.try_query(test.repl->nodes[0], "%s", a.c_str());
 | |
|     }
 | |
| 
 | |
|     // Wait for the users to replicate
 | |
|     test.repl->sync_slaves();
 | |
| 
 | |
|     test.tprintf("Connect with a user that has a default role");
 | |
|     MYSQL* conn = open_conn_db(test.maxscales->rwsplit_port[0], test.maxscales->IP[0], "my_db", "test", "test");
 | |
|     test.assert(mysql_errno(conn) == 0, "Connection failed: %s", mysql_error(conn));
 | |
|     char value[100] {};
 | |
|     find_field(conn, "SELECT CURRENT_ROLE() AS role", "role", value);
 | |
|     test.assert(strcmp(value, "dba") == 0, "Current role should be 'dba' but is: %s", value);
 | |
|     mysql_close(conn);
 | |
| 
 | |
|     test.tprintf("Connect with a user that doesn't have a default role, expect failure");
 | |
|     conn = open_conn_db(test.maxscales->rwsplit_port[0], test.maxscales->IP[0], "my_db", "test2", "test2");
 | |
|     test.assert(mysql_errno(conn) != 0, "Connection should fail");
 | |
|     mysql_close(conn);
 | |
| 
 | |
|     // Cleanup
 | |
|     for (auto a : vector<string>({"DROP DATABASE IF EXISTS my_db",
 | |
|                                   "DROP ROLE IF EXISTS dba",
 | |
|                                   "DROP USER 'test'@'%'",
 | |
|                                   "DROP USER 'test2'@'%'"}))
 | |
|     {
 | |
|         execute_query_silent(test.repl->nodes[0], "%s", a.c_str());
 | |
|     }
 | |
| 
 | |
|     test.repl->disconnect();
 | |
|     return test.global_result;
 | |
| }
 | 
