test: Add API hinting endpoints

Adding new resources to the http api to enable testing of the hinting
functionality.

Signed-off-by: Jack Thomson <jackabt@amazon.com>

Author: JackThomson2

src/firecracker/src/api_server/parsed_request.rs

@@ -328,6 +328,7 @@ pub mod tests {
     use micro_http::HttpConnection;
     use vmm::builder::StartMicrovmError;
     use vmm::cpu_config::templates::test_utils::build_test_template;
+    use vmm::devices::virtio::balloon::device::HintingStatus;
     use vmm::resources::VmmConfig;
     use vmm::rpc_interface::VmmActionError;
     use vmm::vmm_config::balloon::{BalloonDeviceConfig, BalloonStats};
@@ -605,6 +606,9 @@ pub mod tests {
             swap_out: Some(1),
             ..Default::default()
         }));
+        verify_ok_response_with(VmmData::HintingStatus(HintingStatus {
+            ..Default::default()
+        }));
         verify_ok_response_with(VmmData::Empty);
         verify_ok_response_with(VmmData::FullVmConfig(VmmConfig::default()));
         verify_ok_response_with(VmmData::MachineConfiguration(MachineConfig::default()));

src/firecracker/src/api_server/request/balloon.rs

@@ -115,6 +115,7 @@ mod tests {
         parse_get_balloon(["statistics"].into_iter()).unwrap();
 
         parse_get_balloon(["hinting", "status"].into_iter()).unwrap();
+        parse_get_balloon(["hinting", "unrelated"].into_iter()).unwrap_err();
         parse_get_balloon(["hinting"].into_iter()).unwrap_err();
     }
 

src/vmm/src/rpc_interface.rs

@@ -1175,6 +1175,11 @@ mod tests {
         check_unsupported(preboot_request(VmmAction::UpdateBalloon(
             BalloonUpdateConfig { amount_mib: 0 },
         )));
+        check_unsupported(preboot_request(VmmAction::StartFreePageHinting(
+            Default::default(),
+        )));
+        check_unsupported(preboot_request(VmmAction::GetFreePageHintingStatus));
+        check_unsupported(preboot_request(VmmAction::StopFreePageHinting));
         check_unsupported(preboot_request(VmmAction::UpdateBalloonStatistics(
             BalloonUpdateStatsConfig {
                 stats_polling_interval_s: 0,

tests/framework/http_api.py

@@ -127,6 +127,9 @@ def __init__(self, api_usocket_full_name, *, on_error=None):
         self.mmds_config = Resource(self, "/mmds/config")
         self.balloon = Resource(self, "/balloon")
         self.balloon_stats = Resource(self, "/balloon/statistics")
+        self.balloon_hinting_start = Resource(self, "/balloon/hinting/start")
+        self.balloon_hinting_status = Resource(self, "/balloon/hinting/status")
+        self.balloon_hinting_stop = Resource(self, "/balloon/hinting/stop")
         self.vsock = Resource(self, "/vsock")
         self.snapshot_create = Resource(self, "/snapshot/create")
         self.snapshot_load = Resource(self, "/snapshot/load")