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"))