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