Add test for sanitize_type

pedro-mendonca · Jul 14, 2024 · a824561 · a824561
1 parent 3fcc650
commit a824561
Showing 1 changed file with 72 additions and 0 deletions.
diff --git a/tests/phpunit/includes/test-admin-notice.php b/tests/phpunit/includes/test-admin-notice.php
@@ -32,4 +32,76 @@ public function test_message() {
 
 	}
 
+	/**
+	 * Data provider.
+	 *
+	 * @var array
+	 */
+	public function provide_test_sanitize_type() {
+		return array(
+			// Supported.
+			array(
+				'type'            => 'error',
+				'expected_result' => 'notice-warning',
+			),
+			array(
+				'type'            => 'warning',
+				'expected_result' => 'notice-warning',
+			),
+			array(
+				'type'            => 'warning-spin',
+				'expected_result' => 'notice-warning-spin',
+			),
+			array(
+				'type'            => 'success',
+				'expected_result' => 'notice-success',
+			),
+			array(
+				'type'            => 'info',
+				'expected_result' => 'notice-info',
+			),
+			// Not supported.
+			array(
+				'type'            => 'other',
+				'expected_result' => '',
+			),
+			array(
+				'type'            => '',
+				'expected_result' => '',
+			),
+			array(
+				'type'            => null,
+				'expected_result' => '',
+			),
+			array(
+				'type'            => false,
+				'expected_result' => '',
+			),
+			array(
+				'type'            => true,
+				'expected_result' => '',
+			),
+			array(
+				'type'            => array(),
+				'expected_result' => '',
+			),
+		);
+	}
+
+	/**
+	 * Test type sanitization.
+	 *
+	 * @dataProvider provide_test_sanitize_type
+	 */
+	public function test_sanitize_type( $type, $expected_result ) {
+
+		$args = array(
+			'type' => $type;
+		);
+
+		$admin_notice = new Admin_Notice( $args );
+
+		$this->assertSame( $admin_notice->type, $expected_result );
+
+	}
 }