Automattic / jetpack

	 *

	 * @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();

	}

}

	 * @return Mock The mock object.

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

	 */

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

		$builder = new MockBuilder();

		$builder->setNamespace( __NAMESPACE__ )

			->setName( $function_name )

			->setFunction(

				function() use ( &$return_value ) {

					return $return_value;

				}

			);

		$mock = $builder->build();

		$mock->enable();

		return $mock;

	}

}