From 48a05c4727b358d41b18982867e8e9a8de335546 Mon Sep 17 00:00:00 2001
From: Rob Bradford <robert.bradford@intel.com>
Date: Wed, 24 Jun 2020 14:47:14 +0100
Subject: [PATCH] tests: Add simple counters integration test

Add a simple test to check that the data from the counters matches what
is expected and that the value of the counters increases after an
operation that will hit all counters.

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

diff --git a/tests/integration.rs b/tests/integration.rs
index 5f6615ea0..985ce8f26 100644
--- a/tests/integration.rs
+++ b/tests/integration.rs
@@ -4684,4 +4684,85 @@ mod tests {
             Ok(())
         });
     }
+
+    #[derive(PartialEq, PartialOrd)]
+    struct Counters {
+        rx_bytes: u64,
+        rx_frames: u64,
+        tx_bytes: u64,
+        tx_frames: u64,
+        read_bytes: u64,
+        write_bytes: u64,
+        read_ops: u64,
+        write_ops: u64,
+    }
+
+    fn get_counters(api_socket: &str) -> Counters {
+        // Get counters
+        let (cmd_success, cmd_output) = remote_command_w_output(&api_socket, "counters", None);
+        assert!(cmd_success);
+
+        let counters: HashMap<&str, HashMap<&str, u64>> =
+            serde_json::from_slice(&cmd_output).unwrap_or_default();
+
+        let rx_bytes = *counters.get("_net2").unwrap().get("rx_bytes").unwrap();
+        let rx_frames = *counters.get("_net2").unwrap().get("rx_frames").unwrap();
+        let tx_bytes = *counters.get("_net2").unwrap().get("tx_bytes").unwrap();
+        let tx_frames = *counters.get("_net2").unwrap().get("tx_frames").unwrap();
+
+        let read_bytes = *counters.get("_disk0").unwrap().get("read_bytes").unwrap();
+        let write_bytes = *counters.get("_disk0").unwrap().get("write_bytes").unwrap();
+        let read_ops = *counters.get("_disk0").unwrap().get("read_ops").unwrap();
+        let write_ops = *counters.get("_disk0").unwrap().get("write_ops").unwrap();
+
+        Counters {
+            rx_bytes,
+            rx_frames,
+            tx_bytes,
+            tx_frames,
+            read_bytes,
+            read_ops,
+            write_bytes,
+            write_ops,
+        }
+    }
+
+    #[cfg_attr(not(feature = "mmio"), test)]
+    fn test_counters() {
+        test_block!(tb, "", {
+            let mut clear = ClearDiskConfig::new();
+            let guest = Guest::new(&mut clear);
+            let api_socket = temp_api_path(&guest.tmp_dir);
+
+            let mut child = GuestCommand::new(&guest)
+                .args(&["--cpus", "boot=1"])
+                .args(&["--memory", "size=512M"])
+                .args(&["--kernel", guest.fw_path.as_str()])
+                .default_disks()
+                .args(&["--net", guest.default_net_string().as_str()])
+                .args(&["--api-socket", &api_socket])
+                .spawn()
+                .unwrap();
+
+            thread::sleep(std::time::Duration::new(20, 0));
+
+            let orig_counters = get_counters(&api_socket);
+
+            aver!(
+                tb,
+                guest
+                    .ssh_command("dd if=/dev/zero of=test count=8 bs=1M")
+                    .is_ok()
+            );
+
+            let new_counters = get_counters(&api_socket);
+
+            // Check that all the counters have increased
+            aver!(tb, new_counters > orig_counters);
+
+            let _ = child.kill();
+            let _ = child.wait();
+            Ok(())
+        });
+    }
 }