feat: allow controlling each test individually

To allow controlling/enabling each test (which requires an additional
value) individually, this change creates dedicated field in the scenario
file.

Author: ctron

empty.json5

@@ -1,5 +1,9 @@
 // an empty file, disabling all tests
 
 {
-  "large_sbom": null,
+  "get_sbom": null,
+  "get_sbom_advisories": null,
+  "get_sbom_packages": null,
+  "get_sbom_related": null,
+  "get_vulnerability": null
 }

src/main.rs

@@ -113,16 +113,20 @@ async fn main() -> Result<(), anyhow::Error> {
                 .register_transaction(tx!(list_sboms))
                 .register_transaction(tx!(list_sboms_paginated));
 
-            if let Some(large_sbom) = scenario.large_sbom.clone() {
-                s = s
-                    .register_transaction(tx!(get_sbom_info(large_sbom.clone())))
-                    .register_transaction(tx!(get_sbom_advisories(large_sbom.clone())))
-                    .register_transaction(tx!(get_sbom_packages(large_sbom.clone())))
-                    .register_transaction(tx!(get_sbom_related(large_sbom.clone())))
+            if let Some(value) = scenario.get_sbom.clone() {
+                s = s.register_transaction(tx!(get_sbom(value.clone())));
             }
-
-            if let Some(max_vuln) = scenario.max_vuln.clone() {
-                s = s.register_transaction(tx!(get_vulnerability(max_vuln.clone())))
+            if let Some(value) = scenario.get_sbom_advisories.clone() {
+                s = s.register_transaction(tx!(get_sbom_advisories(value.clone())));
+            }
+            if let Some(value) = scenario.get_sbom_packages.clone() {
+                s = s.register_transaction(tx!(get_sbom_packages(value.clone())));
+            }
+            if let Some(value) = scenario.get_sbom_related.clone() {
+                s = s.register_transaction(tx!(get_sbom_related(value.clone())));
+            }
+            if let Some(value) = scenario.get_vulnerability.clone() {
+                s = s.register_transaction(tx!(get_vulnerability(value.clone())))
             }
 
             s

src/restapi.rs

@@ -88,7 +88,7 @@ pub async fn list_vulnerabilities_paginated(user: &mut GooseUser) -> Transaction
     Ok(())
 }
 
-pub async fn get_sbom_info(sbom_id: String, user: &mut GooseUser) -> TransactionResult {
+pub async fn get_sbom(sbom_id: String, user: &mut GooseUser) -> TransactionResult {
     let _response = user.get(&format!("/api/v2/sbom/{sbom_id}")).await?;
 
     Ok(())

src/scenario.rs

@@ -62,10 +62,19 @@ mod required {
 #[derive(Clone, PartialEq, Eq, Debug, serde::Serialize, serde::Deserialize)]
 pub(crate) struct Scenario {
     #[serde(with = "required")]
-    pub large_sbom: Option<String>,
+    pub get_sbom: Option<String>,
 
     #[serde(with = "required")]
-    pub max_vuln: Option<String>,
+    pub get_sbom_advisories: Option<String>,
+
+    #[serde(with = "required")]
+    pub get_sbom_packages: Option<String>,
+
+    #[serde(with = "required")]
+    pub get_sbom_related: Option<String>,
+
+    #[serde(with = "required")]
+    pub get_vulnerability: Option<String>,
 }
 
 impl Scenario {
@@ -88,9 +97,16 @@ impl Scenario {
 
         let loader = Loader::new(db);
 
+        let large_sbom = Some(loader.large_sbom().await?);
+        let max_vuln = Some(loader.max_vuln().await?);
+
         Ok(Self {
-            large_sbom: Some(loader.large_sbom().await?),
-            max_vuln: Some(loader.max_vuln().await?),
+            get_sbom: large_sbom.clone(),
+            get_sbom_advisories: large_sbom.clone(),
+            get_sbom_related: large_sbom.clone(),
+            get_sbom_packages: large_sbom,
+
+            get_vulnerability: max_vuln,
         })
     }
 }
@@ -154,6 +170,12 @@ order by num desc
 mod test {
     use super::*;
 
+    #[derive(Clone, PartialEq, Eq, Debug, serde::Serialize, serde::Deserialize)]
+    pub(crate) struct Scenario {
+        #[serde(with = "required")]
+        pub large_sbom: Option<String>,
+    }
+
     #[test]
     fn missing() {
         let err = serde_json::from_str::<Scenario>(r#"{}"#).expect_err("Must be an error");
@@ -196,7 +218,7 @@ mod test {
     // Ensure the empty file parses
     #[test]
     fn empty() {
-        let _ =
-            serde_json5::from_str::<Scenario>(include_str!("../empty.json5")).expect("Must be ok");
+        let _ = serde_json5::from_str::<super::Scenario>(include_str!("../empty.json5"))
+            .expect("Must be ok");
     }
 }