Automattic / jetpack

	 * @param mixed  $return_value  Return value of the function.

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

	}

}