feat: test match only with filenames

crates/cli_bin/fixtures/match_filename/.grit/grit.yaml

@@ -0,0 +1,23 @@
+version: 0.0.1
+patterns:
+  - name: rewrite_filename
+    body: |
+      engine marzano(0.1)
+      language markdown
+
+      file($name, $body) where {
+        $name <: `index.js`
+      }
+    samples:
+      - input: |
+          // @filename: index.js
+
+          console.log('hello')
+        output: |
+          // @filename: index.js
+
+          console.log('hello')
+      - input: |
+          // @filename: main.js
+
+          // This should be unmatched

crates/cli_bin/tests/patterns_test.rs

@@ -247,3 +247,36 @@ fn tests_patterns_with_foreign_function_call_from_dot_grit_lib() -> Result<()> {
     assert!(output.status.success());
     Ok(())
 }
+
+#[test]
+fn checks_non_matching_yaml_sample() -> Result<()> {
+    let (_temp_dir, dir) = get_fixture("yaml_unmatched", false)?;
+
+    let mut test = get_test_cmd()?;
+    test.arg("patterns")
+        .arg("test")
+        .current_dir(dir);
+
+    let output = test.output()?;
+
+    assert!(output.status.success());
+
+    Ok(())
+}
+
+
+#[test]
+fn tests_match_only_with_file_name() -> Result<()> {
+    let (_temp_dir, dir) = get_fixture("match_filename", false)?;
+
+    let mut test = get_test_cmd()?;
+    test.arg("patterns")
+        .arg("test")
+        .current_dir(dir);
+
+    let output = test.output()?;
+
+    assert!(output.status.success());
+
+    Ok(())
+}

crates/cli_bin/tests/plumbing.rs

@@ -255,19 +255,3 @@ fn lists_imported_patterns() -> Result<()> {
 
     Ok(())
 }
-
-#[test]
-fn checks_non_matching_yaml_sample() -> Result<()> {
-    let (_temp_dir, dir) = get_fixture("yaml_unmatched", false)?;
-
-    let mut test = get_test_cmd()?;
-    test.arg("patterns")
-        .arg("test")
-        .current_dir(dir);
-
-    let output = test.output()?;
-
-    assert!(output.status.success());
-
-    Ok(())
-}

crates/gritmodule/src/testing.rs

@@ -223,7 +223,7 @@ pub fn test_pattern_sample(
                 });
             }
             MatchResult::Match(r) => {
-                if sample.input.contains("// @filename:") {
+                if is_multifile_sample(&sample.input) {
                     continue;
                 }
                 raw_actual_outputs.push(RichFile {
@@ -239,7 +239,7 @@ pub fn test_pattern_sample(
     if raw_actual_outputs.is_empty() {
         if sample.output.is_none() {
             return SampleTestResult::new_passing(matches, false);
-        } else {
+        } else if !is_multifile_sample(&sample.input) {
             return SampleTestResult {
                 matches,
                 state: GritTestResultState::FailedMatch,
@@ -268,7 +268,9 @@ pub fn test_pattern_sample(
 
     let mut raw_expected_outputs = infer_rich_files_from_content(&compiled.language, sample_output);
 
-    if raw_actual_outputs.len() < raw_expected_outputs.len() && compiled.is_multifile {
+    if raw_actual_outputs.len() < raw_expected_outputs.len()
+        && is_multifile_sample(&sample.input)
+    {
         for file in rich_files.iter() {
             if raw_actual_outputs.iter().any(|f| f.path == file.path) {
                 continue;
@@ -327,6 +329,10 @@ pub fn test_pattern_sample(
     }
 }
 
+fn is_multifile_sample(input: &str) -> bool {
+    input.contains("// @filename:")
+}
+
 #[derive(Debug)]
 pub enum MismatchInfo {
     Path(OutputInfo),