def _update_pbm_location(self): if CONF.vmware.pbm_wsdl_location: pbm_wsdl_loc = CONF.vmware.pbm_wsdl_location else: version = vim_util.get_vc_version(self._session) pbm_wsdl_loc = pbm.get_pbm_wsdl_location(version) self._session.pbm_wsdl_loc_set(pbm_wsdl_loc)
def test_get_pbm_wsdl_location(self): wsdl = pbm.get_pbm_wsdl_location(None) self.assertIsNone(wsdl) def expected_wsdl(version): driver_abs_dir = os.path.abspath(os.path.dirname(pbm.__file__)) path = os.path.join(driver_abs_dir, 'wsdl', version, 'pbmService.wsdl') return urlparse.urljoin('file:', urllib.pathname2url(path)) with mock.patch('os.path.exists') as path_exists: path_exists.return_value = True wsdl = pbm.get_pbm_wsdl_location('5') self.assertEqual(expected_wsdl('5'), wsdl) wsdl = pbm.get_pbm_wsdl_location('5.5') self.assertEqual(expected_wsdl('5.5'), wsdl) wsdl = pbm.get_pbm_wsdl_location('5.5.1') self.assertEqual(expected_wsdl('5.5'), wsdl) path_exists.return_value = False wsdl = pbm.get_pbm_wsdl_location('5.5') self.assertIsNone(wsdl)