testhelper.php

<?php
/**
 * Plugin Name: JSON REST API Testing Helper
 * Description: Creates helpers for the testing framework (such as Code Coverage)
 * Author: Ryan McCue
 * Author URI: http://ryanmccue.info/
 * Version: 0.1
 */

$GLOBALS['wp_json_server_testhelper'] = new WP_JSON_Server_TestHelper();

class WP_JSON_Server_TestHelper {
	protected $current_test = null;

	protected $coverage = null;
	protected $reports = array();

	public function __construct() {
		add_action('init', array($this, 'startCoverage'));
		add_filter('json_endpoints', array($this, 'addEndpoints'));
	}

	public function startCoverage() {
		if ( ! isset( $_REQUEST['_jsoncurrenttest'] ) ) {
			return;
		}

		if ( ! class_exists( 'PHP_CodeCoverage' ) ) {
			return;
		}

		$this->reports = get_transient('json_testhelper_coverage');
		if (empty($this->reports))
			$this->reports = array();

		$this->coverage = new PHP_CodeCoverage();
		$current_test = preg_replace( '#[^\w-:]+#i', '', $_REQUEST['current_test'] );
		$this->coverage->start( $current_test );
	}

	public function endCoverage() {
		if ( ! $this->coverage ) {
			return;
		}

		$this->coverage->end();
		$this->reports[] = serialize($this->coverage);
		set_transient('json_testhelper_coverage', $this->reports, 30 * MINUTE_IN_SECONDS);
	}

	public function addEndpoints($routes) {
		$routes['/testhelper/report'] = array(
			array( array( $this, 'getReports' ), WP_JSON_Server::METHOD_POST ),
		);
		return $routes;
	}

	public function getReports() {
		$this->reports = get_transient('json_testhelper_coverage');
		if (empty($this->reports)) {
			return new WP_Error('json_testhelper_no_report', __('No report data available', 'json_testhelper'), array('status' => 400));
		}

		if ( ! class_exists( 'PHP_CodeCoverage' ) ) {
			return new WP_Error('json_testhelper_missing_codecoverage', __('The CodeCoverage classes are missing', 'json_testhelper'), array('status' => 500));
		}

		$master = new PHP_CodeCoverage();
		foreach ($this->reports as $report) {
			$master->merge($report);
		}

		// Clean up
		delete_transient('json_testhelper_coverage');

		$data = array(
			'reports' => count($this->reports),
			'data' => serialize($master)
		);
		return $data;
	}
}