Tests: use onlyMethods() only if it exists

Parsely · Oct 4, 2021 · bd8e7f3 · bd8e7f3
1 parent 8a07bca
commit bd8e7f3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/Unit/Integrations/IntegrationsTest.php b/tests/Unit/Integrations/IntegrationsTest.php
@@ -52,7 +52,13 @@ public function test_an_integration_can_be_registered_to_a_new_Integrations_obje
 	 * @uses \Parsely\Integrations\Integrations::register
 	 */
 	public function test_registered_integrations_have_their_integrate_method_called(): void {
-		$mock_integration = $this->getMockBuilder( Integration::class )->onlyMethods( array( 'integrate' ) )->getMock();
+		$mock_builder = $this->getMockBuilder( Integration::class );
+		// See https://github.com/Parsely/wp-parsely/issues/426.
+		if ( method_exists( $mock_builder, 'onlyMethods' ) ) {
+			$mock_integration = $mock_builder->onlyMethods( array( 'integrate' ) )->getMock();
+		} else {
+			$mock_integration = $mock_builder->setMethods( array( 'integrate' ) )->getMock();
+		}
 		$mock_integration->expects( $this->once() )->method( 'integrate' );
 
 		$integrations = new Integrations();