def test_get_vsphere_location(self): expected = 'vsphere://ok' metadata = {'locations': [{}, {'url': 'http://ko'}, {'url': expected}]} with mock.patch.object(images.IMAGE_API, 'get', return_value=metadata): context = mock.Mock() observed = images.get_vsphere_location(context, 'image_id') self.assertEqual(expected, observed)
def test_get_vsphere_location_no_image(self): context = mock.Mock() observed = images.get_vsphere_location(context, None) self.assertIsNone(observed)
def test_get_no_vsphere_location(self): metadata = {'locations': [{}, {'url': 'http://ko'}]} with mock.patch.object(images.IMAGE_API, 'get', return_value=metadata): context = mock.Mock() observed = images.get_vsphere_location(context, 'image_id') self.assertIsNone(observed)