diff --git a/cli/tests/integration/upgrade_tests.rs b/cli/tests/integration/upgrade_tests.rs index 9b0ad30bde..0d41a1649d 100644 --- a/cli/tests/integration/upgrade_tests.rs +++ b/cli/tests/integration/upgrade_tests.rs @@ -223,7 +223,11 @@ fn upgrade_prompt() { // - We need to use a pty here because the upgrade prompt // doesn't occur except when there's a pty. // - Version comes from the test server. - pty.expect(" 99999.99.99 Run `deno upgrade` to install it."); + pty.expect_any(&[ + " 99999.99.99 Run `deno upgrade` to install it.", + // it builds canary releases on main, so check for this in that case + "Run `deno upgrade --canary` to install it.", + ]); }); } diff --git a/test_util/src/lib.rs b/test_util/src/lib.rs index c38f2bb0fa..0f402fb52e 100644 --- a/test_util/src/lib.rs +++ b/test_util/src/lib.rs @@ -1282,6 +1282,15 @@ async fn main_server( .unwrap(), ); } + (&hyper::Method::GET, "/upgrade/sleep/canary-latest.txt") => { + tokio::time::sleep(Duration::from_secs(45)).await; + return Ok( + Response::builder() + .status(StatusCode::OK) + .body(Body::from("bda3850f84f24b71e02512c1ba2d6bf2e3daa2fd")) + .unwrap(), + ); + } (&hyper::Method::GET, "/release-latest.txt") => { return Ok( Response::builder() @@ -1291,6 +1300,14 @@ async fn main_server( .unwrap(), ); } + (&hyper::Method::GET, "/canary-latest.txt") => { + return Ok( + Response::builder() + .status(StatusCode::OK) + .body(Body::from("bda3850f84f24b71e02512c1ba2d6bf2e3daa2fd")) + .unwrap(), + ); + } _ => { let mut file_path = testdata_path().to_path_buf(); file_path.push(&req.uri().path()[1..].replace("%2f", "/"));