def _assert_get_lb_policy(self, policy_name, policy_args, expected_policy_type,
expected_child_policy_type=None,
should_throw=False):
thrown = False
try:
policy = CassandraHook.get_lb_policy(policy_name, policy_args)
self.assertTrue(isinstance(policy, expected_policy_type))
if expected_child_policy_type:
self.assertTrue(isinstance(policy._child_policy,
expected_child_policy_type))
except Exception:
thrown = True
self.assertEqual(should_throw, thrown)
def _assert_get_lb_policy(
self,
policy_name,
policy_args,
expected_policy_type,
expected_child_policy_type=None,
should_throw=False,
):
thrown = False
try:
policy = CassandraHook.get_lb_policy(policy_name, policy_args)
assert isinstance(policy, expected_policy_type)
if expected_child_policy_type:
assert isinstance(policy._child_policy, expected_child_policy_type)
except Exception: # pylint: disable=broad-except
thrown = True
assert should_throw == thrown
