tests: Enable testing of zstd compressed tarballs

Signed-off-by: Daniel Silverstone <dsilvers@digital-scurf.org>

Author: kinnison

tests/cli-rustup.rs

@@ -323,7 +323,7 @@ info: note that the toolchain 'nightly-{0}' is currently in use (directory overr
 }
 
 #[test]
-fn rustup_xz() {
+fn rustup_zstd() {
     setup(&|config| {
         set_current_dist_date(config, "2015-01-01");
         expect_stderr_ok(
@@ -335,7 +335,7 @@ fn rustup_xz() {
                 "nightly",
                 "--no-self-update",
             ],
-            for_host!(r"dist/2015-01-01/rust-std-nightly-{0}.tar.xz"),
+            for_host!(r"dist/2015-01-01/rust-std-nightly-{0}.tar.zst"),
         );
     });
 }

tests/cli-v2.rs

@@ -328,7 +328,10 @@ fn bad_sha_on_installer() {
             let file = file.unwrap();
             let path = file.path();
             let filename = path.to_string_lossy();
-            if filename.ends_with(".tar.gz") || filename.ends_with(".tar.xz") {
+            if filename.ends_with(".tar.gz")
+                || filename.ends_with(".tar.xz")
+                || filename.ends_with(".tar.zst")
+            {
                 rustup::utils::raw::write_file(&path, "xxx").unwrap();
             }
         }