Automattic / jetpack

	/**

	 * Testing the `connection/register` endpoint.

	 */

	public function test_connection_register() {

		add_filter( 'pre_http_request', array( static::class, 'intercept_register_request' ), 10, 3 );

		$this->request = new WP_REST_Request( 'POST', '/jetpack/v4/connection/register' );

		$this->request->set_header( 'Content-Type', 'application/json' );

		$this->request->set_body( wp_json_encode( array( 'registration_nonce' => wp_create_nonce( 'jetpack-registration-nonce' ) ) ) );

		$response = $this->server->dispatch( $this->request );

		$data     = $response->get_data();

		remove_filter( 'pre_http_request', array( static::class, 'intercept_register_request' ), 10 );

		// Manually clears filter added by Manager::register().

		remove_filter( 'jetpack_use_iframe_authorization_flow', '__return_false', 20 );

		$this->assertEquals( 200, $response->get_status() );

		$this->assertSame( 0, strpos( $data['authorizeUrl'], 'https://jetpack.wordpress.com/jetpack.authorize/' ) );

		// Asserts jetpack_register_site_rest_response filter is being properly hooked to add data from wpcom register endpoint response.

		$this->assertFalse( $data['allowInplaceAuthorization'] );

		$this->assertSame( '', $data['alternateAuthorizeUrl'] );

	}

	/**

	 * Testing the `connection/register` endpoint with allow_inplace_authorization as true.

	/**

	 * Testing the `connection/register` endpoint with alternate_authorization_url

	 */

	public function test_connection_register_with_alternate_auth_url() {

		add_filter( 'pre_http_request', array( static::class, 'intercept_register_request_with_alternate_auth_url' ), 10, 3 );

		$this->request = new WP_REST_Request( 'POST', '/jetpack/v4/connection/register' );

		$this->request->set_header( 'Content-Type', 'application/json' );

		$this->request->set_body( wp_json_encode( array( 'registration_nonce' => wp_create_nonce( 'jetpack-registration-nonce' ) ) ) );

		$response = $this->server->dispatch( $this->request );

		$data     = $response->get_data();

		remove_filter( 'pre_http_request', array( static::class, 'intercept_register_request_with_alternate_auth_url' ), 10 );

		// Manually clears filter added by Manager::register().

		remove_filter( 'jetpack_use_iframe_authorization_flow', '__return_false', 20 );

		$this->assertEquals( 200, $response->get_status() );

		$this->assertSame( 0, strpos( $data['authorizeUrl'], 'https://jetpack.wordpress.com/jetpack.authorize/' ) );

		// Asserts jetpack_register_site_rest_response filter is being properly hooked to add data from wpcom register endpoint response.

		$this->assertFalse( $data['allowInplaceAuthorization'] );

		$this->assertSame( Redirect::get_url( 'https://dummy.com' ), $data['alternateAuthorizeUrl'] );

	}

	/**

	 * This filter callback allow us to skip the database query by `Jetpack_Options` to retrieve the option.