73 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			73 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /**
 | |
|  * @file user_cache.cpp Test user caching mechanism of MaxScale
 | |
|  *
 | |
|  * - Create 'testuser'@'%' user
 | |
|  * - Start up MaxScale with 'testuser' as the service user
 | |
|  * - Delete 'testuser'@'%'
 | |
|  * - Restart MaxScale
 | |
|  * - Check that queries through MaxScale are OK
 | |
|  */
 | |
| 
 | |
| 
 | |
| #include <iostream>
 | |
| #include "testconnections.h"
 | |
| 
 | |
| int main(int argc, char *argv[])
 | |
| {
 | |
|     TestConnections * Test = new TestConnections(argc, argv);
 | |
|     Test->stop_timeout();
 | |
|     Test->maxscales->stop_maxscale(0);
 | |
| 
 | |
|     /** Create the test user and give required grants */
 | |
|     Test->tprintf("Creating 'testuser'@'%'\n");
 | |
|     Test->repl->connect();
 | |
|     execute_query_silent(Test->repl->nodes[0], "CREATE USER 'testuser'@'%' IDENTIFIED BY 'testpasswd'");
 | |
|     execute_query_silent(Test->repl->nodes[0], "GRANT SELECT ON mysql.user TO 'testuser'@'%'");
 | |
|     execute_query_silent(Test->repl->nodes[0], "GRANT SELECT ON mysql.db TO 'testuser'@'%'");
 | |
|     execute_query_silent(Test->repl->nodes[0], "GRANT SELECT ON mysql.tables_priv TO 'testuser'@'%'");
 | |
|     execute_query_silent(Test->repl->nodes[0], "GRANT SHOW DATABASES ON *.* TO 'testuser'@'%'");
 | |
| 
 | |
|     /** Wait for the user to replicate */
 | |
|     Test->tprintf("Waiting for users to replicate\n");
 | |
|     sleep(10);
 | |
| 
 | |
|     /** Test that MaxScale works and initialize the cache */
 | |
|     Test->tprintf("Test that MaxScale works and initialize the cache\n");
 | |
|     Test->maxscales->start_maxscale(0);
 | |
|     Test->maxscales->connect_maxscale(0);
 | |
|     Test->set_timeout(30);
 | |
|     Test->add_result(Test->try_query_all("SHOW DATABASES"), "Initial query without user cache should work\n");
 | |
|     Test->stop_timeout();
 | |
| 
 | |
|     /** Block all nodes */
 | |
|     Test->tprintf("Blocking all nodes\n");
 | |
|     for (int i = 0; i < Test->repl->N; i++)
 | |
|     {
 | |
|         Test->repl->block_node(i);
 | |
|     }
 | |
| 
 | |
|     /** Restart MaxScale and check that the user cache works */
 | |
|     Test->tprintf("Restarting MaxScale\n");
 | |
|     Test->maxscales->restart_maxscale(0);
 | |
|     sleep(5);
 | |
| 
 | |
|     Test->tprintf("Unblocking all nodes\n");
 | |
|     Test->repl->unblock_all_nodes();
 | |
|     sleep(5);
 | |
| 
 | |
|     Test->tprintf("Dropping 'testuser'@'%'\n");
 | |
|     execute_query_silent(Test->repl->nodes[0], "DROP USER 'testuser'@'%'");
 | |
|     sleep(5);
 | |
| 
 | |
|     Test->tprintf("Checking that the user cache works and queries are accepted\n");
 | |
|     Test->set_timeout(30);
 | |
|     Test->maxscales->connect_maxscale(0);
 | |
|     Test->add_result(Test->try_query_all("SHOW DATABASES"), "Second query with user cache should work\n");
 | |
|     Test->stop_timeout();
 | |
| 
 | |
|     int rval = Test->global_result;
 | |
|     delete Test;
 | |
| 
 | |
|     return rval;
 | |
| }
 | 
