Add unit tests for JavaScript semantic coloring

jonhoo · Jun 13, 2019 · d6ffcd6 · d6ffcd6
1 parent 07fe3b7
commit d6ffcd6
Showing 1 changed file with 54 additions and 0 deletions.
diff --git a/src/flamegraph/color/palettes.rs b/src/flamegraph/color/palettes.rs
@@ -100,3 +100,57 @@ pub(super) mod wakeup {
         BasicPalette::Aqua
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use crate::flamegraph::color::BasicPalette;
+
+    struct TestData {
+        input: String,
+        output: BasicPalette,
+    }
+
+    #[test]
+    fn js_returns_correct() {
+        use super::js;
+
+        let test_data = [
+            TestData {
+                input: String::from(" "),
+                output: BasicPalette::Green,
+            },
+            TestData {
+                input: String::from("something_[k]"),
+                output: BasicPalette::Orange,
+            },
+            TestData {
+                input: String::from("something/_[j]"),
+                output: BasicPalette::Green,
+            },
+            TestData {
+                input: String::from("something_[j]"),
+                output: BasicPalette::Aqua,
+            },
+            TestData {
+                input: String::from("some::thing"),
+                output: BasicPalette::Yellow,
+            },
+            TestData {
+                input: String::from("some:thing"),
+                output: BasicPalette::Aqua,
+            },
+            TestData {
+                input: String::from("some/ai.js"),
+                output: BasicPalette::Green,
+            },
+            TestData {
+                input: String::from("someai.js"),
+                output: BasicPalette::Red,
+            },
+        ];
+        for elem in test_data.iter() {
+            let result = js::resolve(&elem.input);
+            assert_eq!(result, elem.output);
+        }
+    }
+}