devlensio 1.0.1 → 1.0.2

package/extractors/rust/src/calls.rs

@@ -41,7 +41,7 @@ pub fn detect_calls(
     let mut edges = vec![];
 
     for pf in &repo.files {
-        if pf.is_test_file {
+        if pf.is_pure_test_file {
             continue;
         }
         for item in &pf.items {

package/extractors/rust/src/enrich.rs

@@ -57,7 +57,7 @@ pub fn enrich_nodes(
     let mut model_table: std::collections::HashMap<String, String> =
         std::collections::HashMap::new();
     for pf in &repo.files {
-        if pf.is_test_file {
+        if pf.is_pure_test_file {
             continue;
         }
         for item in &pf.items {

package/extractors/rust/src/imports.rs

@@ -34,7 +34,7 @@ pub fn detect_imports(
     let mut seen: std::collections::HashSet<(String, String)> = std::collections::HashSet::new();
 
     for pf in &repo.files {
-        if pf.is_test_file {
+        if pf.is_pure_test_file {
             // test files ARE leaf nodes, but they still import — emit edges
             // from the TEST node (same file id — parentFile covers it)
         }

package/extractors/rust/src/inheritance.rs

@@ -37,7 +37,7 @@ pub fn detect_inheritance(
     let mut edges = vec![];
     let mut nodes = vec![];
     for pf in &repo.files {
-        if pf.is_test_file {
+        if pf.is_pure_test_file {
             continue;
         }
         for item in &pf.items {

package/extractors/rust/src/nodes.rs

@@ -2,9 +2,11 @@
 //
 // Contract rules enforced here:
 //   - FILE id `file::rel/path.rs`; file nodes parent themselves
-//   - TEST files (any #[test] fn) = LEAF nodes: nodes=[] for them, test fns
-//     ride in metadata.testCases only (the bug that shipped in Python's
-//     parser — asserted in the bun suite)
+//   - PURE test files (tests/ integration files, or files whose every item is
+//     a #[test] fn) = LEAF TEST nodes; Rust's idiomatic inline #[cfg(test)]
+//     mod tests must NOT collapse a mixed production file — its non-test items
+//     emit normally and #[test] fns emit as FUNCTION nodes flagged
+//     metadata.isTestFunction (test bodies stay out of prod call edges there)
 //   - rawCode + codeHash on every FUNCTION/METHOD/STRUCT/ENUM/TRAIT/IMPL_BLOCK
 //   - deterministic ids (file-scoped; trait-impl method suffix for the
 //     same-name-different-trait collision class)
@@ -17,7 +19,7 @@ use serde_json::{Map, Value};
 pub fn collect_file_nodes(files: &[ParsedFile]) -> Vec<CodeNode> {
     let mut out = vec![];
     for pf in files {
-        if pf.is_test_file {
+        if pf.is_pure_test_file {
             let mut n = file_node(&pf.rel_path, pf.line_count, NODE_TEST);
             let meta = n
                 .metadata
@@ -30,25 +32,43 @@ pub fn collect_file_nodes(files: &[ParsedFile]) -> Vec<CodeNode> {
             }
             out.push(n);
         } else {
-            out.push(file_node(&pf.rel_path, pf.line_count, NODE_FILE));
+            let mut n = file_node(&pf.rel_path, pf.line_count, NODE_FILE);
+            // Mixed files keep a testCases index so consumers that never walk
+            // FUNCTION nodes still see the file's tests.
+            if !pf.test_fns.is_empty() {
+                let meta = n
+                    .metadata
+                    .entry("testCases".to_string())
+                    .or_insert_with(|| Value::Array(vec![]));
+                if let Value::Array(arr) = meta {
+                    for t in &pf.test_fns {
+                        arr.push(Value::from(t.clone()));
+                    }
+                }
+            }
+            out.push(n);
         }
     }
     out
 }
 
-/// Code nodes (FUNCTION/METHOD/STRUCT/ENUM/TRAIT/IMPL_BLOCK) — test fns are
-/// NOT emitted (leaf rule); they feed metadata.testCases only.
+/// Code nodes. PURE test files stay leaves (their fns ride in testCases).
+/// Mixed files emit ALL items; #[test] fns are flagged metadata.isTestFunction
+/// instead of being dropped, so a test function is addressable without a new
+/// node type.
 pub fn collect_code_nodes(files: &[ParsedFile]) -> Vec<CodeNode> {
     let mut out = vec![];
     for pf in files {
-        if pf.is_test_file {
+        if pf.is_pure_test_file {
             continue;
         }
         for item in &pf.items {
+            let mut n = item_node(pf, item);
             if item.is_test {
-                continue;
+                n.metadata
+                    .insert("isTestFunction".to_string(), Value::from(true));
             }
-            out.push(item_node(pf, item));
+            out.push(n);
         }
     }
     out

package/extractors/rust/src/orm_edges.rs

@@ -21,7 +21,7 @@ use std::collections::HashMap;
 pub fn table_to_model_map(repo: &ParsedRepo) -> HashMap<String, String> {
     let mut map = HashMap::new();
     for pf in &repo.files {
-        if pf.is_test_file {
+        if pf.is_pure_test_file {
             continue;
         }
         for item in &pf.items {
@@ -89,7 +89,7 @@ pub fn detect_orm_edges(
     let table_to_schema_file = table_to_schema_file_map(repo);
 
     for pf in &repo.files {
-        if pf.is_test_file {
+        if pf.is_pure_test_file {
             continue;
         }
         for item in &pf.items {

package/extractors/rust/src/parser.rs

@@ -162,7 +162,12 @@ pub struct ParsedFile {
     pub items: Vec<ParsedItem>,
     pub tables: Vec<TableDecl>,
     pub test_fns: Vec<String>,
-    pub is_test_file: bool,
+    /// True only when the file is EXCLUSIVELY tests (every top-level item is a
+    /// #[test] fn, or the file only holds inline #[cfg(test)] mods). Rust's
+    /// idiomatic inline `#[cfg(test)] mod tests` must NOT collapse a mixed
+    /// production file into a TEST leaf — only pure test files get leaf
+    /// treatment (mirrors the intent of go's _test.go rule).
+    pub is_pure_test_file: bool,
     pub line_count: i64,
 }
 
@@ -326,7 +331,7 @@ pub fn parse_file(rel_path: &str, source: &str) -> Result<ParsedFile, String> {
         items: vec![],
         tables: vec![],
         test_fns: vec![],
-        is_test_file: false,
+        is_pure_test_file: false,
         line_count: line_index.line_count() as i64,
     };
 
@@ -342,7 +347,7 @@ pub fn parse_file(rel_path: &str, source: &str) -> Result<ParsedFile, String> {
     }
     pf.ast.items = ast_items;
 
-    pf.is_test_file = !pf.test_fns.is_empty();
+    pf.is_pure_test_file = !pf.test_fns.is_empty() && pf.items.iter().all(|i| i.is_test);
     Ok(pf)
 }
 

package/extractors/rust/src/routes.rs

@@ -96,7 +96,7 @@ pub fn detect_routes(
     let mut func_lit_vars: HashMap<(String, String), String> = HashMap::new();
 
     for pf in &repo.files {
-        if pf.is_test_file {
+        if pf.is_pure_test_file {
             continue;
         }
         for item in &pf.items {
@@ -195,7 +195,7 @@ pub fn detect_routes(
     // ── actix-web / rocket: attribute macros ──
     if framework == "actix-web" || framework == "rocket" {
         for pf in &repo.files {
-            if pf.is_test_file {
+            if pf.is_pure_test_file {
                 continue;
             }
             for item in &pf.items {

package/extractors/rust/src/tests.rs

@@ -22,7 +22,7 @@ pub fn detect_tests(
 ) -> Vec<CodeEdge> {
     let mut edges = vec![];
     for pf in &repo.files {
-        if !pf.is_test_file {
+        if !pf.is_pure_test_file {
             continue;
         }
         let from = format!("file::{}", pf.rel_path);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "devlensio",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "Codebase intelligence engine for TypeScript/JavaScript/React/Next.js, Python, Java, Go, and Rust repositories.",
   "type": "module",
   "main": "./dist/index.js",