1 files changed, 145 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/MwHooksHandler.php b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/MwHooksHandler.php
new file mode 100644
index 00000000..2e842ec1
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/MwHooksHandler.php
@@ -0,0 +1,145 @@
+<?php
+
+namespace SMW\Tests\Utils;
+
+use RuntimeException;
+use SMW\MediaWiki\Hooks\HookRegistry;
+
+/**
+ * @license GNU GPL v2+
+ * @since   1.9
+ *
+ * @author mwjames
+ */
+class MwHooksHandler {
+
+	/**
+	 * @var HookRegistry
+	 */
+	private $hookRegistry = null;
+
+	private $wgHooks = [];
+	private $inTestRegisteredHooks = [];
+
+	private $listOfSmwHooks = [
+		'SMWStore::updateDataBefore',
+
+		// Those shoudl not be disabled so that extension used
+		// by a test will run the registration in case an instance
+		// is cleared
+		//	'smwInitDatatypes',
+		//	'SMW::DataType::initTypes',
+
+		'smwInitProperties',
+		'SMW::Property::initProperties',
+		'SMW::Factbox::BeforeContentGeneration',
+		'SMW::SQLStore::updatePropertyTableDefinitions',
+		'SMW::Store::BeforeQueryResultLookupComplete',
+		'SMW::Store::AfterQueryResultLookupComplete',
+		'SMW::SQLStore::BeforeChangeTitleComplete',
+		'SMW::SQLStore::BeforeDeleteSubjectComplete',
+		'SMW::SQLStore::AfterDeleteSubjectComplete',
+		'SMW::Parser::BeforeMagicWordsFinder',
+		'SMW::SQLStore::BeforeDataRebuildJobInsert',
+		'SMW::SQLStore::AddCustomFixedPropertyTables',
+		'SMW::SQLStore::AfterDataUpdateComplete',
+		'SMW::Browse::AfterIncomingPropertiesLookupComplete',
+		'SMW::Browse::BeforeIncomingPropertyValuesFurtherLinkCreate'
+	];
+
+	/**
+	 * @since  2.0
+	 *
+	 * @return MwHooksHandler
+	 */
+	public function deregisterListedHooks() {
+
+		$listOfHooks = array_merge(
+			$this->listOfSmwHooks,
+			$this->getHookRegistry()->getHandlerList()
+		);
+
+		foreach ( $listOfHooks as $hook ) {
+
+			// MW 1.19
+			if ( method_exists( 'Hooks', 'clear' ) ) {
+				\Hooks::clear( $hook );
+			}
+
+			if ( !isset( $GLOBALS['wgHooks'][$hook] ) ) {
+				continue;
+			}
+
+			$this->wgHooks[$hook] = $GLOBALS['wgHooks'][$hook];
+			$GLOBALS['wgHooks'][$hook] = [];
+		}
+
+		return $this;
+	}
+
+	/**
+	 * @since  2.0
+	 *
+	 * @return MwHooksHandler
+	 */
+	public function restoreListedHooks() {
+
+		foreach ( $this->inTestRegisteredHooks as $hook ) {
+			unset( $GLOBALS['wgHooks'][$hook] );
+		}
+
+		foreach ( $this->wgHooks as $hook => $definition ) {
+			$GLOBALS['wgHooks'][$hook] = $definition;
+			unset( $this->wgHooks[$hook] );
+		}
+
+		return $this;
+	}
+
+	/**
+	 * @since  2.1
+	 *
+	 * @return MwHooksHandler
+	 */
+	public function register( $name, callable $callback ) {
+
+		$listOfHooks = array_merge(
+			$this->listOfSmwHooks,
+			$this->getHookRegistry()->getHandlerList()
+		);
+
+		if ( !in_array( $name, $listOfHooks ) ) {
+			throw new RuntimeException( "$name is not listed as registrable hook" );
+		}
+
+		$this->inTestRegisteredHooks[] = $name;
+		$GLOBALS['wgHooks'][$name][] = $callback;
+
+		return $this;
+	}
+
+	/**
+	 * @since  2.1
+	 *
+	 * @return MwHooksHandler
+	 */
+	public function invokeHooksFromRegistry() {
+		$this->getHookRegistry()->register();
+		return $this;
+	}
+
+	/**
+	 * @since  2.1
+	 *
+	 * @return HookRegistry
+	 */
+	public function getHookRegistry() {
+
+		if ( $this->hookRegistry === null ) {
+			 $this->hookRegistry = new HookRegistry( $GLOBALS, '' );
+		}
+
+		return $this->hookRegistry;
+	}
+
+}