def test_get_latest_image_id_returns_id_of_single_image(self):
images = [
{'ImageId': 'image1', 'CreationDate': datetime.datetime(2015, 1, 6, 15, 1, 24, 78915)}
]
result = Ec2Api.get_latest_image_id(images)
self.assertEqual('image1', result)
def test_get_latest_taupage_image_id_returns_only_image(
self, get_taupage_images_mock, _):
image_1 = Mock(spec=Image)
image_1.id = "image1"
image_1.creationDate = datetime.datetime(2015, 1, 6, 15, 1, 24, 78915)
get_taupage_images_mock.return_value = [image_1]
self.assertEqual('image1', Ec2Api().get_latest_taupage_image_id())
def test_get_latest_image_id_returns_the_most_recent_image_id(self):
images = [
{'ImageId': 'image1', 'CreationDate': datetime.datetime(2015, 1, 6, 15, 1, 24, 78915)},
{'ImageId': 'image2', 'CreationDate': datetime.datetime(2015, 1, 6, 15, 8, 24, 78915)},
{'ImageId': 'image3', 'CreationDate': datetime.datetime(2012, 11, 6, 15, 7, 24, 78915)}
]
result = Ec2Api.get_latest_image_id(images)
self.assertEqual('image2', result)
def test_get_latest_image_id_returns_id_of_single_image(self):
images = [{
'ImageId':
'image1',
'CreationDate':
datetime.datetime(2015, 1, 6, 15, 1, 24, 78915)
}]
result = Ec2Api.get_latest_image_id(images)
self.assertEqual('image1', result)
def test_get_latest_taupage_image_id_returns_id_of_eldest_image(
self, get_taupage_images_mock, _):
image_1 = Mock(spec=Image)
image_1.id = "image1"
image_1.creationDate = datetime.datetime(2015, 1, 6, 15, 1, 24, 78915)
image_2 = Mock(spec=Image)
image_2.id = "image2"
image_2.creationDate = datetime.datetime(2015, 1, 6, 15, 8, 24, 78915)
image_3 = Mock(spec=Image)
image_3.id = "image3"
image_3.creationDate = datetime.datetime(2015, 1, 6, 15, 7, 24, 78915)
get_taupage_images_mock.return_value = [image_1, image_2, image_3]
self.assertEqual('image2', Ec2Api().get_latest_taupage_image_id())
def test_get_latest_image_id_returns_the_most_recent_image_id(self):
images = [{
'ImageId':
'image1',
'CreationDate':
datetime.datetime(2015, 1, 6, 15, 1, 24, 78915)
}, {
'ImageId':
'image2',
'CreationDate':
datetime.datetime(2015, 1, 6, 15, 8, 24, 78915)
}, {
'ImageId':
'image3',
'CreationDate':
datetime.datetime(2012, 11, 6, 15, 7, 24, 78915)
}]
result = Ec2Api.get_latest_image_id(images)
self.assertEqual('image2', result)
def test_get_taupage_images_raises_exception_on_empty_response(
self, ec2_mock):
ec2_mock.return_value.get_all_images.return_value = []
with self.assertRaises(CfnSphereException):
Ec2Api().get_taupage_images()
def __init__(self, region="eu-west-1"):
self.logger = get_logger()
self.cfn = CloudFormation(region)
self.ec2 = Ec2Api(region)
self.kms = KMS(region)
def __init__(self, cfn, region=DEFAULT_REGION):
self.logger = get_logger()
self.cfn = cfn
self.ec2 = Ec2Api(region)
self.kms = KMS(region)
self.ssm = SSM(region)
def test_get_images_raises_exception_on_empty_response(self, boto_client):
boto_client.return_value.describe_images.return_value = {'Images': []}
with self.assertRaises(CfnSphereException):
print(Ec2Api().get_images("foo"))
