| @@ 121-129 (lines=9) @@ | ||
| 118 | * @param mixed $return_value Return value of the function. |
|
| 119 | * @return phpmock\Mock The mock object. |
|
| 120 | */ |
|
| 121 | protected function mock_function( $function_name, $return_value = null ) { |
|
| 122 | $builder = new MockBuilder(); |
|
| 123 | $builder->setNamespace( __NAMESPACE__ ) |
|
| 124 | ->setName( $function_name ) |
|
| 125 | ->setFunction( function() use ( &$return_value ) { |
|
| 126 | return $return_value; |
|
| 127 | } ); |
|
| 128 | return $builder->build()->enable(); |
|
| 129 | } |
|
| 130 | } |
|
| 131 | ||
| @@ 258-266 (lines=9) @@ | ||
| 255 | * @param mixed $return_value Return value of the function. |
|
| 256 | * @return phpmock\Mock The mock object. |
|
| 257 | */ |
|
| 258 | protected function mock_function( $function_name, $return_value = null ) { |
|
| 259 | $builder = new MockBuilder(); |
|
| 260 | $builder->setNamespace( __NAMESPACE__ ) |
|
| 261 | ->setName( $function_name ) |
|
| 262 | ->setFunction( function() use ( &$return_value ) { |
|
| 263 | return $return_value; |
|
| 264 | } ); |
|
| 265 | return $builder->build()->enable(); |
|
| 266 | } |
|
| 267 | ||
| 268 | /** |
|
| 269 | * Mock $wpdb->get_var() and make it return a certain value. |
|