From b4d04bdff6a7e2c3da39fdb5b1906a228c38223e Mon Sep 17 00:00:00 2001
From: Rob Bradford <robert.bradford@intel.com>
Date: Tue, 28 Jan 2020 10:38:42 +0000
Subject: [PATCH] tests: Add CLI <-> API validation test for --disk changes

Check that the CLI generates the JSON data as expected.

Signed-off-by: Rob Bradford <robert.bradford@intel.com>
---
 src/main.rs | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index acdb20a08..771502460 100755
--- a/src/main.rs
+++ b/src/main.rs
@@ -681,6 +681,36 @@ mod unit_tests {
                 }"#,
                 false,
             ),
+            (
+                vec![
+                    "cloud-hypervisor",
+                    "--disk",
+                    "path=/path/to/disk/1,vhost_user=true,socket=/tmp/socket1",
+                    "path=/path/to/disk/2",
+                ],
+                r#"{
+                    "disks": [
+                        {"path": "/path/to/disk/1", "vhost_user":true, "vhost_socket":"/tmp/socket1"},
+                        {"path": "/path/to/disk/2"}
+                    ]
+                }"#,
+                true,
+            ),
+            (
+                vec![
+                    "cloud-hypervisor",
+                    "--disk",
+                    "path=/path/to/disk/1,vhost_user=true,socket=/tmp/socket1,wce=true",
+                    "path=/path/to/disk/2",
+                ],
+                r#"{
+                    "disks": [
+                        {"path": "/path/to/disk/1", "vhost_user":true, "vhost_socket":"/tmp/socket1", "wce":true},
+                        {"path": "/path/to/disk/2"}
+                    ]
+                }"#,
+                true,
+            ),
         ]
         .iter()
         .for_each(|(cli, openapi, equal)| {