class TestMySQLHelper(unittest.TestCase):
def setUp(self):
mysql_host = os.getenv('MYSQL_TEST_IP')
mysql_user = os.getenv('MYSQL_TEST_USER')
mysql_password = os.getenv('MYSQL_TEST_PASSWORD')
mysql_port = os.getenv('MYSQL_TEST_PORT')
if not mysql_host or not mysql_user or not mysql_password or not mysql_port:
self.fail(msg='MySQL database connection information not set')
self.mysql_conn = MySQLHelper(host=mysql_host, port=mysql_port, user=mysql_user, password=mysql_password)
if not self.mysql_conn.connect():
self.fail(msg='Could not connect to the MySQL database')
self.mysql_version = os.getenv('MYSQL_TEST_VERSION')
def tearDown(self):
self.mysql_conn.disconnect()
def test_is_read_only_query(self):
self.assertEqual(MySQLHelper.is_read_only_query('SELECT 1'), True)
self.assertEqual(MySQLHelper.is_read_only_query('INSERT INTO foo VALUES ("bar")'), False)
def test_get_version(self):
self.assertEqual(self.mysql_conn.get_version(), self.mysql_version)
def test_get_connection_id(self):
self.assertNotEqual(self.mysql_conn.get_connection_id(), None)
def test_get_current_user(self):
self.assertNotEqual(self.mysql_conn.get_current_user(), None)
def test_set_read_only(self):
self.mysql_conn.set_read_only()
self.assertEqual(self.mysql_conn.is_read_only(), True)
def test_unset_read_only(self):
self.mysql_conn.unset_read_only()
self.assertEqual(self.mysql_conn.is_read_only(), False)
def test_get_processlist(self):
processlist = self.mysql_conn.get_processlist()
self.assertTrue(len(processlist) >= 1)
def test_kill_connection(self):
mysql_host = os.getenv('MYSQL_TEST_IP')
mysql_user = os.getenv('MYSQL_TEST_USER')
mysql_password = os.getenv('MYSQL_TEST_PASSWORD')
mysql_port = int(os.getenv('MYSQL_TEST_PORT'))
mysql_conn = MySQLHelper(host=mysql_host, port=mysql_port, user=mysql_user, password=mysql_password)
if not mysql_conn.connect():
self.fail(msg='Could not connect to the MySQL database')
mysql_conn_id = mysql_conn.get_connection_id()
self.assertTrue(self.mysql_conn.kill_connection(mysql_conn_id))
del mysql_conn
def tearDown(self):
# We remove the VIP just to have a clean slate at the end of the test
VIPMetalHelper(self.orig_master_host, self.orig_master_ip, self.orig_master_ssh_user,
self.orig_master_ssh_port).remove_vip()
VIPMetalHelper(self.new_master_host, self.new_master_ip, self.new_master_ssh_user,
self.new_master_ssh_port).remove_vip()
# We unset read only on original master to have a clean slate at the end of the test
orig_mysql = MySQLHelper(self.orig_master_ip, self.orig_master_port, self.orig_master_user,
self.orig_master_password)
orig_mysql.connect()
orig_mysql.unset_read_only()
# We set read only on new master to have a clean slate at the end of the test
new_mysql = MySQLHelper(self.new_master_ip, self.new_master_port, self.new_master_user,
self.new_master_password)
new_mysql.connect()
new_mysql.set_read_only()
class TestMySQLHelper(unittest.TestCase):
def setUp(self):
mysql_host = os.getenv('MYSQL_TEST_IP')
mysql_user = os.getenv('MYSQL_TEST_USER')
mysql_password = os.getenv('MYSQL_TEST_PASSWORD')
mysql_port = os.getenv('MYSQL_TEST_PORT')
if not mysql_host or not mysql_user or not mysql_password or not mysql_port:
self.fail(msg='MySQL database connection information not set')
self.mysql_conn = MySQLHelper(host=mysql_host,
port=mysql_port,
user=mysql_user,
password=mysql_password)
if not self.mysql_conn.connect():
self.fail(msg='Could not connect to the MySQL database')
self.mysql_version = os.getenv('MYSQL_TEST_VERSION')
def tearDown(self):
self.mysql_conn.disconnect()
def test_is_read_only_query(self):
self.assertEqual(MySQLHelper.is_read_only_query('SELECT 1'), True)
self.assertEqual(
MySQLHelper.is_read_only_query('INSERT INTO foo VALUES ("bar")'),
False)
def test_get_version(self):
self.assertEqual(self.mysql_conn.get_version(), self.mysql_version)
def test_get_connection_id(self):
self.assertNotEqual(self.mysql_conn.get_connection_id(), None)
def test_get_current_user(self):
self.assertNotEqual(self.mysql_conn.get_current_user(), None)
def test_set_read_only(self):
self.mysql_conn.set_read_only()
self.assertEqual(self.mysql_conn.is_read_only(), True)
def test_unset_read_only(self):
self.mysql_conn.unset_read_only()
self.assertEqual(self.mysql_conn.is_read_only(), False)
def test_get_processlist(self):
processlist = self.mysql_conn.get_processlist()
self.assertTrue(len(processlist) >= 1)
def test_kill_connection(self):
mysql_host = os.getenv('MYSQL_TEST_IP')
mysql_user = os.getenv('MYSQL_TEST_USER')
mysql_password = os.getenv('MYSQL_TEST_PASSWORD')
mysql_port = int(os.getenv('MYSQL_TEST_PORT'))
mysql_conn = MySQLHelper(host=mysql_host,
port=mysql_port,
user=mysql_user,
password=mysql_password)
if not mysql_conn.connect():
self.fail(msg='Could not connect to the MySQL database')
mysql_conn_id = mysql_conn.get_connection_id()
self.assertTrue(self.mysql_conn.kill_connection(mysql_conn_id))
del mysql_conn