new: Add --cloud-init flag to image upload plugin (#560)

Author: lgarber-akamai

linodecli/plugins/image-upload.py

@@ -94,6 +94,11 @@ def call(args, context):
         nargs="?",
         help="A description for this Image.  Blank if omitted.",
     )
+    parser.add_argument(
+        "--cloud-init",
+        action="store_true",
+        help="If given, the new image will be flagged as cloud-init compatible.",
+    )
     parser.add_argument(
         "file",
         metavar="FILE",
@@ -149,6 +154,9 @@ def call(args, context):
     if parsed.description:
         call_args += ["--description", parsed.description]
 
+    if parsed.cloud_init:
+        call_args += ["--cloud_init", "true"]
+
     status, resp = context.client.call_operation("images", "upload", call_args)
 
     if status != 200:

tests/integration/image/test_plugin_image_upload.py

@@ -10,7 +10,7 @@
 
 from tests.integration.helpers import get_random_text
 
-REGION = "us-southeast"
+REGION = "us-iad"
 BASE_CMD = ["linode-cli", "image-upload", "--region", REGION]
 
 # A minimal gzipped image that will be accepted by the API
@@ -96,3 +96,42 @@ def test_file_upload(
     # Delete the image
     process = exec_test_command(["linode-cli", "images", "rm", image[0]["id"]])
     assert process.returncode == 0
+
+
+@pytest.mark.skipif(platform == "win32", reason="Test N/A on Windows")
+def test_file_upload_cloud_init(
+    fake_image_file,
+):
+    file_path = fake_image_file
+    label = f"cli-test-{get_random_text()}"
+    description = "test description"
+
+    # Upload the test image
+    process = exec_test_command(
+        BASE_CMD
+        + [
+            "--label",
+            label,
+            "--description",
+            description,
+            "--cloud-init",
+            file_path,
+        ]
+    )
+
+    assert process.returncode == 0
+
+    # Get the new image from the API
+    process = exec_test_command(
+        ["linode-cli", "images", "ls", "--json", "--label", label]
+    )
+    assert process.returncode == 0
+
+    image = json.loads(process.stdout.decode())
+
+    assert image[0]["label"] == label
+    assert "cloud-init" in image[0]["capabilities"]
+
+    # Delete the image
+    process = exec_test_command(["linode-cli", "images", "rm", image[0]["id"]])
+    assert process.returncode == 0