[api] Add listQuarantinedTests support (#489)

Add support for invoking this endpoint. It will be used in a follow-up PR.

Author: gnalh

api/src/client.rs

@@ -184,6 +184,41 @@ impl ApiClient {
         .await
     }
 
+    pub async fn list_quarantined_tests(
+        &self,
+        request: &message::ListQuarantinedTestsRequest,
+    ) -> anyhow::Result<message::ListQuarantinedTestsResponse> {
+        CallApi {
+            action: || async {
+                let response = self
+                    .trunk_api_client
+                    .post(format!("{}{}/flaky-tests/list-quarantined-tests", self.api_host, self.version_path_prefix))
+                    .json(&request)
+                    .send()
+                    .await?;
+
+                let response = status_code_help(
+                    response,
+                    CheckUnauthorized::Check,
+                    CheckNotFound::DoNotCheck,
+                    |_| String::from("Failed to list quarantined tests."),
+                    &self.api_host,
+                    &self.org_url_slug,
+                )?;
+
+                self.deserialize_response::<message::ListQuarantinedTestsResponse>(response).await
+            },
+            log_progress_message: |time_elapsed, _| {
+                format!("Listing quarantined tests from Trunk services is taking longer than expected. It has taken {} seconds so far.", time_elapsed.as_secs())
+            },
+            report_slow_progress_message: |time_elapsed| {
+                format!("Listing quarantined tests from Trunk services is taking longer than {} seconds", time_elapsed.as_secs())
+            },
+        }
+        .call_api()
+        .await
+    }
+
     pub async fn put_bundle_to_s3<U: AsRef<str>, B: AsRef<Path>>(
         &self,
         url: U,

api/src/message.rs

@@ -63,3 +63,56 @@ pub struct CreateBundleUploadIntentResponse {
 pub struct TelemetryUploadMetricsRequest {
     pub upload_metrics: proto::upload_metrics::trunk::UploadMetrics,
 }
+
+#[derive(Debug, Serialize, Clone, Deserialize, PartialEq)]
+pub struct PageQuery {
+    #[serde(rename = "pageSize")]
+    pub page_size: i32,
+    #[serde(rename = "pageToken")]
+    pub page_token: String,
+}
+
+#[derive(Debug, Serialize, Clone, Deserialize, PartialEq)]
+pub struct Page {
+    pub total_rows: i32,
+    pub total_pages: i32,
+    pub next_page_token: String,
+    pub prev_page_token: String,
+    pub last_page_token: String,
+    pub page_index: i32,
+}
+
+#[derive(Debug, Serialize, Clone, Deserialize, PartialEq)]
+pub struct ListQuarantinedTestsRequest {
+    pub repo: RepoUrlParts,
+    #[serde(rename = "orgUrlSlug")]
+    pub org_url_slug: String,
+    #[serde(rename = "pageQuery")]
+    pub page_query: PageQuery,
+}
+
+#[derive(Debug, Serialize, Clone, Deserialize, PartialEq)]
+pub enum QuarantineSetting {
+    #[serde(rename = "ALWAYS_QUARANTINE")]
+    AlwaysQuarantine,
+    #[serde(rename = "AUTO_QUARANTINE")]
+    AutoQuarantine,
+}
+
+#[derive(Debug, Serialize, Clone, Deserialize, PartialEq)]
+pub struct QuarantinedTest {
+    pub name: String,
+    pub parent: Option<String>,
+    pub file: Option<String>,
+    #[serde(rename = "className")]
+    pub class_name: Option<String>,
+    pub status: String,
+    pub quarantine_setting: QuarantineSetting,
+    pub test_case_id: String,
+}
+
+#[derive(Debug, Serialize, Clone, Deserialize, PartialEq)]
+pub struct ListQuarantinedTestsResponse {
+    pub quarantined_tests: Vec<QuarantinedTest>,
+    pub page: Page,
+}