Add tests for REST API unauthorized error handling

WordPress · Dec 19, 2024 · a0f460d · a0f460d
1 parent 12a229c
commit a0f460d
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/...s/optimization-detective/tests/test-optimization-detective-rest-api-site-health-check.php b/...s/optimization-detective/tests/test-optimization-detective-rest-api-site-health-check.php
@@ -43,9 +43,31 @@ public function test_rest_api_available(): void {
 			),
 		);
 
-		$result = od_optimization_detective_rest_api_test();
+		$result           = od_optimization_detective_rest_api_test();
+		$od_rest_api_info = get_option( 'od_rest_api_info', array() );
 
 		$this->assertSame( 'good', $result['status'] );
+		$this->assertSame( 'ok', isset( $od_rest_api_info['status'] ) ? $od_rest_api_info['status'] : '' );
+		$this->assertTrue( isset( $od_rest_api_info['available'] ) ? $od_rest_api_info['available'] : false );
+	}
+
+	/**
+	 * Test behavior when REST API returns unauthorized error.
+	 */
+	public function test_rest_api_unauthorized(): void {
+		$this->mocked_responses = array(
+			get_rest_url( null, OD_REST_API_NAMESPACE . OD_URL_METRICS_ROUTE ) => $this->build_mock_response(
+				401,
+				'Unauthorized'
+			),
+		);
+
+		$result           = od_optimization_detective_rest_api_test();
+		$od_rest_api_info = get_option( 'od_rest_api_info', array() );
+
+		$this->assertSame( 'recommended', $result['status'] );
+		$this->assertSame( 'unauthorized', isset( $od_rest_api_info['status'] ) ? $od_rest_api_info['status'] : '' );
+		$this->assertFalse( isset( $od_rest_api_info['available'] ) ? $od_rest_api_info['available'] : true );
 	}
 
 	/**