Changeset 43629

Timestamp:

09/05/2018 11:01:36 AM (6 years ago)

Author:

flixos90

Message:

Upgrade/Install: Introduce populate_site_meta().

Fixes #44896. See #41333.

Location:

trunk

Files:

• src/wp-admin/includes/schema.php (modified) (1 diff)
• tests/phpunit/tests/admin/includesSchema.php (modified) (5 diffs)

trunk/src/wp-admin/includes/schema.php

@@ -1293 +1293 @@
     $wpdb->query( "INSERT INTO $wpdb->sitemeta ( site_id, meta_key, meta_value ) VALUES " . $insert ); // phpcs:ignore WordPress.WP.PreparedSQL.NotPrepared
 }
+
+/**
+ * Creates WordPress site meta and sets the default values.
+ *
+ * @since 5.0.0
+ *
+ * @global wpdb $wpdb WordPress database abstraction object.
+ *
+ * @param int   $site_id Site ID to populate meta for.
+ * @param array $meta    Optional. Custom meta $key => $value pairs to use. Default empty array.
+ */
+function populate_site_meta( $site_id, array $meta = array() ) {
+    global $wpdb;
+
+    $site_id = (int) $site_id;
+
+    if ( ! is_site_meta_supported() ) {
+        return;
+    }
+
+    if ( empty( $meta ) ) {
+        return;
+    }
+
+    $insert = '';
+    foreach ( $meta as $meta_key => $meta_value ) {
+        if ( is_array( $meta_value ) ) {
+            $meta_value = serialize( $meta_value );
+        }
+        if ( ! empty( $insert ) ) {
+            $insert .= ', ';
+        }
+        $insert .= $wpdb->prepare( '( %d, %s, %s)', $site_id, $meta_key, $meta_value );
+    }
+
+    $wpdb->query( "INSERT INTO $wpdb->blogmeta ( blog_id, meta_key, meta_value ) VALUES " . $insert ); // phpcs:ignore WordPress.WP.PreparedSQL.NotPrepared
+
+    wp_cache_set( 'last_changed', microtime(), 'sites' );
+}

trunk/tests/phpunit/tests/admin/includesSchema.php

@@ -8 +8 @@
 
     private static $options;
+    private static $blogmeta;
     private static $sitemeta;
 
@@ -17 +18 @@
 
         self::$options  = 'testprefix_options';
+        self::$blogmeta = 'testprefix_blogmeta';
         self::$sitemeta = 'testprefix_sitemeta';
 
         $options  = self::$options;
+        $blogmeta = self::$blogmeta;
         $sitemeta = self::$sitemeta;
 
@@ -36 +39 @@
                 PRIMARY KEY  (option_id),
                 UNIQUE KEY option_name (option_name)
+            ) {$charset_collate}
+            "
+        );
+        $wpdb->query(
+            "
+            CREATE TABLE {$blogmeta} (
+                meta_id bigint(20) unsigned NOT NULL auto_increment,
+                blog_id bigint(20) unsigned NOT NULL default '0',
+                meta_key varchar(255) default NULL,
+                meta_value longtext,
+                PRIMARY KEY  (meta_id),
+                KEY meta_key (meta_key({$max_index_length})),
+                KEY blog_id (blog_id)
             ) {$charset_collate}
             "
@@ -61 +77 @@
 
         $options  = self::$options;
+        $blogmeta = self::$blogmeta;
         $sitemeta = self::$sitemeta;
 
         $wpdb->query( "DROP TABLE IF EXISTS {$options}" );
+        $wpdb->query( "DROP TABLE IF EXISTS {$blogmeta}" );
         $wpdb->query( "DROP TABLE IF EXISTS {$sitemeta}" );
     }
@@ -158 +176 @@
 
     /**
+     * @ticket 44896
+     * @group multisite
+     * @group ms-required
+     * @dataProvider data_populate_site_meta
+     */
+    function test_populate_site_meta( $meta, $expected ) {
+        global $wpdb;
+
+        $orig_blogmeta  = $wpdb->blogmeta;
+        $wpdb->blogmeta = self::$blogmeta;
+
+        populate_site_meta( 42, $meta );
+
+        $results = array();
+        foreach ( $expected as $meta_key => $value ) {
+            $results[ $meta_key ] = get_site_meta( 42, $meta_key, true );
+        }
+
+        $wpdb->query( "TRUNCATE TABLE {$wpdb->blogmeta}" );
+
+        $wpdb->blogmeta = $orig_blogmeta;
+
+        $this->assertEquals( $expected, $results );
+    }
+
+    public function data_populate_site_meta() {
+        return array(
+            array(
+                array(),
+                array(
+                    'unknown_value' => '',
+                ),
+            ),
+            array(
+                array(
+                    'custom_meta' => '1',
+                ),
+                array(
+                    'custom_meta' => '1',
+                ),
+            ),
+        );
+    }
+
+    /**
      * @ticket 44895
+     * @group multisite
      * @dataProvider data_populate_network_meta
      */