Implement PDO::exec() API

Author: JanJakes

tests/WP_PDO_MySQL_On_SQLite_PDO_API_Tests.php

@@ -21,6 +21,38 @@ public function test_query(): void {
 		$this->assertEquals( 1, $result->fetchColumn() );
 	}
 
+	public function test_exec(): void {
+		$result = $this->driver->exec( 'SELECT 1' );
+		$this->assertEquals( 0, $result );
+
+		$result = $this->driver->exec( 'CREATE TABLE t (id INT)' );
+		$this->assertEquals( 0, $result );
+
+		$result = $this->driver->exec( 'INSERT INTO t (id) VALUES (1)' );
+		$this->assertEquals( 1, $result );
+
+		$result = $this->driver->exec( 'INSERT INTO t (id) VALUES (2), (3)' );
+		$this->assertEquals( 2, $result );
+
+		$result = $this->driver->exec( 'UPDATE t SET id = 10 + id WHERE id = 0' );
+		$this->assertEquals( 0, $result );
+
+		$result = $this->driver->exec( 'UPDATE t SET id = 10 + id WHERE id = 1' );
+		$this->assertEquals( 1, $result );
+
+		$result = $this->driver->exec( 'UPDATE t SET id = 10 + id WHERE id < 10' );
+		$this->assertEquals( 2, $result );
+
+		$result = $this->driver->exec( 'DELETE FROM t WHERE id = 11' );
+		$this->assertEquals( 1, $result );
+
+		$result = $this->driver->exec( 'DELETE FROM t' );
+		$this->assertEquals( 2, $result );
+
+		$result = $this->driver->exec( 'DROP TABLE t' );
+		$this->assertEquals( 0, $result );
+	}
+
 	public function test_begin_transaction(): void {
 		$result = $this->driver->beginTransaction();
 		$this->assertTrue( $result );

wp-includes/sqlite-ast/class-wp-pdo-mysql-on-sqlite.php

@@ -792,6 +792,17 @@ public function query( string $query, ?int $fetch_mode = PDO::FETCH_COLUMN, ...$
 		}
 	}
 
+	/**
+	 * PDO API: Execute a MySQL statement and return the number of affected rows.
+	 *
+	 * @return int|false The number of affected rows or false on failure.
+	 */
+	#[ReturnTypeWillChange]
+	public function exec( string $query ) {
+		$stmt = $this->query( $query );
+		return $stmt->rowCount();
+	}
+
 	/**
 	 * PDO API: Begin a transaction.
 	 *