Skip to content

Update example_ec2 to Create Instance Which Supports Hibernation (#35790) #35839

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
vincbeck merged 6 commits into from
Dec 5, 2023
Merged

Update example_ec2 to Create Instance Which Supports Hibernation (#35790) #35839

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
c42c673
Update example_ec2 instance to support hibernation
dashton90 Nov 24, 2023
ae55a85
Add wait_for_completion for hibernate and reboot operators in EC2 exa…
dashton90 Nov 24, 2023
00e06ce
Change max_attempts to 75 for example_ec2 DAG
dashton90 Nov 29, 2023
b6c134a
Merge branch 'main' into bugfix/ec2_example
ferruzzi Nov 30, 2023
b8e6568
Move configurations out of code block
dashton90 Dec 1, 2023
4089321
add comments to ec2 example
dashton90 Dec 4, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 17 additions & 2 deletions tests/system/providers/amazon/aws/example_ec2.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -50,11 +50,20 @@ def get_latest_ami_id():
# on how they name the new images. This page should have AL2022 info when
# it comes available: https://aws.amazon.com/linux/amazon-linux-2022/faqs/
image_prefix = "Amazon Linux*"
root_device_name = "/dev/xvda"

images = boto3.client("ec2").describe_images(
Filters=[
{"Name": "description", "Values": [image_prefix]},
{"Name": "architecture", "Values": ["arm64"]},
{
"Name": "architecture",
"Values": ["x86_64"],
}, # t3 instances are only compatible with x86 architecture
{
"Name": "root-device-type",
"Values": ["ebs"],
}, # instances which are capable of hibernation need to use an EBS-backed AMI
{"Name": "root-device-name", "Values": [root_device_name]},
],
Owners=["amazon"],
)
Expand Down Expand Up @@ -97,7 +106,7 @@ def parse_response(instance_ids: list):
image_id = get_latest_ami_id()

config = {
"InstanceType": "t4g.micro",
"InstanceType": "t3.micro",
"KeyName": key_name,
"TagSpecifications": [
{"ResourceType": "instance", "Tags": [{"Key": "Name", "Value": instance_name}]}
Expand All @@ -106,6 +115,9 @@ def parse_response(instance_ids: list):
# https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/configuring-instance-metadata-service.html
"MetadataOptions": {"HttpEndpoint": "enabled", "HttpTokens": "required"},
"HibernationOptions": {"Configured": True},
"BlockDeviceMappings": [
{"DeviceName": "/dev/xvda", "Ebs": {"Encrypted": True, "DeleteOnTermination": True}}
],
}

# EC2CreateInstanceOperator creates and starts the EC2 instances. To test the EC2StartInstanceOperator,
Expand Down Expand Up @@ -151,13 +163,16 @@ def parse_response(instance_ids: list):
instance_ids=instance_id,
)
# [END howto_operator_ec2_reboot_instance]
reboot_instance.wait_for_completion = True

# [START howto_operator_ec2_hibernate_instance]
hibernate_instance = EC2HibernateInstanceOperator(
task_id="hibernate_instace",
instance_ids=instance_id,
)
# [END howto_operator_ec2_hibernate_instance]
hibernate_instance.wait_for_completion = True
hibernate_instance.max_attempts = 75

# [START howto_operator_ec2_terminate_instance]
terminate_instance = EC2TerminateInstanceOperator(
Expand Down