class TestImage(unittest.TestCase):
@classmethod
def setUpClass(self):
self.resource = resource()
self.client = ProfitBricksService(
username=configuration.USERNAME,
password=configuration.PASSWORD,
headers=configuration.HEADERS)
# Find an Ubuntu image for testing.
for item in self.client.list_images()['items']:
if 'Ubuntu-15' in item['properties']['name'] and item['properties']['location'] == configuration.LOCATION:
self.image = item
def test_list_images(self):
images = self.client.list_images()
self.assertGreater(len(images), 0)
self.assertEqual(images['items'][0]['type'], 'image')
assertRegex(self, images['items'][0]['id'], self.resource['uuid_match'])
def test_get_image(self):
image = self.client.get_image(self.image['id'])
self.assertEqual(image['type'], 'image')
self.assertEqual(image['id'], self.image['id'])
self.assertIn(image['properties']['imageType'], ['HDD', 'CDROM'])
class TestImage(unittest.TestCase):
def setUp(self):
self.image = ProfitBricksService(
username='******', password='******')
def test_list_images(self):
images = self.image.list_images()
self.assertEqual(len(images), 4)
self.assertEqual(
images['items'][0]['id'], '7df81087-5835-41c6-a10b-3e098593bbd2')
def test_get_image(self):
image_id = '7df81087-5835-41c6-a10b-3e098593bbd2'
image = self.image.get_image(image_id)
self.assertEqual(image['properties']['name'], 'Ubuntu 14.04')
def test_delete_image(self):
image_id = '7df81087-5835-41c6-a10b-3e098593bbd2'
image = self.image.delete_image(image_id)
self.assertTrue(image)
def test_update_image(self):
image_id = '7df81087-5835-41c6-a10b-3e098593bbd2'
image = self.image.update_image(
image_id,
name='New name')
self.assertEqual(image['properties']['name'], 'New name')
