fix: make SnapTestRunner::with_fn more flexible

unimarkup · Oct 12, 2024 · c262f6c · c262f6c
1 parent 38b7519
commit c262f6c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 19 deletions.
diff --git a/commons/src/test_runner/snap_test_runner.rs b/commons/src/test_runner/snap_test_runner.rs
@@ -14,7 +14,7 @@ pub struct SnapTestRunner<'a, I = ()> {
 }
 
 impl<'a> SnapTestRunner<'a> {
-    pub fn with_fn_any<I, F>(name: &str, input: &'a I, mut func: F) -> SnapTestRunner<'a, ()>
+    pub fn with_fn<I, F>(name: &str, input: &'a I, mut func: F) -> SnapTestRunner<'a, ()>
     where
         I: AsRef<str>,
         F: FnMut(&I) -> String,
@@ -30,23 +30,6 @@ impl<'a> SnapTestRunner<'a> {
             snapshot,
         }
     }
-
-    pub fn with_fn<S, PF>(name: &str, input: &'a S, mut parser: PF) -> SnapTestRunner<'a, ()>
-    where
-        S: AsRef<[Token<'a>]>,
-        PF: for<'s, 'i> FnMut(&'s [Token<'i>]) -> String,
-    {
-        let snapshot = parser(input.as_ref());
-
-        SnapTestRunner {
-            info: None,
-            desc: None,
-            input: Token::flatten(input.as_ref()),
-            name: name.into(),
-            sub_path: None,
-            snapshot,
-        }
-    }
 }
 
 impl<'a, I> SnapTestRunner<'a, I>

diff --git a/frontend/tests/lexer/mod.rs b/frontend/tests/lexer/mod.rs
@@ -39,7 +39,7 @@ pub(crate) fn collect_snapshot_tests() -> Vec<Trial> {
 }
 
 fn run_snap_test(case: test_runner::test_file::TestCase) {
-    let runner = SnapTestRunner::with_fn_any(&case.test.name, &case.test.input, |input_str| {
+    let runner = SnapTestRunner::with_fn(&case.test.name, &case.test.input, |input_str| {
         let token_stream = unimarkup_frontend::lexer::TokenStream::tokenize(input_str);
 
         let token_snaps = token_stream