def test_extract_hardware_specific_load_balancing(self, policy, is_output_empty):
from . import CLEAR_POLICY, ROUND_ROBIN, FAIL_OVER_ONLY, LEAST_BLOCKS, LEAST_QUEUE_DEPTH
if is_output_empty:
output = "\n".join(["", "No target-level default load balance policies have been set.", ""])
else:
if policy is None:
raise unittest.SkipTest
output = "\n".join(["",
""""Target H/W Identifier " LB Policy """,
"""-------------------------------------------------------------------------------""",
""""ABC 123 " %s """ % policy,
""])
if is_output_empty:
self.assertEqual(MultipathClaim._extract_hardware_specific_load_balacing_policy(output, self.HARDWARE_ID),
CLEAR_POLICY)
else:
self.assertIn(MultipathClaim._extract_hardware_specific_load_balacing_policy(output, self.HARDWARE_ID),
[ROUND_ROBIN, FAIL_OVER_ONLY, LEAST_BLOCKS, LEAST_QUEUE_DEPTH])
