WIP

Author: Kobzol

.github/workflows/check.yml

@@ -13,16 +13,16 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        rust-version: [ 1.78.0, stable, beta, nightly ]
+        rust-version: [ 1.80.0, stable, beta, nightly ]
     steps:
       - name: Checkout sources
         uses: actions/checkout@v2
 
-      - name: Install stable toolchain
-        uses: actions-rust-lang/setup-rust-toolchain@v1
-        with:
-          toolchain: ${{ matrix.rust-version }}
-          components: clippy, rustfmt, llvm-tools-preview
+      - name: Install toolchain
+        run: |
+          rustup toolchain install ${{ matrix.rust-version }}
+          rustup default ${{ matrix.rust-version }}
+          rustup component add clippy rustfmt llvm-tools-preview
 
       - name: Build
         run: cargo build --all --all-targets
@@ -31,22 +31,23 @@ jobs:
         run: cargo test
 
       - name: Lint
-        run: clippy --all -- -D warnings
+        run: cargo clippy --all -- -D warnings
 
       - name: Check Rust formatting
         run: cargo fmt --all -- --check
+
   test-windows:
     name: Test on Windows
     runs-on: windows-latest
     steps:
       - name: Checkout sources
         uses: actions/checkout@v2
 
-      - name: Install stable toolchain
-        uses: actions-rust-lang/setup-rust-toolchain@v1
-        with:
-          toolchain: 1.78.0
-          components: clippy, rustfmt, llvm-tools-preview
+      - name: Install toolchain
+        run: |
+          rustup toolchain install 1.80.0
+          rustup default 1.80.0
+          rustup component add clippy rustfmt llvm-tools-preview
 
       - name: Build
         run: cargo build --all --all-targets

Cargo.lock

@@ -2,7 +2,7 @@
 name = "cargo-pgo"
 version = "0.2.8"
 edition = "2021"
-rust-version = "1.78.0"
+rust-version = "1.80.0"
 
 description = "Cargo subcommand for optimizing Rust binaries with PGO and BOLT."
 repository = "https://github.com/kobzol/cargo-pgo"
@@ -30,9 +30,9 @@ which = "7"
 clap = { version = "4.5", features = ["derive"] }
 log = "0.4"
 env_logger = "0.11"
-colored = "2"
+colored = "3"
 cargo_metadata = "0.19"
-humansize = "1"
+humansize = "2"
 semver = "1"
 tempfile = "3.14"
 regex = "1.11"
@@ -42,6 +42,3 @@ walkdir = "2.5"
 shellwords = "1.1"
 blake3 = "1.4"
 version_check = "0.9"
-
-[dev-dependencies]
-version_check = "0.9"

Cargo.toml

@@ -5,8 +5,7 @@ use anyhow::anyhow;
 use cargo_metadata::diagnostic::DiagnosticLevel;
 use cargo_metadata::{CompilerMessage, Message};
 use colored::Colorize;
-use humansize::file_size_opts::BINARY;
-use humansize::FileSize;
+use humansize::{format_size, BINARY};
 use once_cell::sync::OnceCell;
 use regex::Regex;
 use rustc_demangle::{demangle, Demangle};
@@ -153,7 +152,7 @@ fn print_pgo_profile_stats(stats: &ProfileStats, pgo_dir: &Path) -> anyhow::Resu
         } else {
             "file"
         },
-        stats.total_size.file_size(BINARY).unwrap().yellow(),
+        format_size(stats.total_size, BINARY).yellow(),
         cli_format_path(pgo_dir.display())
     );
     Ok(())

src/pgo/optimize.rs

@@ -326,9 +326,8 @@ rustflags = ["-Ctarget-cpu=native"]
     );
 
     let output = project.cmd(&["build", "--", "-v"]).run()?;
-    println!("{}", output.stderr());
-    assert!(output.stderr().contains("-Ctarget-cpu=native"));
-    assert!(output.stderr().contains("-Cprofile-generate"));
+    output.assert_stderr_contains("-Ctarget-cpu=native");
+    output.assert_stderr_contains("-Cprofile-generate");
     output.assert_ok();
 
     Ok(())

tests/integration/pgo.rs

@@ -167,6 +167,8 @@ pub trait OutputExt {
     fn assert_ok(self) -> Self;
     fn assert_error(self) -> Self;
 
+    fn assert_stderr_contains(&self, needle: &str);
+
     fn stdout(&self) -> String;
     fn stderr(&self) -> String;
 }
@@ -190,6 +192,13 @@ impl OutputExt for Output {
         self
     }
 
+    fn assert_stderr_contains(&self, needle: &str) {
+        let stderr = self.stderr();
+        if !stderr.contains(needle) {
+            panic!("STDERR\n{stderr}\nDOES NOT CONTAIN\n{needle}");
+        }
+    }
+
     fn stdout(&self) -> String {
         String::from_utf8_lossy(&self.stdout).to_string()
     }