Test utils

1 files changed, 7 insertions, 32 deletions

diff --git a/tests/lexer.rs b/tests/lexer.rs
index beca19c24..6c7531596 100644
--- a/tests/lexer.rs
+++ b/tests/lexer.rs
@@ -1,56 +1,31 @@
 extern crate file;
-#[macro_use(assert_diff)]
-extern crate difference;
 extern crate libsyntax2;
+extern crate testutils;
 
-use std::path::{PathBuf, Path};
-use std::fs::read_dir;
+use std::path::{Path};
 use std::fmt::Write;
 
 use libsyntax2::{Token, tokenize};
+use testutils::{assert_equal_text, collect_tests};
 
 #[test]
 fn lexer_tests() {
-    for test_case in lexer_test_cases() {
+    for test_case in collect_tests(&["lexer"]) {
         lexer_test_case(&test_case);
     }
 }
 
-fn lexer_test_dir() -> PathBuf {
-    let dir = env!("CARGO_MANIFEST_DIR");
-    PathBuf::from(dir).join("tests/data/lexer")
-}
-
-fn lexer_test_cases() -> Vec<PathBuf> {
-    let mut acc = Vec::new();
-    let dir = lexer_test_dir();
-    for file in read_dir(&dir).unwrap() {
-        let file = file.unwrap();
-        let path = file.path();
-        if path.extension().unwrap_or_default() == "rs" {
-            acc.push(path);
-        }
-    }
-    acc.sort();
-    acc
-}
-
 fn lexer_test_case(path: &Path) {
     let actual = {
         let text = file::get_text(path).unwrap();
         let tokens = tokenize(&text);
         dump_tokens(&tokens, &text)
     };
-    let expected = file::get_text(&path.with_extension("txt")).unwrap();
+    let path = path.with_extension("txt");
+    let expected = file::get_text(&path).unwrap();
     let expected = expected.as_str();
     let actual = actual.as_str();
-    if expected == actual {
-        return
-    }
-    if expected.trim() == actual.trim() {
-        panic!("Whitespace difference!")
-    }
-    assert_diff!(expected, actual, "\n", 0)
+    assert_equal_text(expected, actual, &path)
 }
 
 fn dump_tokens(tokens: &[Token], text: &str) -> String {