test: reinstate some unit test cases

Signed-off-by: Ahmed <ahmedtadde@gmail.com>
ahmedtadde · Nov 22, 2023 · 66dbeee · 66dbeee
1 parent 263cc9a
commit 66dbeee
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/crates/wash-cli/src/common/registry_cmd.rs b/crates/wash-cli/src/common/registry_cmd.rs
@@ -300,7 +300,8 @@ mod tests {
 
         let pull_basic: Cmd = Parser::try_parse_from(["reg", "pull", ECHO_WASM]).unwrap();
         let pull_all_flags: Cmd =
-            Parser::try_parse_from(["reg", "pull", "--allow-latest", "--insecure"]).unwrap();
+            Parser::try_parse_from(["reg", "pull", ECHO_WASM, "--allow-latest", "--insecure"])
+                .unwrap();
         let pull_all_options: Cmd = Parser::try_parse_from([
             "reg",
             "pull",
@@ -329,7 +330,7 @@ mod tests {
                 opts,
                 ..
             }) => {
-                assert_eq!(url, None);
+                assert_eq!(url, Some(ECHO_WASM.to_string()));
                 assert!(allow_latest);
                 assert!(opts.insecure);
             }
@@ -391,9 +392,11 @@ mod tests {
         };
 
         // Push logging.par.gz and pull from local registry
+        let logging_push_all_flags = format!("{LOCAL_REGISTRY}/logging:allflags");
         let push_all_flags: Cmd = Parser::try_parse_from([
             "reg",
             "push",
+            &logging_push_all_flags,
             "--artifact",
             &format!("{TESTDIR}/logging.par.gz"),
             "--insecure",
@@ -409,7 +412,7 @@ mod tests {
                 allow_latest,
                 ..
             }) => {
-                assert_eq!(url, None);
+                assert_eq!(url, Some(logging_push_all_flags));
                 assert_eq!(artifact, format!("{TESTDIR}/logging.par.gz"));
                 assert!(opts.insecure);
                 assert!(allow_latest);