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

@@ -341,6 +341,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};
@@ -621,6 +622,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

@@ -1219,6 +1219,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

@@ -173,6 +173,9 @@ def __init__(self, api_usocket_full_name, *, validate=True, 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")