code coverage suite

Cargo.toml

@@ -15,11 +15,13 @@ node-subtensor-runtime = { path = "runtime", version = "4.0.0-dev" }
 
 [build-dependencies]
 subtensor-linting = { path = "support/linting", version = "0.1.0" }
+subtensor-code-coverage = { path = "support/code-coverage", version = "0.1.0" }
+build-print.workspace = true
 syn.workspace = true
 quote.workspace = true
 proc-macro2.workspace = true
 walkdir.workspace = true
-rayon = "1.10"
+rayon.workspace = true
 
 [workspace]
 members = [
@@ -34,8 +36,8 @@ members = [
     "support/macros",
     "support/linting",
     "support/procedural-fork",
+    "support/code-coverage",
 ]
-exclude = ["support/procedural-fork"]
 resolver = "2"
 
 [workspace.lints.clippy]
@@ -46,6 +48,8 @@ unwrap-used = "deny"
 manual_inspect = "allow"
 
 [workspace.dependencies]
+rayon = "1"
+build-print = "0.1"
 cargo-husky = { version = "1", default-features = false }
 clap = "4.5.4"
 codec = { version = "3.2.2", default-features = false }
@@ -157,6 +161,8 @@ substrate-frame-rpc-system = { git = "https://github.com/paritytech/polkadot-sdk
 substrate-wasm-builder = { git = "https://github.com/paritytech/polkadot-sdk.git", tag = "v1.16.0-rc1" }
 frame-metadata = "16"
 
+procedural-fork = { version = "1.10.0-rc3", path = "support/procedural-fork" }
+
 [profile.release]
 panic = "unwind"
 

build.rs

@@ -1,11 +1,6 @@
 use rayon::prelude::*;
-use std::{
-    env, fs,
-    path::{Path, PathBuf},
-    str::FromStr,
-    sync::mpsc::channel,
-};
-use walkdir::WalkDir;
+use std::{env, fs, path::Path, str::FromStr, sync::mpsc::channel};
+use subtensor_code_coverage::collect_rust_files;
 
 use subtensor_linting::*;
 
@@ -14,7 +9,6 @@ fn main() {
     println!("cargo:rerun-if-changed=pallets");
     println!("cargo:rerun-if-changed=node");
     println!("cargo:rerun-if-changed=runtime");
-    println!("cargo:rerun-if-changed=lints");
     println!("cargo:rerun-if-changed=build.rs");
     println!("cargo:rerun-if-changed=src");
     println!("cargo:rerun-if-changed=support");
@@ -25,6 +19,9 @@ fn main() {
     // Collect all Rust source files in the workspace
     let rust_files = collect_rust_files(workspace_root);
 
+    // Generate code coverage report
+    subtensor_code_coverage::analyze_files(&rust_files, workspace_root);
+
     // Channel used to communicate errors back to the main thread from the parallel processing
     // as we process each Rust file
     let (tx, rx) = channel();
@@ -49,8 +46,7 @@ fn main() {
             for error in errors {
                 let loc = error.span().start();
                 let file_path = relative_path.display();
-                // note that spans can't go across thread boundaries without losing their location
-                // info so we we serialize here and send a String
+                // note that spans can't go across thread boundaries so we serialize here and send a String
                 tx.send(format!(
                     "cargo:warning={}:{}:{}: {}",
                     file_path, loc.line, loc.column, error,
@@ -59,6 +55,7 @@ fn main() {
             }
         };
 
+        track_lint(DisallowV1Benchmarks::lint(&parsed_file));
         track_lint(ForbidAsPrimitiveConversion::lint(&parsed_file));
         track_lint(RequireFreezeStruct::lint(&parsed_file));
         track_lint(RequireExplicitPalletIndex::lint(&parsed_file));
@@ -71,30 +68,3 @@ fn main() {
         println!("{error}");
     }
 }
-
-/// Recursively collects all Rust files in the given directory
-fn collect_rust_files(dir: &Path) -> Vec<PathBuf> {
-    let mut rust_files = Vec::new();
-
-    for entry in WalkDir::new(dir) {
-        let Ok(entry) = entry else {
-            continue;
-        };
-        let path = entry.path();
-
-        // Skip any path that contains "target" directory
-        if path
-            .components()
-            .any(|component| component.as_os_str() == "target")
-            || path.ends_with("build.rs")
-        {
-            continue;
-        }
-
-        if path.is_file() && path.extension().and_then(|ext| ext.to_str()) == Some("rs") {
-            rust_files.push(path.to_path_buf());
-        }
-    }
-
-    rust_files
-}