TestManager: Add flexibility to displayed columns

bottlerocket-os · Mar 1, 2023 · dbf1ece · dbf1ece
1 parent 609fe27
commit dbf1ece
Show file tree

Hide file tree

Showing 3 changed files with 259 additions and 311 deletions.
diff --git a/cli/src/status.rs b/cli/src/status.rs
@@ -1,9 +1,7 @@
 use anyhow::{Context, Result};
 use clap::Parser;
 use terminal_size::{Height, Width};
-use testsys_model::test_manager::{
-    CrdState, CrdType, SelectionParams, StatusProgress, TestManager,
-};
+use testsys_model::test_manager::{CrdState, CrdType, SelectionParams, StatusColumn, TestManager};
 
 /// Check the status of a TestSys object.
 #[derive(Debug, Parser)]
@@ -20,10 +18,6 @@ pub(crate) struct Status {
     #[clap(long, short = 'p')]
     progress: bool,
 
-    /// Include the `Test` status, too. Requires `--progress`
-    #[clap(long, requires("progress"))]
-    with_test: bool,
-
     /// Include the time the CRD was last updated
     #[clap(long, short = 'u')]
     with_time: bool,
@@ -68,14 +62,19 @@ impl Status {
             .await
             .context("Unable to get status")?;
 
-        if self.with_test {
-            status.with_progress(StatusProgress::WithTests);
-        } else if self.progress {
-            status.with_progress(StatusProgress::Resources);
+        status.add_column(StatusColumn::name());
+        status.add_column(StatusColumn::crd_type());
+        status.add_column(StatusColumn::state());
+        status.add_column(StatusColumn::passed());
+        status.add_column(StatusColumn::failed());
+        status.add_column(StatusColumn::skipped());
+
+        if self.progress {
+            status.add_column(StatusColumn::progress());
         }
 
         if self.with_time {
-            status.with_time();
+            status.add_column(StatusColumn::last_update());
         }
 
         if self.json {

diff --git a/model/src/test_manager/mod.rs b/model/src/test_manager/mod.rs
@@ -3,7 +3,7 @@ pub use error::{Error, Result};
 pub use manager::{read_manifest, TestManager};
 use serde::{Deserialize, Serialize};
 use serde_plain::derive_fromstr_from_deserialize;
-pub use status::StatusSnapshot;
+pub use status::{StatusColumn, StatusSnapshot};
 use std::collections::HashMap;
 
 mod delete;
@@ -85,10 +85,3 @@ pub enum ResourceState {
 }
 
 derive_fromstr_from_deserialize!(ResourceState);
-
-/// `StatusProgress` represents whether a `Test`'s `other_info` should be included or not.
-#[derive(Debug)]
-pub enum StatusProgress {
-    WithTests,
-    Resources,
-}