Add tests for extract_full_block_and_advance()

dmsnell · sirreal · dmsnell · commit b8e88ceccd8a · 2026-01-21T10:13:34.000-07:00
Initially ported from sirreal#20 Co-authored-by: Jon Surrell <jonsurrell@git.wordpress.org>
diff --git a/tests/phpunit/tests/block-processor/wpBlockProcessor.php b/tests/phpunit/tests/block-processor/wpBlockProcessor.php
@@ -1296,6 +1296,98 @@ public function test_scans_directly_to_requested_block_type( string $html, strin
 		);
 	}
 
+	/**
+	 * Ensures that block extraction matches the behavior of the default block parser.
+	 *
+	 * @ticket {TICKET_NUMBER}
+	 *
+	 * @dataProvider data_various_block_posts
+	 *
+	 * @param string $test_document An HTML document to parse as blocks.
+	 */
+	public function test_extracts_equivalent_parses_as_parse_blocks( string $test_document ) {
+		$processor = new WP_Block_Processor( $test_document );
+		$blocks    = array();
+
+		while ( $processor->next_block( '*' ) ) {
+			$blocks[] = $processor->extract_full_block_and_advance();
+		}
+
+		$this->assertSame(
+			parse_blocks( $test_document ),
+			$blocks,
+			'Failed to properly parse the block structure.'
+		);
+	}
+
+	/**
+	 * Data provider.
+	 *
+	 * @return Generator
+	 */
+	public static function data_various_block_posts() {
+		yield 'Empty post' => array( '' );
+
+		yield 'Void block' => array( '<!-- wp:void /-->' );
+
+		yield 'Empty block' => array( '<!-- wp:empty --><!-- /wp:empty -->' );
+
+		yield 'Paragraph block' => array( '<!-- wp:paragraph --><p>Test</p><!-- /wp:paragraph -->' );
+
+		yield 'Paragraph block with attributes' => array(
+			'<!-- wp:paragraph {"dropCaps": true} --><p>Test</p><!-- /wp:paragraph -->',
+		);
+
+		yield 'Group with void inner' => array(
+			'<!-- wp:group --><!-- wp:void /--><!-- /wp:group -->',
+		);
+
+		yield 'Empty columns' => array(
+			'<!-- wp:columns --><!-- wp:column --><!-- wp:column /--><!-- /wp:columns -->',
+		);
+
+		yield 'Contentful columns' => array(
+			<<<HTML
+			<!-- wp:columns -->
+			<ul>
+			<!-- wp:column -->
+			<li>A good point.</li>
+			<!-- wp:column /-->
+			</ul>
+			<!-- /wp:columns -->
+HTML
+		);
+
+		yield 'Group with mixed content' => array(
+			<<<HTML
+			<!-- wp:group -->
+			<div>
+			<!-- wp:paragraph --><p>Test</p><!-- /wp:paragraph -->
+			This is freeform.
+			<!-- wp:void /-->
+			End
+			<!-- wp:footer ->
+			<footer>That&rsquo;s it!</footer>
+			<!-- /wp:footer -->
+			</div>
+			<!-- /wp:group -->
+HTML
+		);
+
+		yield 'Nested blocks' => array(
+			<<<HTML
+			<!-- wp:a -->
+			<div>
+			<!-- wp:b -->
+			<span><!-- wp:c /--></span>
+			<!-- /wp:b -->
+			</div>
+			<!-- /wp:a -->
+HTML
+
+		);
+	}
+
 	/**
 	 * Data provider.
 	 *
