def test_extract_load_balancing(self, policy): from . import CLEAR_POLICY, ROUND_ROBIN, FAIL_OVER_ONLY, LEAST_BLOCKS, LEAST_QUEUE_DEPTH if policy: output = "\n".join(["", "MSDSM-wide Load Balance Policy: %s" % policy , ""]) else: output = "\n".join(["", "No MSDSM-wide default load balance policy has been set", ""]) self.assertIn(MultipathClaim._extract_load_balancing_from_output(output), [CLEAR_POLICY, ROUND_ROBIN, FAIL_OVER_ONLY, LEAST_BLOCKS, LEAST_QUEUE_DEPTH])