Function get_test_image_dir

dstack-mr/tests/tdvf_parse.rs:31–75

Download and extract dstack release tarball if not already cached

()

/// Download and extract dstack release tarball if not already cached
fn get_test_image_dir() -> Result<PathBuf> {
    let cache_dir = std::env::temp_dir().join("dstack-mr-test-cache");
    let version_dir = cache_dir.join(DSTACK_VERSION);
    let image_dir = version_dir.join("dstack-0.5.5");
    let metadata_path = image_dir.join("metadata.json");

    // Return cached version if it exists
    if metadata_path.exists() {
        return Ok(image_dir);
    }

    eprintln!("Downloading dstack {DSTACK_VERSION} release for testing...",);
    std::fs::create_dir_all(&version_dir)?;

    // Download tarball
    let tarball_path = version_dir.join("dstack.tar.gz");
    let response =
        reqwest::blocking::get(DSTACK_RELEASE_URL).context("failed to download dstack release")?;

    if !response.status().is_success() {
        anyhow::bail!("failed to download: HTTP {}", response.status());
    }

    let bytes = response.bytes().context("failed to read response")?;
    std::fs::write(&tarball_path, bytes).context("failed to write tarball")?;

    eprintln!("Extracting tarball...");

    // Extract tarball
    let tarball = std::fs::File::open(&tarball_path)?;
    let decoder = flate2::read::GzDecoder::new(tarball);
    let mut archive = tar::Archive::new(decoder);
    archive
        .unpack(&version_dir)
        .context("failed to extract tarball")?;

    // Verify extraction
    if !metadata_path.exists() {
        anyhow::bail!("metadata.json not found after extraction");
    }

    eprintln!("Test image ready at: {}", image_dir.display());

    Ok(image_dir)
}

#[test]
#[ignore] // Run with: cargo test --release -- --ignored