From c76c620e4ec7d504fe42c9ca66bb55511fc9b23f Mon Sep 17 00:00:00 2001
From: Charles Lew <crlf0710@gmail.com>
Date: Tue, 12 Oct 2021 22:34:53 +0800
Subject: [PATCH] Add a test for `unsafe_code` lint.
---
.../unsafe_code/auxiliary/forge_unsafe_block.rs | 16 ++++++++++++++++
.../ui/lint/unsafe_code/forge_unsafe_block.rs | 16 ++++++++++++++++
2 files changed, 32 insertions(+)
create mode 100644 src/test/ui/lint/unsafe_code/auxiliary/forge_unsafe_block.rs
create mode 100644 src/test/ui/lint/unsafe_code/forge_unsafe_block.rs
diff --git a/src/test/ui/lint/unsafe_code/auxiliary/forge_unsafe_block.rs b/src/test/ui/lint/unsafe_code/auxiliary/forge_unsafe_block.rs
new file mode 100644
index 0000000000000..26871c98dbef4
--- /dev/null
+++ b/src/test/ui/lint/unsafe_code/auxiliary/forge_unsafe_block.rs
@@ -0,0 +1,16 @@
+// force-host
+// no-prefer-dynamic
+
+#![crate_type = "proc-macro"]
+
+extern crate proc_macro;
+
+use proc_macro::{Delimiter, Group, Ident, Span, TokenStream, TokenTree};
+
+#[proc_macro]
+pub fn forge_unsafe_block(input: TokenStream) -> TokenStream {
+ let mut output = TokenStream::new();
+ output.extend(Some(TokenTree::from(Ident::new("unsafe", Span::call_site()))));
+ output.extend(Some(TokenTree::from(Group::new(Delimiter::Brace, input))));
+ output
+}
diff --git a/src/test/ui/lint/unsafe_code/forge_unsafe_block.rs b/src/test/ui/lint/unsafe_code/forge_unsafe_block.rs
new file mode 100644
index 0000000000000..a1bd7b4131984
--- /dev/null
+++ b/src/test/ui/lint/unsafe_code/forge_unsafe_block.rs
@@ -0,0 +1,16 @@
+// check-pass
+// aux-build:forge_unsafe_block.rs
+
+#[macro_use]
+extern crate forge_unsafe_block;
+
+unsafe fn foo() {}
+
+#[forbid(unsafe_code)]
+fn main() {
+ // `forbid` doesn't work for non-user-provided unsafe blocks.
+ // see `UnsafeCode::check_expr`.
+ forge_unsafe_block! {
+ foo();
+ }
+}