From cdd0873db690cbcea1e5a011bbfc7fc94c5d5a6e Mon Sep 17 00:00:00 2001
From: woppopo <woppopo@protonmail.com>
Date: Fri, 28 Jan 2022 18:41:35 +0900
Subject: [PATCH] Add a test case for using NonNull::new in const context

---
 library/core/tests/lib.rs |  2 ++
 library/core/tests/ptr.rs | 15 +++++++++++++++
 2 files changed, 17 insertions(+)

diff --git a/library/core/tests/lib.rs b/library/core/tests/lib.rs
index 841c114063dc..e5ae65f1ab5e 100644
--- a/library/core/tests/lib.rs
+++ b/library/core/tests/lib.rs
@@ -16,7 +16,9 @@
 #![feature(const_maybe_uninit_as_mut_ptr)]
 #![feature(const_maybe_uninit_assume_init)]
 #![feature(const_maybe_uninit_assume_init_read)]
+#![feature(const_nonnull_new)]
 #![feature(const_num_from_num)]
+#![feature(const_ptr_as_ref)]
 #![feature(const_ptr_read)]
 #![feature(const_ptr_write)]
 #![feature(const_ptr_offset)]
diff --git a/library/core/tests/ptr.rs b/library/core/tests/ptr.rs
index b9c0d75b702e..bbfd2d64dda4 100644
--- a/library/core/tests/ptr.rs
+++ b/library/core/tests/ptr.rs
@@ -274,6 +274,21 @@ fn test_unsized_nonnull() {
     assert!(ys == zs);
 }
 
+#[test]
+fn test_const_nonnull_new() {
+    const {
+        assert!(NonNull::new(core::ptr::null_mut::<()>()).is_none());
+
+        let value = &mut 0u32;
+        let mut ptr = NonNull::new(value).unwrap();
+        unsafe { *ptr.as_mut() = 42 };
+
+        let reference = unsafe { &*ptr.as_ref() };
+        assert!(*reference == *value);
+        assert!(*reference == 42);
+    };
+}
+
 #[test]
 #[allow(warnings)]
 // Have a symbol for the test below. It doesn’t need to be an actual variadic function, match the