Automattic / jetpack

	 * @return Mock The mock object.

	 * @throws MockEnabledException PHPUnit wasn't able to enable mock functions.

	 */

	protected function mock_function( $function_name, $return_value = null, $namespace = __NAMESPACE__ ) {

		$builder = new MockBuilder();

		$builder->setNamespace( $namespace )

			->setName( $function_name )

			->setFunction(

				function () use ( &$return_value ) {

					return $return_value;

				}

			);

		$mock = $builder->build();

		$mock->enable();

		return $mock;

	}

	/**

	 * Filter to set the default constant values.

	 *

	 * @return phpmock\Mock The mock object.

	 */

	protected function mock_function( $function_name, $return_value = null, $called_with = null ) {

		$builder = new MockBuilder();

		$builder->setNamespace( __NAMESPACE__ )

				->setName( $function_name )

				->setFunction(

					function ( $value ) use ( &$return_value, $called_with ) {

						if ( $called_with ) {

							$this->assertEquals( $value, $called_with );

						}

						return $return_value;

					}

				);

		return $builder->build()->enable();

	}

}