Fix

Author: InSyncWithFoo

crates/ruff_linter/src/noqa.rs

@@ -1253,19 +1253,25 @@ mod tests {
     #[test]
     fn noqa_no_code() {
         let source = "# noqa:";
-        assert_debug_snapshot!(Directive::try_extract(source, TextSize::default()));
+        let directive = lex_inline_noqa(TextRange::up_to(source.text_len()), source);
+        assert_debug_snapshot!(directive);
+        assert_lexed_ranges_match_slices(directive, source);
     }
 
     #[test]
     fn noqa_no_code_invalid_suffix() {
         let source = "# noqa: foo";
-        assert_debug_snapshot!(Directive::try_extract(source, TextSize::default()));
+        let directive = lex_inline_noqa(TextRange::up_to(source.text_len()), source);
+        assert_debug_snapshot!(directive);
+        assert_lexed_ranges_match_slices(directive, source);
     }
 
     #[test]
     fn noqa_no_code_trailing_content() {
         let source = "# noqa:  # Foo";
-        assert_debug_snapshot!(Directive::try_extract(source, TextSize::default()));
+        let directive = lex_inline_noqa(TextRange::up_to(source.text_len()), source);
+        assert_debug_snapshot!(directive);
+        assert_lexed_ranges_match_slices(directive, source);
     }
 
     #[test]
@@ -1503,19 +1509,25 @@ mod tests {
     #[test]
     fn flake8_noqa_no_code() {
         let source = "# flake8: # noqa:";
-        assert_debug_snapshot!(Directive::try_extract(source, TextSize::default()));
+        let exemption = lex_file_exemption(TextRange::up_to(source.text_len()), source);
+        assert_debug_snapshot!(exemption);
+        assert_lexed_ranges_match_slices(exemption, source);
     }
 
     #[test]
     fn flake8_noqa_no_code_invalid_suffix() {
         let source = "# flake8: # noqa: foo";
-        assert_debug_snapshot!(Directive::try_extract(source, TextSize::default()));
+        let exemption = lex_file_exemption(TextRange::up_to(source.text_len()), source);
+        assert_debug_snapshot!(exemption);
+        assert_lexed_ranges_match_slices(exemption, source);
     }
 
     #[test]
     fn flake8_noqa_no_code_trailing_content() {
         let source = "# flake8: # noqa:  # Foo";
-        assert_debug_snapshot!(Directive::try_extract(source, TextSize::default()));
+        let exemption = lex_file_exemption(TextRange::up_to(source.text_len()), source);
+        assert_debug_snapshot!(exemption);
+        assert_lexed_ranges_match_slices(exemption, source);
     }
 
     #[test]
@@ -1529,19 +1541,25 @@ mod tests {
     #[test]
     fn ruff_noqa_no_code() {
         let source = "# ruff: # noqa:";
-        assert_debug_snapshot!(Directive::try_extract(source, TextSize::default()));
+        let exemption = lex_file_exemption(TextRange::up_to(source.text_len()), source);
+        assert_debug_snapshot!(exemption);
+        assert_lexed_ranges_match_slices(exemption, source);
     }
 
     #[test]
     fn ruff_noqa_no_code_invalid_suffix() {
         let source = "# ruff: # noqa: foo";
-        assert_debug_snapshot!(Directive::try_extract(source, TextSize::default()));
+        let exemption = lex_file_exemption(TextRange::up_to(source.text_len()), source);
+        assert_debug_snapshot!(exemption);
+        assert_lexed_ranges_match_slices(exemption, source);
     }
 
     #[test]
     fn ruff_noqa_no_code_trailing_content() {
         let source = "# ruff: # noqa:  # Foo";
-        assert_debug_snapshot!(Directive::try_extract(source, TextSize::default()));
+        let exemption = lex_file_exemption(TextRange::up_to(source.text_len()), source);
+        assert_debug_snapshot!(exemption);
+        assert_lexed_ranges_match_slices(exemption, source);
     }
 
     #[test]

crates/ruff_linter/src/snapshots/ruff_linter__noqa__tests__flake8_noqa_no_code.snap

@@ -1,7 +1,7 @@
 ---
 source: crates/ruff_linter/src/noqa.rs
-expression: "Directive::try_extract(source, TextSize::default())"
+expression: exemption
 ---
-Err(
-    MissingCodes,
+Ok(
+    None,
 )

crates/ruff_linter/src/snapshots/ruff_linter__noqa__tests__flake8_noqa_no_code_invalid_suffix.snap

@@ -1,7 +1,7 @@
 ---
 source: crates/ruff_linter/src/noqa.rs
-expression: "Directive::try_extract(source, TextSize::default())"
+expression: exemption
 ---
-Err(
-    MissingCodes,
+Ok(
+    None,
 )

crates/ruff_linter/src/snapshots/ruff_linter__noqa__tests__flake8_noqa_no_code_trailing_content.snap

@@ -1,7 +1,7 @@
 ---
 source: crates/ruff_linter/src/noqa.rs
-expression: "Directive::try_extract(source, TextSize::default())"
+expression: exemption
 ---
-Err(
-    MissingCodes,
+Ok(
+    None,
 )

crates/ruff_linter/src/snapshots/ruff_linter__noqa__tests__ruff_noqa_no_code.snap

@@ -1,7 +1,7 @@
 ---
 source: crates/ruff_linter/src/noqa.rs
-expression: "Directive::try_extract(source, TextSize::default())"
+expression: exemption
 ---
-Err(
-    MissingCodes,
+Ok(
+    None,
 )

crates/ruff_linter/src/snapshots/ruff_linter__noqa__tests__ruff_noqa_no_code_invalid_suffix.snap

@@ -1,7 +1,7 @@
 ---
 source: crates/ruff_linter/src/noqa.rs
-expression: "Directive::try_extract(source, TextSize::default())"
+expression: exemption
 ---
-Err(
-    MissingCodes,
+Ok(
+    None,
 )

crates/ruff_linter/src/snapshots/ruff_linter__noqa__tests__ruff_noqa_no_code_trailing_content.snap

@@ -1,7 +1,7 @@
 ---
 source: crates/ruff_linter/src/noqa.rs
-expression: "Directive::try_extract(source, TextSize::default())"
+expression: exemption
 ---
-Err(
-    MissingCodes,
+Ok(
+    None,
 )