For multisite installs, reconstruct WP schema for all sites

Author: JanJakes

tests/WP_SQLite_Information_Schema_Reconstructor_Tests.php

@@ -39,8 +39,8 @@ public static function setUpBeforeClass(): void {
 		if ( ! defined( 'ABSPATH' ) ) {
 			define( 'ABSPATH', __DIR__ );
 		}
-		if ( ! function_exists( 'wp_installing' ) ) {
-			function wp_installing() {
+		if ( ! function_exists( 'is_multisite' ) ) {
+			function is_multisite() {
 				return false;
 			}
 		}

wp-includes/sqlite-ast/class-wp-sqlite-information-schema-reconstructor.php

@@ -145,7 +145,23 @@ private function get_wp_create_table_statements(): array {
 			throw new Exception( 'The "wp_get_db_schema()" function was not defined.' );
 		}
 
+		// Get schema for global tables and the main site.
 		$schema = wp_get_db_schema();
+
+		// For multisite installs, add schema definitions for all sites.
+		if ( is_multisite() ) {
+			$site_ids = get_sites(
+				array(
+					'fields' => 'ids',
+					'number' => PHP_INT_MAX,
+				)
+			);
+			foreach ( $site_ids as $site_id ) {
+				$schema .= wp_get_db_schema( 'blog', $site_id );
+			}
+		}
+
+		// Parse the schema.
 		$parser = $this->driver->create_parser( $schema );
 		while ( $parser->next_query() ) {
 			$ast = $parser->get_query_ast();