chore: create the #@exit-status: CODE annotation

This lets us verify a specific exit code is set by `bevy_lint`, rather than just ignoring it completely.

Author: BD103

bevy_lint/tests/ui-cargo/duplicate_bevy_dependencies/long_version_format/fail/Cargo.stderr

@@ -1,13 +1,13 @@
 error: multiple versions of the `bevy` crate found
-  --> Cargo.toml:12:26
+  --> Cargo.toml:14:26
    |
-12 | leafwing-input-manager = "0.13"
+14 | leafwing-input-manager = "0.13"
    |                          ^^^^^^
    |
 help: expected all crates to use `bevy` 0.17.2, but `leafwing-input-manager` uses `bevy` ^0.13
-  --> Cargo.toml:11:8
+  --> Cargo.toml:13:8
    |
-11 | bevy = { version = "0.17.2", default-features = false }
+13 | bevy = { version = "0.17.2", default-features = false }
    |        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
 note: the lint level is defined here
   --> src/main.rs:3:9

bevy_lint/tests/ui-cargo/duplicate_bevy_dependencies/long_version_format/fail/Cargo.toml

@@ -1,3 +1,5 @@
+#@exit-status: 101
+
 [package]
 name = "multiple-bevy-versions"
 publish = false

bevy_lint/tests/ui-cargo/duplicate_bevy_dependencies/long_version_format/pass/Cargo.toml

@@ -1,3 +1,5 @@
+#@check-pass
+
 [package]
 name = "single-bevy-version"
 publish = false

bevy_lint/tests/ui-cargo/duplicate_bevy_dependencies/short_version_format/fail/Cargo.stderr

@@ -1,13 +1,13 @@
 error: multiple versions of the `bevy` crate found
-  --> Cargo.toml:12:26
+  --> Cargo.toml:14:26
    |
-12 | leafwing-input-manager = "0.13"
+14 | leafwing-input-manager = "0.13"
    |                          ^^^^^^
    |
 help: expected all crates to use `bevy` 0.17.2, but `leafwing-input-manager` uses `bevy` ^0.13
-  --> Cargo.toml:11:8
+  --> Cargo.toml:13:8
    |
-11 | bevy = "0.17.2"
+13 | bevy = "0.17.2"
    |        ^^^^^^^^
 note: the lint level is defined here
   --> src/main.rs:3:9

bevy_lint/tests/ui-cargo/duplicate_bevy_dependencies/short_version_format/fail/Cargo.toml

@@ -1,3 +1,5 @@
+#@exit-status: 101
+
 [package]
 name = "multiple-bevy-versions"
 publish = false

bevy_lint/tests/ui-cargo/duplicate_bevy_dependencies/short_version_format/pass/Cargo.toml

@@ -1,3 +1,5 @@
+#@check-pass
+
 [package]
 name = "single-bevy-version"
 publish = false

bevy_lint/tests/ui_cargo.rs

@@ -4,7 +4,7 @@ use std::{
     process::{Command, Stdio},
 };
 
-use ui_test::{CommandBuilder, Config, status_emitter};
+use ui_test::{CommandBuilder, Config, OptWithLine, status_emitter};
 
 // This is set by Cargo to the absolute paths of `bevy_lint` and `bevy_lint_driver`.
 const LINTER_PATH: &str = env!("CARGO_BIN_EXE_bevy_lint");
@@ -36,12 +36,21 @@ fn main() {
     };
 
     let defaults = config.comment_defaults.base();
-    // The driver returns a '101' on error.
-    // This allows for any status code to be considered a success.
-    defaults.exit_status = None.into();
-
     defaults.require_annotations = None.into();
 
+    // Create the `#@exit-status: CODE` annotation. This can be used to ensure a UI test exits with
+    // a specific exit code (e.g. `bevy_lint` exits with code 101 when a denied lint is found).
+    config
+        .custom_comments
+        .insert("exit-status", |parser, args, _span| {
+            parser.exit_status = OptWithLine::new(
+                args.content
+                    .parse()
+                    .expect("expected `i32` as input for `exit-status`"),
+                args.span,
+            );
+        });
+
     // Run this `Config` for all paths that end with `Cargo.toml` resulting
     // only in the `Cargo` lints.
     ui_test::run_tests_generic(