Automattic /
jetpack
| @@ 523-561 (lines=39) @@ | ||
| 520 | * |
|
| 521 | * @param string|null $invalid_token The invalid token to be returned in the response. Null if the tokens should be valid. |
|
| 522 | */ |
|
| 523 | private function setup_reconnect_test( $invalid_token ) { |
|
| 524 | switch ( $invalid_token ) { |
|
| 525 | case 'blog_token': |
|
| 526 | add_filter( |
|
| 527 | 'pre_http_request', |
|
| 528 | array( |
|
| 529 | $this, |
|
| 530 | 'intercept_validate_tokens_request_invalid_blog_token', |
|
| 531 | ), |
|
| 532 | 10, |
|
| 533 | 3 |
|
| 534 | ); |
|
| 535 | break; |
|
| 536 | case 'user_token': |
|
| 537 | add_filter( |
|
| 538 | 'pre_http_request', |
|
| 539 | array( |
|
| 540 | $this, |
|
| 541 | 'intercept_validate_tokens_request_invalid_user_token', |
|
| 542 | ), |
|
| 543 | 10, |
|
| 544 | 3 |
|
| 545 | ); |
|
| 546 | break; |
|
| 547 | case null: |
|
| 548 | add_filter( |
|
| 549 | 'pre_http_request', |
|
| 550 | array( |
|
| 551 | $this, |
|
| 552 | 'intercept_validate_tokens_request_valid_tokens', |
|
| 553 | ), |
|
| 554 | 10, |
|
| 555 | 3 |
|
| 556 | ); |
|
| 557 | break; |
|
| 558 | } |
|
| 559 | ||
| 560 | add_filter( 'jetpack_options', array( $this, 'mock_jetpack_options' ), 10, 2 ); |
|
| 561 | } |
|
| 562 | ||
| 563 | /** |
|
| 564 | * Restore the environment after the `reconnect` test has been run. |
|
| @@ 568-604 (lines=37) @@ | ||
| 565 | * |
|
| 566 | * @param string|null $invalid_token The invalid token to be returned in the response. Null if the tokens should be valid. |
|
| 567 | */ |
|
| 568 | private function shutdown_reconnect_test( $invalid_token ) { |
|
| 569 | switch ( $invalid_token ) { |
|
| 570 | case 'blog_token': |
|
| 571 | remove_filter( |
|
| 572 | 'pre_http_request', |
|
| 573 | array( |
|
| 574 | $this, |
|
| 575 | 'intercept_validate_tokens_request_invalid_blog_token', |
|
| 576 | ), |
|
| 577 | 10 |
|
| 578 | ); |
|
| 579 | break; |
|
| 580 | case 'user_token': |
|
| 581 | remove_filter( |
|
| 582 | 'pre_http_request', |
|
| 583 | array( |
|
| 584 | $this, |
|
| 585 | 'intercept_validate_tokens_request_invalid_user_token', |
|
| 586 | ), |
|
| 587 | 10 |
|
| 588 | ); |
|
| 589 | break; |
|
| 590 | case null: |
|
| 591 | remove_filter( |
|
| 592 | 'pre_http_request', |
|
| 593 | array( |
|
| 594 | $this, |
|
| 595 | 'intercept_validate_tokens_request_valid_tokens', |
|
| 596 | ), |
|
| 597 | 10 |
|
| 598 | ); |
|
| 599 | break; |
|
| 600 | } |
|
| 601 | ||
| 602 | remove_filter( 'jetpack_options', array( $this, 'mock_jetpack_options' ), 10 ); |
|
| 603 | remove_filter( 'pre_http_request', array( $this, 'intercept_validate_tokens_request' ), 10 ); |
|
| 604 | } |
|
| 605 | ||
| 606 | } |
|
| 607 | ||
