From d0c96186056c296e338ff9e69de07bf4d3b56af7 Mon Sep 17 00:00:00 2001
From: Carsten Bach <mail@carsten-bach.de>
Date: Tue, 1 Oct 2024 00:19:43 +0200
Subject: [PATCH] NEW test for allow_query_vars() method

---
 .../classes/endpoints/class-test-endpoint.php | 97 +++++++++++++++----
 1 file changed, 76 insertions(+), 21 deletions(-)

diff --git a/test/unit/php/includes/core/classes/endpoints/class-test-endpoint.php b/test/unit/php/includes/core/classes/endpoints/class-test-endpoint.php
index 838347ac4..fe885b9e9 100644
--- a/test/unit/php/includes/core/classes/endpoints/class-test-endpoint.php
+++ b/test/unit/php/includes/core/classes/endpoints/class-test-endpoint.php
@@ -38,7 +38,6 @@ public function test___construct(): void {
 			new Endpoint_Redirect( 'endpoint_redirect_1', $callback ),
 		);
 		$reg_ex    = 'reg_ex';
-
 		$instance  = new Endpoint(
 			$query_var,
 			$post_type,
@@ -67,18 +66,23 @@ public function test_get_slugs(): void {
 		// ini_set('display_startup_errors', '1');
 		// error_reporting(E_ALL);
 
-		$callback = function(){};
-		$instance = new Endpoint(
-			'query_var',
-			'gatherpress_event',
+		$query_var = 'query_var';
+		$post_type = 'gatherpress_event';
+		$callback  = function(){};
+		$types     = array(
+			new Endpoint_Template( 'endpoint_template_1', $callback ),
+			new Endpoint_Template( 'endpoint_template_2', $callback ),
+			new Endpoint_Redirect( 'endpoint_redirect_1', $callback ),
+		);
+		$reg_ex    = 'reg_ex';
+		$instance  = new Endpoint(
+			$query_var,
+			$post_type,
 			$callback,
-			array(
-				new Endpoint_Template( 'endpoint_template_1', $callback ),
-				new Endpoint_Template( 'endpoint_template_2', $callback ),
-				new Endpoint_Redirect( 'endpoint_redirect_1', $callback ),
-			),
-			'reg_ex',
+			$types,
+			$reg_ex,
 		);
+
 		$this->assertSame(
 			array(
 				'endpoint_template_1',
@@ -98,18 +102,23 @@ public function test_get_slugs(): void {
 	 * @return void
 	 */
 	public function test_get_rewrite_atts(): void {
-		$callback = function(){};
-		$instance = new Endpoint(
-			'query_var',
-			'gatherpress_event',
+		$query_var = 'query_var';
+		$post_type = 'gatherpress_event';
+		$callback  = function(){};
+		$types     = array(
+			new Endpoint_Template( 'endpoint_template_1', $callback ),
+			new Endpoint_Template( 'endpoint_template_2', $callback ),
+			new Endpoint_Redirect( 'endpoint_redirect_1', $callback ),
+		);
+		$reg_ex    = 'reg_ex';
+		$instance  = new Endpoint(
+			$query_var,
+			$post_type,
 			$callback,
-			array(
-				new Endpoint_Template( 'endpoint_template_1', $callback ),
-				new Endpoint_Template( 'endpoint_template_2', $callback ),
-				new Endpoint_Redirect( 'endpoint_redirect_1', $callback ),
-			),
-			'reg_ex',
+			$types,
+			$reg_ex,
 		);
+
 		$this->assertSame(
 			array(
 				'gatherpress_event' => '$matches[1]',
@@ -119,4 +128,50 @@ public function test_get_rewrite_atts(): void {
 			'Failed to assert that rewrite attributes match.'
 		);
 	}
+
+	/**
+	 * Coverage for maybe_flush_rewrite_rules method.
+	 *
+	 * @covers ::maybe_flush_rewrite_rules
+	 *
+	 * @return void
+	 */
+	public function test_maybe_flush_rewrite_rules(): void {}
+
+	/**
+	 * Coverage for allow_query_vars method.
+	 *
+	 * @covers ::allow_query_vars
+	 *
+	 * @return void
+	 */
+	public function test_allow_query_vars(): void {
+		$query_var = 'query_var';
+		$post_type = 'gatherpress_event';
+		$callback  = function(){};
+		$types     = array(
+			new Endpoint_Template( 'endpoint_template_1', $callback ),
+			new Endpoint_Template( 'endpoint_template_2', $callback ),
+			new Endpoint_Redirect( 'endpoint_redirect_1', $callback ),
+		);
+		$reg_ex    = 'reg_ex';
+		$instance  = new Endpoint(
+			$query_var,
+			$post_type,
+			$callback,
+			$types,
+			$reg_ex,
+		);
+
+		$this->assertSame(
+			array(
+				'apples',
+				'oranges',
+				'query_var',
+			),
+			$instance->allow_query_vars( array( 'apples', 'oranges' )),
+			'Failed to assert that merged query variables match.'
+		);
+	}
+
 }