Rust: improve modules test case

Author: aibaars

rust/ql/test/extractor-tests/crate_graph/crates.expected

@@ -1,23 +1,23 @@
-#-----| Crate(allocator_api2@0.2.21)
+#-----| Crate(allocator_api2@0.2.18)
 
 #-----| Crate(core@0.0.0)
 #-----|  -> Crate(rand@0.8.5)
 #-----|  -> Crate(rand_xorshift@0.3.0)
 
-#-----| Crate(compiler_builtins@0.1.140)
+#-----| Crate(compiler_builtins@0.1.138)
 #-----|  -> Crate(core@0.0.0)
 
 #-----| Crate(cfg_if@1.0.0)
 #-----|  -> Crate(core@0.0.0)
-#-----|  -> Crate(compiler_builtins@0.1.140)
+#-----|  -> Crate(compiler_builtins@0.1.138)
 
 #-----| Crate(std@0.0.0)
 #-----|  -> Crate(core@0.0.0)
-#-----|  -> Crate(compiler_builtins@0.1.140)
+#-----|  -> Crate(compiler_builtins@0.1.138)
 #-----|  -> Crate(cfg_if@1.0.0)
 #-----|  -> Crate(alloc@0.0.0)
-#-----|  -> Crate(hashbrown@0.15.2)
-#-----|  -> Crate(libc@0.2.169)
+#-----|  -> Crate(hashbrown@0.15.0)
+#-----|  -> Crate(libc@0.2.162)
 #-----|  -> Crate(rand@0.8.5)
 #-----|  -> Crate(rand_xorshift@0.3.0)
 #-----|  -> Crate(rustc_demangle@0.1.24)
@@ -28,7 +28,7 @@
 
 #-----| Crate(unicode_width@0.1.14)
 #-----|  -> Crate(core@0.0.0)
-#-----|  -> Crate(compiler_builtins@0.1.140)
+#-----|  -> Crate(compiler_builtins@0.1.138)
 #-----|  -> Crate(std@0.0.0)
 
 #-----| Crate(getopts@0.2.21)
@@ -38,17 +38,17 @@
 
 #-----| Crate(alloc@0.0.0)
 #-----|  -> Crate(core@0.0.0)
-#-----|  -> Crate(compiler_builtins@0.1.140)
+#-----|  -> Crate(compiler_builtins@0.1.138)
 #-----|  -> Crate(rand@0.8.5)
 #-----|  -> Crate(rand_xorshift@0.3.0)
 
-#-----| Crate(hashbrown@0.15.2)
-#-----|  -> Crate(allocator_api2@0.2.21)
+#-----| Crate(hashbrown@0.15.0)
+#-----|  -> Crate(allocator_api2@0.2.18)
 #-----|  -> Crate(core@0.0.0)
-#-----|  -> Crate(compiler_builtins@0.1.140)
+#-----|  -> Crate(compiler_builtins@0.1.138)
 #-----|  -> Crate(alloc@0.0.0)
 
-#-----| Crate(libc@0.2.169)
+#-----| Crate(libc@0.2.162)
 #-----|  -> Crate(core@0.0.0)
 
 #-----| Crate(rand_core@0.6.4)
@@ -61,27 +61,27 @@
 
 #-----| Crate(rustc_demangle@0.1.24)
 #-----|  -> Crate(core@0.0.0)
-#-----|  -> Crate(compiler_builtins@0.1.140)
+#-----|  -> Crate(compiler_builtins@0.1.138)
 
 #-----| Crate(panic_abort@0.0.0)
 #-----|  -> Crate(core@0.0.0)
-#-----|  -> Crate(compiler_builtins@0.1.140)
+#-----|  -> Crate(compiler_builtins@0.1.138)
 #-----|  -> Crate(cfg_if@1.0.0)
 #-----|  -> Crate(alloc@0.0.0)
-#-----|  -> Crate(libc@0.2.169)
+#-----|  -> Crate(libc@0.2.162)
 
 #-----| Crate(unwind@0.0.0)
 #-----|  -> Crate(core@0.0.0)
-#-----|  -> Crate(compiler_builtins@0.1.140)
+#-----|  -> Crate(compiler_builtins@0.1.138)
 #-----|  -> Crate(cfg_if@1.0.0)
-#-----|  -> Crate(libc@0.2.169)
+#-----|  -> Crate(libc@0.2.162)
 
 #-----| Crate(panic_unwind@0.0.0)
 #-----|  -> Crate(core@0.0.0)
