From d137529bd5ca7afbc3b94abdde2f338ef2c89899 Mon Sep 17 00:00:00 2001
From: Morgan Phillips <winter2718@gmail.com>
Date: Mon, 25 Jul 2016 18:49:39 -0700
Subject: [PATCH] Use cton_reader to simplify cfg traversal tests.

---
 cranelift/src/libcretonne/tests/cfg.rs | 85 ++++++++++++++++++++++++++
 cranelift/src/libcretonne/tests/lib.rs |  1 +
 2 files changed, 86 insertions(+)
 create mode 100644 cranelift/src/libcretonne/tests/cfg.rs
 create mode 100644 cranelift/src/libcretonne/tests/lib.rs

diff --git a/cranelift/src/libcretonne/tests/cfg.rs b/cranelift/src/libcretonne/tests/cfg.rs
new file mode 100644
index 0000000000..6ad024aefd
--- /dev/null
+++ b/cranelift/src/libcretonne/tests/cfg.rs
@@ -0,0 +1,85 @@
+extern crate cretonne;
+extern crate cton_reader;
+
+use self::cton_reader::parser::Parser;
+use self::cretonne::ir::entities::Ebb;
+use self::cretonne::cfg::ControlFlowGraph;
+
+fn test_reverse_postorder_traversal(function_source: &str, ebb_order: Vec<u32>) {
+    let func = &Parser::parse(function_source).unwrap()[0];
+    let cfg = ControlFlowGraph::new(&func);
+    let ebbs = ebb_order.iter().map(|n| Ebb::with_number(*n).unwrap())
+                               .collect::<Vec<Ebb>>();
+    for (ebb, key) in cfg.reverse_postorder_ebbs() {
+        assert_eq!(ebb, ebbs[key]);
+    }
+}
+
+#[test]
+fn simple_traversal() {
+    test_reverse_postorder_traversal("
+        function test(i32) {
+            ebb0(v0: i32):
+               brz v0, ebb1
+               jump ebb2
+            ebb1:
+                jump ebb3
+            ebb2:
+                v1 = iconst.i32 1
+                v2 = iadd v1, v0
+                brz v2, ebb2
+                v3 = iadd v1, v2
+                brz v3, ebb1
+                v4 = iadd v1, v3
+                brz v4, ebb4
+                jump ebb5
+            ebb3:
+                trap
+            ebb4:
+                trap
+            ebb5:
+                trap
+        }
+    ", vec![0, 2, 5, 4, 1, 3]);
+}
+
+#[test]
+fn loops_one() {
+    test_reverse_postorder_traversal("
+        function test(i32) {
+            ebb0(v0: i32):
+                jump ebb1
+            ebb1:
+                brnz v0, ebb3
+                jump ebb2
+            ebb2:
+                jump ebb3
+            ebb3:
+                return
+        }
+    ", vec![0, 1, 2, 3]);
+}
+
+#[test]
+fn loops_two() {
+    test_reverse_postorder_traversal("
+        function test(i32) {
+            ebb0(v0: i32):
+                brz v0, ebb1
+                jump ebb2
+            ebb1:
+                jump ebb3
+            ebb2:
+                brz v0, ebb4
+                jump ebb5
+            ebb3:
+                jump ebb4
+            ebb4:
+                brz v0, ebb3
+                jump ebb5
+            ebb5:
+                brz v0, ebb4
+                return
+        }
+    ", vec![0, 2, 5, 4, 3, 1]);
+}
diff --git a/cranelift/src/libcretonne/tests/lib.rs b/cranelift/src/libcretonne/tests/lib.rs
new file mode 100644
index 0000000000..26cb1041ac
--- /dev/null
+++ b/cranelift/src/libcretonne/tests/lib.rs
@@ -0,0 +1 @@
+pub mod cfg;