Automattic /
jetpack
| @@ 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. |
|
| @@ 145-153 (lines=9) @@ | ||
| 142 | * @param mixed $return_value Return value of the function. |
|
| 143 | * @return phpmock\Mock The mock object. |
|
| 144 | */ |
|
| 145 | protected function mock_function( $function_name, $return_value = null ) { |
|
| 146 | $builder = new MockBuilder(); |
|
| 147 | $builder->setNamespace( __NAMESPACE__ ) |
|
| 148 | ->setName( $function_name ) |
|
| 149 | ->setFunction( function() use ( &$return_value ) { |
|
| 150 | return $return_value; |
|
| 151 | } ); |
|
| 152 | return $builder->build()->enable(); |
|
| 153 | } |
|
| 154 | } |
|
| 155 | ||