-#-----|  -> Crate(compiler_builtins@0.1.140)
+#-----|  -> Crate(compiler_builtins@0.1.138)
 #-----|  -> Crate(cfg_if@1.0.0)
 #-----|  -> Crate(alloc@0.0.0)
-#-----|  -> Crate(libc@0.2.169)
+#-----|  -> Crate(libc@0.2.162)
 #-----|  -> Crate(unwind@0.0.0)
 
 #-----| Crate(proc_macro@0.0.0)
@@ -90,16 +90,16 @@
 
 #-----| Crate(std_detect@0.1.5)
 #-----|  -> Crate(core@0.0.0)
-#-----|  -> Crate(compiler_builtins@0.1.140)
+#-----|  -> Crate(compiler_builtins@0.1.138)
 #-----|  -> Crate(cfg_if@1.0.0)
 #-----|  -> Crate(alloc@0.0.0)
-#-----|  -> Crate(libc@0.2.169)
+#-----|  -> Crate(libc@0.2.162)
 
 #-----| Crate(test@0.0.0)
 #-----|  -> Crate(core@0.0.0)
 #-----|  -> Crate(std@0.0.0)
 #-----|  -> Crate(getopts@0.2.21)
-#-----|  -> Crate(libc@0.2.169)
+#-----|  -> Crate(libc@0.2.162)
 
 #-----| Crate(test@0.0.1)
 #-----|  -> Crate(core@0.0.0)

rust/ql/test/extractor-tests/crate_graph/modules.expected

@@ -1,40 +1,37 @@
-#-----| fn as_string
+#-----| Const
 
-#-----| fn length
+#-----| Static
 
-#-----| fn write
+#-----| enum X
+
+#-----| fn as_string
 
 #-----| fn as_string
 
 #-----| fn fmt
 
-#-----| struct X_List
+#-----| fn length
 
-#-----| Static
+#-----| impl ...::Display for ...::X { ... }
+#-----|  -> fn fmt
+
+#-----| impl AsString for ...::X { ... }
+#-----|  -> fn as_string
+
+#-----| mod crate
+#-----|  -> mod module
 
 #-----| mod module
+#-----|  -> Const
+#-----|  -> Static
+#-----|  -> enum X
 #-----|  -> fn length
-#-----|  -> fn write
+#-----|  -> impl ...::Display for ...::X { ... }
+#-----|  -> impl AsString for ...::X { ... }
 #-----|  -> struct X_List
-#-----|  -> Static
 #-----|  -> trait AsString
-#-----|  -> Const
-#-----|  -> impl AsString for ...::X { ... }
-#-----|  -> impl ...::Display for ...::X { ... }
-#-----|  -> enum X
 
-#-----| mod crate
-#-----|  -> mod module
+#-----| struct X_List
 
 #-----| trait AsString
 #-----|  -> fn as_string
-
-#-----| Const
-
-#-----| impl AsString for ...::X { ... }
-#-----|  -> fn as_string
-
-#-----| impl ...::Display for ...::X { ... }
-#-----|  -> fn fmt
-
-#-----| enum X

rust/ql/test/extractor-tests/crate_graph/modules.ql

@@ -6,12 +6,40 @@
 
 import rust
 
-query predicate nodes(Item i) {
-  i.getParentNode*() = any(Crate m | m.getName() = "test" and m.getVersion() = "0.0.1").getModule()
+predicate nodes(Item i) { i instanceof RelevantNode }
+
+class RelevantNode extends Item {
+  RelevantNode() {
+    this.getParentNode*() =
+      any(Crate m | m.getName() = "test" and m.getVersion() = "0.0.1").getModule()
+  }
 }
 
-query predicate edges(Item container, Item element) {
+predicate edges(RelevantNode container, RelevantNode element) {
   element = container.(Module).getItemList().getAnItem() or
   element = container.(Impl).getAssocItemList().getAnAssocItem() or
   element = container.(Trait).getAssocItemList().getAnAssocItem()
 }
+
+query predicate nodes(RelevantNode node, string attr, string val) {
+  nodes(node) and
+  (
+    attr = "semmle.label" and
+    val = node.toString()
+    or
+    attr = "semmle.order" and
+    val =
+      any(int i | node = rank[i](RelevantNode n | nodes(n) | n order by n.toString())).toString()
+  )
+}
+
+query predicate edges(RelevantNode pred, RelevantNode succ, string attr, string val) {
+  edges(pred, succ) and
+  (
+    attr = "semmle.label" and
+    val = ""
+    or
+    attr = "semmle.order" and
+    val = any(int i | succ = rank[i](Item s | edges(pred, s) | s order by s.toString())).toString()
+  )
+}