TC-3071 add upload advisory and delete advisory scale test

Author: bxf12315

scenarios/full-20250604.json5

@@ -9,5 +9,6 @@
   "sbom_license_ids": "urn:uuid:01973124-d4ab-7163-b104-331632a21144",
   "analyze_purl": "pkg:rpm/redhat/eap7-activemq-artemis-native@1.0.2-3.redhat_00004.1.el7eap?arch=noarch&epoch=1",
   "get_purl_details": "b00df2ca-df21-5c10-8874-304e9c54e2bd",
-  "get_recommendations": ["pkg:rpm/redhat/eap7-activemq-artemis-native@1.0.2, pkg:maven/io.quarkus/quarkus-smallrye-metrics@2.13.8"]
+  "get_recommendations": ["pkg:rpm/redhat/eap7-activemq-artemis-native@1.0.2, pkg:maven/io.quarkus/quarkus-smallrye-metrics@2.13.8"],
+  "upload_advisory_file": "./test-data/advisories/rhba-2024_11505.json"
 }

src/main.rs

@@ -150,13 +150,30 @@ async fn main() -> Result<(), anyhow::Error> {
                 "RestAPIUserSlow",
                 wait_time_from,
                 wait_time_to,
-                custom_client,
+                custom_client.clone(),
             )?
             .set_weight(1)?
             .register_transaction(tx!(search_licenses))
             .register_transaction(tx!(search_sboms_by_license))
             .register_transaction(tx!(search_purls_by_license))
         })
+        .register_scenario({
+            let mut s = create_scenario(
+                "RestAPIUploadAndDeleteFiles",
+                wait_time_from,
+                wait_time_to,
+                custom_client.clone(),
+            )?
+            .set_weight(1)?;
+
+            if let Some(file) = &scenario.upload_advisory_file {
+                s = s.register_transaction(tx!(upload_advisories_and_delete_advisories(
+                    file.clone()
+                )));
+            }
+
+            s
+        })
         // .register_scenario(
         //     scenario!("GraphQLUser")
         //         // .set_weight(1)?

src/restapi.rs

@@ -1,9 +1,27 @@
-use goose::goose::{GooseUser, TransactionResult};
+use goose::goose::{GooseUser, TransactionError, TransactionResult};
 use serde_json::json;
 use urlencoding::encode;
 
 use crate::utils::DisplayVec;
 
+pub async fn upload_advisories_and_delete_advisories(
+    advisory_file: String,
+    user: &mut GooseUser,
+) -> TransactionResult {
+    let file_bytes = tokio::fs::read(advisory_file.clone()).await.map_err(|e| {
+        TransactionError::Custom(format!("Failed to read file {}: {:?}", advisory_file, e))
+    })?;
+
+    let response = user.post("/api/v2/advisory", file_bytes).await?;
+    let v = response.response?.json::<serde_json::Value>().await?;
+
+    let uri = format!("/api/v2/advisory/{}", v["id"].as_str().unwrap_or_default());
+
+    let _response = user.delete(&uri).await?;
+
+    Ok(())
+}
+
 pub async fn list_advisory(user: &mut GooseUser) -> TransactionResult {
     let _response = user.get("/api/v2/advisory").await?;
 

src/scenario/mod.rs

@@ -93,6 +93,9 @@ pub(crate) struct Scenario {
     pub get_purl_details: Option<String>,
 
     pub get_recommendations: Option<DisplayVec<String>>,
+
+    #[serde(with = "required")]
+    pub upload_advisory_file: Option<String>,
 }
 
 impl Scenario {
@@ -125,6 +128,9 @@ impl Scenario {
         let get_purl_details = Some(loader.purl_details().await?);
         let recommendations_purl = Some(loader.purl_with_recommendations().await?);
 
+        // Load advisory files from UPLOAD_FILE_PATH directory
+        let advisory_files = loader.load_advisory_files();
+
         Ok(Self {
             get_sbom: large_sbom_digest.clone(),
             get_sbom_advisories: large_sbom_digest.clone(),
@@ -138,6 +144,7 @@ impl Scenario {
             analyze_purl,
             get_purl_details,
             get_recommendations: recommendations_purl,
+            upload_advisory_file: advisory_files.first().cloned(),
         })
     }
 }
@@ -311,6 +318,29 @@ LIMIT 10;
             Ok(DisplayVec(result))
         })
     }
+
+    /// Load advisory files from UPLOAD_FILE_PATH directory
+    fn load_advisory_files(&self) -> Vec<String> {
+        std::env::var("UPLOAD_FILE_PATH")
+            .ok()
+            .and_then(|upload_file_path| {
+                std::fs::read_dir(format!("{}/advisories", &upload_file_path))
+                    .ok()
+                    .map(|entries| {
+                        entries
+                            .filter_map(|entry| {
+                                let path = entry.ok()?.path();
+                                if path.extension()? == "json" {
+                                    Some(path.to_string_lossy().to_string())
+                                } else {
+                                    None
+                                }
+                            })
+                            .collect::<Vec<String>>()
+                    })
+            })
+            .unwrap_or_default()
+    }
 }
 
 #[cfg(test)]