TC-3071 scale test for put and patch advisory lables

Author: bxf12315

scenarios/full-20250604.json5

@@ -111,5 +111,7 @@
     "urn:uuid:01973122-384d-7002-886f-11dddf5040ad",
     "urn:uuid:01973122-3a0e-73f0-9596-baceb84383aa",
     "urn:uuid:01973122-3bdd-78a3-898a-fb67c06387ed"
-  ]
+  ],
+  "put_advisory_lables": "urn:uuid:24ae57c3-4b57-4f4e-82c1-83ae26059a89",
+  "patch_advisory_lables": "urn:uuid:9ffb1a1b-14e0-4abb-943a-76906daa54aa"
 }

src/main.rs

@@ -191,6 +191,10 @@ async fn main() -> Result<(), anyhow::Error> {
             tx!(s.post_vulnerability_analyze?(scenario.analyze_purl.clone()));
             tx!(s.get_purl_details?(scenario.get_purl_details.clone()));
             tx!(s.get_recommendations?(scenario.get_recommendations.clone()));
+            tx!(s.put_advisory_labels?(scenario.put_advisory_lables.clone()));
+            tx!(s.patch_advisory_labels?(
+                scenario.patch_advisory_lables.clone()
+            ));
 
             s
         })

src/restapi.rs

@@ -1,4 +1,5 @@
-use goose::goose::{GooseUser, TransactionResult};
+use goose::goose::{GooseMethod, GooseRequest, GooseUser, TransactionResult};
+use reqwest::{Client, RequestBuilder};
 use serde_json::json;
 use std::sync::{
     Arc,
@@ -37,6 +38,56 @@ pub async fn search_advisory(user: &mut GooseUser) -> TransactionResult {
     Ok(())
 }
 
+async fn send_advisory_label_request(
+    advisory_id: String,
+    user: &mut GooseUser,
+    method: GooseMethod,
+    source: &str,
+    client_method: fn(&Client, String) -> RequestBuilder,
+) -> TransactionResult {
+    let path = format!("/api/v2/advisory/{}/label", advisory_id);
+    let json = json!({
+        "source": source,
+        "foo": "bar",
+        "space": "with space",
+        "empty": "",
+    });
+
+    let url = user.build_url(&path)?;
+
+    let reqwest_request_builder = client_method(&user.client, url);
+    let goose_request = GooseRequest::builder()
+        .method(method)
+        .path(path.as_str())
+        .set_request_builder(reqwest_request_builder.json(&json))
+        .build();
+    let _response = user.request(goose_request).await?;
+
+    Ok(())
+}
+
+pub async fn put_advisory_labels(advisory_id: String, user: &mut GooseUser) -> TransactionResult {
+    send_advisory_label_request(
+        advisory_id,
+        user,
+        GooseMethod::Put,
+        "It's a put request",
+        Client::put,
+    )
+    .await
+}
+
+pub async fn patch_advisory_labels(advisory_id: String, user: &mut GooseUser) -> TransactionResult {
+    send_advisory_label_request(
+        advisory_id,
+        user,
+        GooseMethod::Patch,
+        "It's a patch request",
+        Client::patch,
+    )
+    .await
+}
+
 pub async fn list_importer(user: &mut GooseUser) -> TransactionResult {
     let _response = user.get("/api/v2/importer").await?;
 

src/scenario/mod.rs

@@ -96,6 +96,12 @@ pub(crate) struct Scenario {
 
     #[serde(default, skip_serializing_if = "Option::is_none")]
     pub delete_sbom_pool: Option<Vec<String>>,
+
+    #[serde(with = "required")]
+    pub put_advisory_lables: Option<String>,
+
+    #[serde(with = "required")]
+    pub patch_advisory_lables: Option<String>,
 }
 
 impl Scenario {
@@ -136,6 +142,12 @@ impl Scenario {
                 .collect(),
         );
 
+        let put_advisory_lables = Some(format!("urn:uuid:{}", loader.put_advisory_lables().await?));
+        let patch_advisory_lables = Some(format!(
+            "urn:uuid:{}",
+            loader.patch_advisory_lables().await?
+        ));
+
         Ok(Self {
             get_sbom: large_sbom_digest.clone(),
             get_sbom_advisories: large_sbom_digest.clone(),
@@ -150,6 +162,8 @@ impl Scenario {
             get_purl_details,
             get_recommendations: recommendations_purl,
             delete_sbom_pool,
+            put_advisory_lables,
+            patch_advisory_lables,
         })
     }
 }
@@ -356,6 +370,28 @@ LIMIT 100
             .map(|row| row.get::<String, _>("id"))
             .collect())
     }
+
+    /// A advisory ID for put labels
+    pub async fn put_advisory_lables(&self) -> anyhow::Result<String> {
+        self.find(
+            r#"
+SELECT id::text as result
+FROM public.advisory  where labels is not null order by modified desc limit 1;
+"#,
+        )
+        .await
+    }
+
+    /// A advisory ID for patch labels
+    pub async fn patch_advisory_lables(&self) -> anyhow::Result<String> {
+        self.find(
+            r#"
+SELECT id::text as result
+FROM public.advisory  where labels is not null order by modified desc OFFSET 1 limit 1;
+"#,
+        )
+        .await
+    }
 }
 
 #[cfg(test)]