Automattic /
jetpack
| @@ 256-264 (lines=9) @@ | ||
| 253 | * @param mixed $return_value Return value of the function. |
|
| 254 | * @return phpmock\Mock The mock object. |
|
| 255 | */ |
|
| 256 | protected function mock_function( $function_name, $return_value = null ) { |
|
| 257 | $builder = new MockBuilder(); |
|
| 258 | $builder->setNamespace( __NAMESPACE__ ) |
|
| 259 | ->setName( $function_name ) |
|
| 260 | ->setFunction( function() use ( &$return_value ) { |
|
| 261 | return $return_value; |
|
| 262 | } ); |
|
| 263 | return $builder->build()->enable(); |
|
| 264 | } |
|
| 265 | ||
| 266 | /** |
|
| 267 | * Mock $wpdb->get_var() and make it return a certain value. |
|
| @@ 214-222 (lines=9) @@ | ||
| 211 | * @param mixed $return_value Return value of the function. |
|
| 212 | * @return phpmock\Mock The mock object. |
|
| 213 | */ |
|
| 214 | protected function mock_function( $function_name, $return_value = null ) { |
|
| 215 | $builder = new MockBuilder(); |
|
| 216 | $builder->setNamespace( __NAMESPACE__ ) |
|
| 217 | ->setName( $function_name ) |
|
| 218 | ->setFunction( function() use ( &$return_value ) { |
|
| 219 | return $return_value; |
|
| 220 | } ); |
|
| 221 | return $builder->build()->enable(); |
|
| 222 | } |
|
| 223 | } |
|
| 224 | ||
