def testGetMountVmIds(self):
pd = PersistentDisk()
pd.endpoint = "https://example.com:8445/pwd"
pd._raiseOnErrors = pd._setPDiskUserCredentials = pd._initPDiskConnection = Mock()
pd._getJson = Mock(return_value=("", '[{"vmId":"1234", "device":"static"}]'))
assert ["1234"] == pd.getMountVmIds("foo")
pd._getJson.assert_called_once_with(*("https://example.com:8445/pwd/disks/foo/mounts",), **{})
pd._getJson = Mock(
return_value=("", '[{"vmId":"1234", "device":"static"}, {"vmId":"4567", "device":"static"}]')
)
assert ["1234", "4567"] == pd.getMountVmIds("bar")
pd._getJson.assert_called_with(*("https://example.com:8445/pwd/disks/bar/mounts",), **{})