test: use TestManager

awslabs · Oct 3, 2023 · 4f36010 · 4f36010
1 parent 795220d
commit 4f36010
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/apply.rs b/tests/apply.rs
@@ -21,8 +21,9 @@ use predicates::prelude::*; // Used for writing assertions
 
 #[tokio::test]
 async fn test_apply() -> Result<(), Box<dyn std::error::Error>> {
-    let mut c = util::setup().await?;
-    let cmd = c.args(&["--region", "local", "admin", "apply", "--dev"]);
+    let tm = util::setup().await?;
+    let mut c = tm.command()?;
+    let cmd = c.args(&["--region", "local", "admin", "apply"]);
     cmd.assert()
         .success()
         .stdout(predicate::str::contains("not yet implemented"));
@@ -32,7 +33,8 @@ async fn test_apply() -> Result<(), Box<dyn std::error::Error>> {
 #[tokio::test]
 #[should_panic(expected = "not yet implemented")]
 async fn test_apply_with_dev() {
-    let mut c = util::setup().await.unwrap();
+    let tm = util::setup().await.unwrap();
+    let mut c = tm.command().unwrap();
     let cmd = c.args(&["--region", "local", "admin", "apply", "--dev"]);
     cmd.unwrap();
 }