From c116c57eb80c1f2b03b5da31d1b6e83c1d9d79b2 Mon Sep 17 00:00:00 2001
From: Nuwan Goonasekera <nuwan.ag@gmail.com>
Date: Fri, 15 Sep 2017 21:30:32 +0530
Subject: [PATCH] Ensure root device is listed in block device mappings

---
 moto/ec2/responses/amis.py  | 2 +-
 tests/test_ec2/test_amis.py | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/moto/ec2/responses/amis.py b/moto/ec2/responses/amis.py
index 74767aa6b..d9be77812 100755
--- a/moto/ec2/responses/amis.py
+++ b/moto/ec2/responses/amis.py
@@ -115,7 +115,7 @@ DESCRIBE_IMAGES_RESPONSE = """<DescribeImagesResponse xmlns="http://ec2.amazonaw
           {% endif %}
           <description>{{ image.description }}</description>
           <rootDeviceType>ebs</rootDeviceType>
-          <rootDeviceName>/dev/sda</rootDeviceName>
+          <rootDeviceName>/dev/sda1</rootDeviceName>
           <blockDeviceMapping>
             <item>
               <deviceName>/dev/sda1</deviceName>
diff --git a/tests/test_ec2/test_amis.py b/tests/test_ec2/test_amis.py
index ed251f527..cf9f73f0e 100755
--- a/tests/test_ec2/test_amis.py
+++ b/tests/test_ec2/test_amis.py
@@ -57,6 +57,10 @@ def test_ami_create_and_delete():
         "Auto-created snapshot for AMI {0}".format(image.id))
     snapshot.volume_id.should.equal(volume.id)
 
+    # root device should be in AMI's block device mappings
+    root_mapping = image.block_device_mapping.get(image.root_device_name)
+    root_mapping.should_not.be.none
+
     # Deregister
     with assert_raises(EC2ResponseError) as ex:
         success = conn.deregister_image(image_id, dry_run=True)