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. |
|
| @@ 202-210 (lines=9) @@ | ||
| 199 | * @param mixed $return_value Return value of the function. |
|
| 200 | * @return phpmock\Mock The mock object. |
|
| 201 | */ |
|
| 202 | protected function mock_function( $function_name, $return_value = null ) { |
|
| 203 | $builder = new MockBuilder(); |
|
| 204 | $builder->setNamespace( __NAMESPACE__ ) |
|
| 205 | ->setName( $function_name ) |
|
| 206 | ->setFunction( function() use ( &$return_value ) { |
|
| 207 | return $return_value; |
|
| 208 | } ); |
|
| 209 | return $builder->build()->enable(); |
|
| 210 | } |
|
| 211 | } |
|
| 212 | ||
