def test_custom_assess_status_check(self):
_check = mock.MagicMock()
_check.return_value = None, None
_conn_check = mock.MagicMock()
_conn_check.return_value = True
# All is well
mrc = mysql_router.MySQLRouterCharm()
mrc.check_if_paused = _check
mrc.check_interfaces = _check
mrc.check_mandatory_config = _check
mrc.check_mysql_connection = _conn_check
self.assertEqual((None, None), mrc.custom_assess_status_check())
self.assertEqual(3, len(_check.mock_calls))
_conn_check.assert_called_once_with()
# First checks fail
_check.return_value = "blocked", "for some reason"
self.assertEqual(("blocked", "for some reason"),
mrc.custom_assess_status_check())
# MySQL connect fails
_check.return_value = None, None
_conn_check.return_value = False
self.assertEqual(("blocked", "Failed to connect to MySQL"),
mrc.custom_assess_status_check())
def test_stop_mysqlrouter(self):
_user = "******"
_port = "3306"
mrc = mysql_router.MySQLRouterCharm()
mrc.options.system_user = _user
mrc.options.base_port = _port
# Successful
mrc.stop_mysqlrouter()
self.subprocess.Popen.assert_called_once_with(
["/home/ubuntu/mysqlrouter/stop.sh"],
bufsize=1,
stdout=self.stdout,
stderr=self.stdout,
universal_newlines=True)
self.clear_flag.assert_called_once_with(
mysql_router.MYSQL_ROUTER_STARTED)
# Fail
self.subprocess.reset_mock()
self.clear_flag.reset_mock()
self.subprocess.CalledProcessError = FakeException
self.subprocess.Popen.side_effect = self.subprocess.CalledProcessError
mrc.stop_mysqlrouter()
self.clear_flag.assert_not_called()
def test_cluster_address(self):
_json_addr = '"10.10.10.50"'
_addr = "10.10.10.50"
self.endpoint_from_flag.return_value = self.db_router
self.db_router.db_host.return_value = _json_addr
mrc = mysql_router.MySQLRouterCharm()
self.assertEqual(mrc.cluster_address, _addr)
def test_db_router_password(self):
_json_pass = '******'
_pass = "******"
self.endpoint_from_flag.return_value = self.db_router
self.db_router.password.return_value = _json_pass
mrc = mysql_router.MySQLRouterCharm()
self.assertEqual(mrc.db_router_password, _pass)
def test_check_mysql_connection(self):
self.patch_object(mysql_router.mysql, "MySQL8Helper")
_helper = mock.MagicMock()
_json_pass = '******'
_pass = "******"
_user = "******"
_addr = "127.0.0.1"
self.endpoint_from_flag.return_value = self.db_router
self.db_router.password.return_value = _json_pass
self.patch_object(mysql_router.mysql.MySQLdb, "_exceptions")
self._exceptions.OperationalError = Exception
_helper = mock.MagicMock()
mrc = mysql_router.MySQLRouterCharm()
mrc.get_db_helper = mock.MagicMock()
mrc.get_db_helper.return_value = _helper
# Connects
self.assertTrue(mrc.check_mysql_connection())
_helper.connect.assert_called_once_with(_user, _pass, _addr)
# Fails
_helper.reset_mock()
_helper.connect.side_effect = self._exceptions.OperationalError
self.assertFalse(mrc.check_mysql_connection())
_helper.connect.assert_called_once_with(_user, _pass, _addr)
def test_restart_mysqlrouter(self):
mrc = mysql_router.MySQLRouterCharm()
mrc.stop_mysqlrouter = mock.MagicMock()
mrc.start_mysqlrouter = mock.MagicMock()
mrc.restart_mysqlrouter()
mrc.stop_mysqlrouter.assert_called_once()
mrc.start_mysqlrouter.assert_called_once()
def test_install(self):
self.patch_object(mysql_router.charms_openstack.charm.OpenStackCharm,
"install", "super_install")
mrc = mysql_router.MySQLRouterCharm()
mrc.configure_source = mock.MagicMock()
mrc.install()
self.super_install.assert_called_once()
mrc.configure_source.assert_called_once()
def test_proxy_db_and_user_requests_no_prefix(self):
mrc = mysql_router.MySQLRouterCharm()
mrc.proxy_db_and_user_requests(self.keystone_shared_db, self.db_router)
_calls = [
mock.call('keystone',
'keystone',
self.keystone_unit_ip,
prefix=mrc._unprefixed)
]
self.db_router.configure_proxy_db.assert_has_calls(_calls)
def test_proxy_db_and_user_requests_prefixed(self):
mrc = mysql_router.MySQLRouterCharm()
mrc.proxy_db_and_user_requests(self.nova_shared_db, self.db_router)
_calls = [
mock.call('nova', 'nova', self.nova_unit_ip, prefix="nova"),
mock.call('nova_api', 'nova', self.nova_unit_ip, prefix="novaapi"),
mock.call('nova_cell0',
'nova',
self.nova_unit_ip,
prefix="novacell0")
]
self.db_router.configure_proxy_db.assert_has_calls(_calls)
def test_get_db_helper(self):
self.patch_object(mysql_router.mysql, "MySQL8Helper")
_helper = mock.MagicMock()
_json_addr = '"10.10.10.70"'
_json_pass = '******'
self.endpoint_from_flag.return_value = self.db_router
self.db_router.db_host.return_value = _json_addr
self.db_router.password.return_value = _json_pass
mrc = mysql_router.MySQLRouterCharm()
self.MySQL8Helper.return_value = _helper
self.assertEqual(_helper, mrc.get_db_helper())
self.MySQL8Helper.assert_called_once()
def test_states_to_check(self):
self.patch_object(mysql_router.charms_openstack.charm.OpenStackCharm,
"states_to_check", "super_states")
self.super_states.return_value = {}
_required_rels = ["shared-db", "db-router"]
mrc = mysql_router.MySQLRouterCharm()
_results = mrc.states_to_check(_required_rels)
_states_to_check = [x[0] for x in _results["charm"]]
self.super_states.assert_called_once_with(_required_rels)
self.assertTrue(
mysql_router.MYSQL_ROUTER_BOOTSTRAPPED in _states_to_check)
self.assertTrue(mysql_router.MYSQL_ROUTER_STARTED in _states_to_check)
self.assertTrue(
mysql_router.DB_ROUTER_PROXY_AVAILABLE in _states_to_check)
def test_proxy_db_and_user_responses_unprefixed(self):
_json_pass = '******'
_pass = json.loads(_json_pass)
_local_unit = "kmr/5"
self.db_router.password.return_value = _json_pass
self.local_unit.return_value = _local_unit
mrc = mysql_router.MySQLRouterCharm()
self.db_router.get_prefixes.return_value = [
mrc._unprefixed, mrc.db_prefix
]
# Allowed Units unset
self.db_router.allowed_units.return_value = '""'
mrc.proxy_db_and_user_responses(self.db_router,
self.keystone_shared_db)
self.keystone_shared_db.set_db_connection_info.assert_called_once_with(
self.keystone_shared_db.relation_id,
mrc.shared_db_address,
_pass,
None,
prefix=None)
# Allowed Units set correctly
self.keystone_shared_db.set_db_connection_info.reset_mock()
self.db_router.allowed_units.return_value = json.dumps(_local_unit)
mrc.proxy_db_and_user_responses(self.db_router,
self.keystone_shared_db)
self.keystone_shared_db.set_db_connection_info.assert_called_once_with(
self.keystone_shared_db.relation_id,
mrc.shared_db_address,
_pass,
self.keystone_unit_name,
prefix=None)
# Confirm msyqlrouter credentials are not sent over the shared-db
# relation
for call in self.keystone_shared_db.set_db_connection_info.mock_calls:
self.assertNotEqual(mrc.db_prefix, call.kwargs.get("prefix"))
def test_bootstrap_mysqlrouter(self):
_json_addr = '"10.10.10.60"'
_json_pass = '******'
_pass = json.loads(_json_pass)
_addr = json.loads(_json_addr)
_user = "******"
_port = "3006"
self.endpoint_from_flag.return_value = self.db_router
self.db_router.password.return_value = _json_pass
self.db_router.db_host.return_value = _json_addr
mrc = mysql_router.MySQLRouterCharm()
mrc.options.system_user = _user
mrc.options.base_port = _port
# Successful
mrc.bootstrap_mysqlrouter()
self.subprocess.check_output.assert_called_once_with(
[
mrc.mysqlrouter_bin, "--user", _user, "--bootstrap",
"{}:{}@{}".format(mrc.db_router_user, _pass,
_addr), "--directory", mrc.mysqlrouter_dir,
"--conf-use-sockets", "--conf-base-port", _port
],
stderr=self.stdout)
self.set_flag.assert_called_once_with(
mysql_router.MYSQL_ROUTER_BOOTSTRAPPED)
# Fail
self.subprocess.reset_mock()
self.set_flag.reset_mock()
self.subprocess.CalledProcessError = FakeException
self.subprocess.check_output.side_effect = (
self.subprocess.CalledProcessError)
mrc.bootstrap_mysqlrouter()
self.set_flag.assert_not_called()
def test_db_prefix(self):
mrc = mysql_router.MySQLRouterCharm()
self.assertEqual(mrc.db_prefix, "mysqlrouter")
def test_db_router_user(self):
mrc = mysql_router.MySQLRouterCharm()
self.assertEqual(mrc.db_router_user, "mysqlrouteruser")
def test_db_router_address(self):
_addr = "10.10.10.30"
self.get_relation_ip.return_value = _addr
mrc = mysql_router.MySQLRouterCharm()
self.assertEqual(mrc.db_router_address, _addr)
def test_shared_db_address(self):
mrc = mysql_router.MySQLRouterCharm()
self.assertEqual(mrc.shared_db_address, "127.0.0.1")
def test_mysqlrouter_dir(self):
_user = "******"
mrc = mysql_router.MySQLRouterCharm()
mrc.options.system_user = _user
self.assertEqual(mrc.mysqlrouter_dir,
"/home/{}/mysqlrouter".format(_user))
def test_db_router_endpoint(self):
self.endpoint_from_flag.return_value = self.db_router
mrc = mysql_router.MySQLRouterCharm()
self.assertEqual(mrc.db_router_endpoint, self.db_router)
def test_proxy_db_and_user_responses_prefixed(self):
_json_pass = '******'
_pass = json.loads(_json_pass)
_local_unit = "nmr/5"
_nova = "nova"
_novaapi = "novaapi"
_novacell0 = "novacell0"
self.db_router.password.return_value = _json_pass
self.local_unit.return_value = _local_unit
mrc = mysql_router.MySQLRouterCharm()
self.db_router.get_prefixes.return_value = [
mrc.db_prefix, _nova, _novaapi, _novacell0
]
# Allowed Units unset
self.db_router.allowed_units.return_value = '""'
mrc.proxy_db_and_user_responses(self.db_router, self.nova_shared_db)
_calls = [
mock.call(self.nova_shared_db.relation_id,
mrc.shared_db_address,
_pass,
None,
prefix=_nova),
mock.call(self.nova_shared_db.relation_id,
mrc.shared_db_address,
_pass,
None,
prefix=_novaapi),
mock.call(self.nova_shared_db.relation_id,
mrc.shared_db_address,
_pass,
None,
prefix=_novacell0),
]
self.nova_shared_db.set_db_connection_info.assert_has_calls(_calls)
# Allowed Units set correctly
self.nova_shared_db.set_db_connection_info.reset_mock()
self.db_router.allowed_units.return_value = json.dumps(_local_unit)
mrc.proxy_db_and_user_responses(self.db_router, self.nova_shared_db)
_calls = [
mock.call(self.nova_shared_db.relation_id,
mrc.shared_db_address,
_pass,
self.nova_unit_name,
prefix=_nova),
mock.call(self.nova_shared_db.relation_id,
mrc.shared_db_address,
_pass,
self.nova_unit_name,
prefix=_novaapi),
mock.call(self.nova_shared_db.relation_id,
mrc.shared_db_address,
_pass,
self.nova_unit_name,
prefix=_novacell0),
]
self.nova_shared_db.set_db_connection_info.assert_has_calls(_calls)
# Confirm msyqlrouter credentials are not sent over the shared-db
# relation
for call in self.nova_shared_db.set_db_connection_info.mock_calls:
self.assertNotEqual(mrc.db_prefix, call.kwargs.get("prefix"))
def test_mysqlrouter_bin(self):
mrc = mysql_router.MySQLRouterCharm()
self.assertEqual(mrc.mysqlrouter_bin, "/usr/bin/mysqlrouter")
