X-Git-Url: https://scripts.mit.edu/gitweb/autoinstallsdev/mediawiki.git/blobdiff_plain/19e297c21b10b1b8a3acad5e73fc71dcb35db44a..6932310fd58ebef145fa01eb76edf7150284d8ea:/tests/parser/DbTestRecorder.php

diff --git a/tests/parser/DbTestRecorder.php b/tests/parser/DbTestRecorder.php
new file mode 100644
index 00000000..2089f64a
--- /dev/null
+++ b/tests/parser/DbTestRecorder.php
@@ -0,0 +1,87 @@
+<?php
+/**
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ *
+ * @file
+ * @ingroup Testing
+ */
+
+use Wikimedia\Rdbms\IMaintainableDatabase;
+
+class DbTestRecorder extends TestRecorder {
+	public $version;
+	/** @var Database */
+	private $db;
+
+	public function __construct( IMaintainableDatabase $db ) {
+		$this->db = $db;
+	}
+
+	/**
+	 * Set up result recording; insert a record for the run with the date
+	 * and all that fun stuff
+	 */
+	function start() {
+		$this->db->begin( __METHOD__ );
+
+		if ( !$this->db->tableExists( 'testrun' )
+			|| !$this->db->tableExists( 'testitem' )
+		) {
+			print "WARNING> `testrun` table not found in database. Trying to create table.\n";
+			$updater = DatabaseUpdater::newForDB( $this->db );
+			$this->db->sourceFile( $updater->patchPath( $this->db, 'patch-testrun.sql' ) );
+			echo "OK, resuming.\n";
+		}
+
+		$this->db->insert( 'testrun',
+			[
+				'tr_date' => $this->db->timestamp(),
+				'tr_mw_version' => $this->version,
+				'tr_php_version' => PHP_VERSION,
+				'tr_db_version' => $this->db->getServerVersion(),
+				'tr_uname' => php_uname()
+			],
+			__METHOD__ );
+		if ( $this->db->getType() === 'postgres' ) {
+			$this->curRun = $this->db->currentSequenceValue( 'testrun_id_seq' );
+		} else {
+			$this->curRun = $this->db->insertId();
+		}
+	}
+
+	/**
+	 * Record an individual test item's success or failure to the db
+	 *
+	 * @param array $test
+	 * @param ParserTestResult $result
+	 */
+	function record( $test, ParserTestResult $result ) {
+		$this->db->insert( 'testitem',
+			[
+				'ti_run' => $this->curRun,
+				'ti_name' => $test['desc'],
+				'ti_success' => $result->isSuccess() ? 1 : 0,
+			],
+			__METHOD__ );
+	}
+
+	/**
+	 * Commit transaction and clean up for result recording
+	 */
+	function end() {
+		$this->db->commit( __METHOD__ );
+	}
+}