Automattic /
jetpack
| @@ 569-607 (lines=39) @@ | ||
| 566 | * |
|
| 567 | * @param string|null $invalid_token The invalid token to be returned in the response. Null if the tokens should be valid. |
|
| 568 | */ |
|
| 569 | private function setup_reconnect_test( $invalid_token ) { |
|
| 570 | switch ( $invalid_token ) { |
|
| 571 | case 'blog_token': |
|
| 572 | add_filter( |
|
| 573 | 'pre_http_request', |
|
| 574 | array( |
|
| 575 | $this, |
|
| 576 | 'intercept_validate_tokens_request_invalid_blog_token', |
|
| 577 | ), |
|
| 578 | 10, |
|
| 579 | 3 |
|
| 580 | ); |
|
| 581 | break; |
|
| 582 | case 'user_token': |
|
| 583 | add_filter( |
|
| 584 | 'pre_http_request', |
|
| 585 | array( |
|
| 586 | $this, |
|
| 587 | 'intercept_validate_tokens_request_invalid_user_token', |
|
| 588 | ), |
|
| 589 | 10, |
|
| 590 | 3 |
|
| 591 | ); |
|
| 592 | break; |
|
| 593 | case null: |
|
| 594 | add_filter( |
|
| 595 | 'pre_http_request', |
|
| 596 | array( |
|
| 597 | $this, |
|
| 598 | 'intercept_validate_tokens_request_valid_tokens', |
|
| 599 | ), |
|
| 600 | 10, |
|
| 601 | 3 |
|
| 602 | ); |
|
| 603 | break; |
|
| 604 | } |
|
| 605 | ||
| 606 | add_filter( 'jetpack_options', array( $this, 'mock_jetpack_options' ), 10, 2 ); |
|
| 607 | } |
|
| 608 | ||
| 609 | /** |
|
| 610 | * Restore the environment after the `reconnect` test has been run. |
|
| @@ 614-650 (lines=37) @@ | ||
| 611 | * |
|
| 612 | * @param string|null $invalid_token The invalid token to be returned in the response. Null if the tokens should be valid. |
|
| 613 | */ |
|
| 614 | private function shutdown_reconnect_test( $invalid_token ) { |
|
| 615 | switch ( $invalid_token ) { |
|
| 616 | case 'blog_token': |
|
| 617 | remove_filter( |
|
| 618 | 'pre_http_request', |
|
| 619 | array( |
|
| 620 | $this, |
|
| 621 | 'intercept_validate_tokens_request_invalid_blog_token', |
|
| 622 | ), |
|
| 623 | 10 |
|
| 624 | ); |
|
| 625 | break; |
|
| 626 | case 'user_token': |
|
| 627 | remove_filter( |
|
| 628 | 'pre_http_request', |
|
| 629 | array( |
|
| 630 | $this, |
|
| 631 | 'intercept_validate_tokens_request_invalid_user_token', |
|
| 632 | ), |
|
| 633 | 10 |
|
| 634 | ); |
|
| 635 | break; |
|
| 636 | case null: |
|
| 637 | remove_filter( |
|
| 638 | 'pre_http_request', |
|
| 639 | array( |
|
| 640 | $this, |
|
| 641 | 'intercept_validate_tokens_request_valid_tokens', |
|
| 642 | ), |
|
| 643 | 10 |
|
| 644 | ); |
|
| 645 | break; |
|
| 646 | } |
|
| 647 | ||
| 648 | remove_filter( 'jetpack_options', array( $this, 'mock_jetpack_options' ), 10 ); |
|
| 649 | remove_filter( 'pre_http_request', array( $this, 'intercept_validate_tokens_request' ), 10 ); |
|
| 650 | } |
|
| 651 | ||
| 652 | } |
|
| 653 | ||
