- added test for get_graphql_admin_notices()

jasonbahl · jasonbahl · commit 4453a6b7e5e4 · 2024-09-09T11:29:00.000-06:00
diff --git a/access-functions.php b/access-functions.php
@@ -974,11 +974,6 @@ static function ( \WPGraphQL\Admin\AdminNotices $admin_notices ) use ( $slug, $c
  * @return array|mixed[][] An array of admin notices
  */
 function get_graphql_admin_notices() {
-
-	if ( ! is_admin() ) {
-		return [];
-	}
-
 	$admin_notices = \WPGraphQL\Admin\AdminNotices::get_instance();
 	return $admin_notices->get_admin_notices();
 }
diff --git a/tests/wpunit/AdminNoticesTest.php b/tests/wpunit/AdminNoticesTest.php
@@ -48,6 +48,30 @@ public function testAddAndGetAdminNotices(): void {
 		$this->assertSame($config, $notices[$slug]);
 	}
 
+    /**
+     * Test adding and retrieving admin notices.
+     */
+    public function testGetAdminNotices(): void {
+        $adminNotices = AdminNotices::get_instance();
+
+        $slug = 'test-notice';
+        $config = [
+            'message' => 'Test Notice Message',
+            'type' => 'warning',
+            'is_dismissable' => true
+        ];
+
+        $adminNotices->add_admin_notice($slug, $config);
+
+        $notices = $adminNotices->get_admin_notices();
+        $this->assertArrayHasKey($slug, $notices);
+        $this->assertSame($config, $notices[$slug]);
+
+        $get_admin_notices = get_graphql_admin_notices();
+
+        $this->assertSame($get_admin_notices, $notices);
+    }
+
 	/**
 	 * Test removing admin notices.
 	 */
