Automattic /
jetpack
| @@ 72-92 (lines=21) @@ | ||
| 69 | * @param array $args Array of argument sets, last value of each set is used as a return value. |
|
| 70 | * @return phpmock\Mock The mock object. |
|
| 71 | */ |
|
| 72 | protected function mock_function_with_args( $function_name, $args = array() ) { |
|
| 73 | $builder = new MockBuilder(); |
|
| 74 | $builder->setNamespace( __NAMESPACE__ ) |
|
| 75 | ->setName( $function_name ) |
|
| 76 | ->setFunction( |
|
| 77 | function() use ( &$args ) { |
|
| 78 | $current_args = func_get_args(); |
|
| 79 | ||
| 80 | foreach ( $args as $arg ) { |
|
| 81 | if ( array_slice( $arg, 0, -1 ) === $current_args ) { |
|
| 82 | return array_pop( $arg ); |
|
| 83 | } |
|
| 84 | } |
|
| 85 | } |
|
| 86 | ); |
|
| 87 | ||
| 88 | $mock = $builder->build(); |
|
| 89 | $mock->enable(); |
|
| 90 | ||
| 91 | return $mock; |
|
| 92 | } |
|
| 93 | ||
| 94 | } |
|
| 95 | ||
| @@ 154-174 (lines=21) @@ | ||
| 151 | * @param array $args Array of argument sets, last value of each set is used as a return value. |
|
| 152 | * @return phpmock\Mock The mock object. |
|
| 153 | */ |
|
| 154 | protected function mock_function_with_args( $function_name, $args = array() ) { |
|
| 155 | $builder = new MockBuilder(); |
|
| 156 | $builder->setNamespace( __NAMESPACE__ ) |
|
| 157 | ->setName( $function_name ) |
|
| 158 | ->setFunction( |
|
| 159 | function() use ( &$args ) { |
|
| 160 | $current_args = func_get_args(); |
|
| 161 | ||
| 162 | foreach ( $args as $arg ) { |
|
| 163 | if ( array_slice( $arg, 0, -1 ) === $current_args ) { |
|
| 164 | return array_pop( $arg ); |
|
| 165 | } |
|
| 166 | } |
|
| 167 | } |
|
| 168 | ); |
|
| 169 | ||
| 170 | $mock = $builder->build(); |
|
| 171 | $mock->enable(); |
|
| 172 | ||
| 173 | return $mock; |
|
| 174 | } |
|
| 175 | ||
| 176 | /** |
|
| 177 | * Mock a set of filters. |
|
