[pdm-devel] [PATCH proxmox-datacenter-manager 12/15] fake remote: generate fake task data

[Lukas Wagner]

This one helps us with the evaluation of the performance characteristics
of huge setups.

Signed-off-by: Lukas Wagner <l.wagner at proxmox.com>
---
 server/src/test_support/fake_remote.rs | 64 +++++++++++++++++++++++++-
 1 file changed, 62 insertions(+), 2 deletions(-)

diff --git a/server/src/test_support/fake_remote.rs b/server/src/test_support/fake_remote.rs
index 2419e69..52584a0 100644
--- a/server/src/test_support/fake_remote.rs
+++ b/server/src/test_support/fake_remote.rs
@@ -6,8 +6,9 @@ use pdm_config::remotes::RemoteConfig;
 use proxmox_product_config::ApiLockGuard;
 use proxmox_section_config::typed::SectionConfigData;
 use pve_api_types::{
-    client::PveClient, ClusterMetrics, ClusterMetricsData, ClusterResource, ClusterResourceKind,
-    ClusterResourceType, StorageContent,
+    client::PveClient, ClusterMetrics, ClusterMetricsData, ClusterNodeIndexResponse,
+    ClusterNodeIndexResponseStatus, ClusterResource, ClusterResourceKind, ClusterResourceType,
+    ListTasks, ListTasksResponse, PveUpid, StorageContent,
 };
 use serde::Deserialize;
 
@@ -351,4 +352,63 @@ impl PveClient for FakePveClient {
 
         Ok(ClusterMetrics { data })
     }
+
+    async fn list_nodes(&self) -> Result<Vec<ClusterNodeIndexResponse>, proxmox_client::Error> {
+        Ok((0..self.nr_of_nodes)
+            .into_iter()
+            .map(|i| ClusterNodeIndexResponse {
+                node: format!("pve-{i}"),
+                cpu: None,
+                level: None,
+                maxcpu: None,
+                maxmem: None,
+                mem: None,
+                ssl_fingerprint: None,
+                status: ClusterNodeIndexResponseStatus::Online,
+                uptime: None,
+            })
+            .collect())
+    }
+
+    async fn get_task_list(
+        &self,
+        node: &str,
+        params: ListTasks,
+    ) -> Result<Vec<ListTasksResponse>, proxmox_client::Error> {
+        let make_task = |starttime| {
+            let endtime = Some(starttime + 4);
+
+            let upid_str =
+                format!("UPID:{node}:0000C530:001C9BEC:{starttime:08X}:stopall::root at pam:",);
+            let upid: PveUpid = upid_str.parse().unwrap();
+
+            ListTasksResponse {
+                node: node.to_string(),
+                endtime,
+                pid: upid.pid as i64,
+                pstart: upid.pstart as i64,
+                starttime,
+                status: Some("OK".to_string()),
+                ty: upid.worker_type,
+                user: upid.auth_id,
+                upid: upid_str,
+                id: upid.worker_id.unwrap_or_default(),
+            }
+        };
+
+        let limit = params.limit.unwrap_or(1500);
+        let since = params.since.unwrap_or(0);
+
+        let now = proxmox_time::epoch_i64();
+
+        // Let's fake a new task every 5 seconds
+        let number_of_tasks = (now - since) / 5;
+
+        let number_of_tasks = limit.min(number_of_tasks as u64);
+
+        Ok((0..number_of_tasks)
+            .into_iter()
+            .map(|i| make_task(now - i as i64 * 5))
+            .collect())
+    }
 }
-- 
2.39.5