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