Add a test for the retry mechanism of the `API\Google\Ads::get_mercha…

…nt_link` method.
woocommerce · May 27, 2024 · 6e54e69 · 6e54e69
1 parent cdd6807
commit 6e54e69
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/tests/Unit/API/Google/AdsTest.php b/tests/Unit/API/Google/AdsTest.php
@@ -179,6 +179,30 @@ public function test_accept_merchant_link_already_accepted() {
 		$this->ads->accept_merchant_link( self::TEST_MERCHANT_ID );
 	}
 
+	public function test_accept_merchant_link_with_retry_get_merchant_link() {
+		$this->options->method( 'get_ads_id' )->willReturn( self::TEST_ADS_ID );
+		$link = new ProductLinkInvitation();
+		$mc   = $this->createMock( MerchantCenterLinkInvitationIdentifier::class );
+
+		$mc
+			->expects( $this->exactly( 4 ) )
+			->method( 'getMerchantCenterId' )
+			->willReturnOnConsecutiveCalls(
+				self::TEST_MERCHANT_ID + 3,
+				self::TEST_MERCHANT_ID + 2,
+				self::TEST_MERCHANT_ID + 1,
+				self::TEST_MERCHANT_ID
+			);
+
+		$link->setStatus( ProductLinkInvitationStatus::PENDING_APPROVAL );
+		$link->setMerchantCenter( $mc );
+
+		$service = $this->generate_mc_link_mock( [ $link ] );
+		$service->expects( $this->once() )->method( 'updateProductLinkInvitation' );
+
+		$this->ads->accept_merchant_link( self::TEST_MERCHANT_ID );
+	}
+
 	public function test_accept_merchant_link() {
 		$this->options->method( 'get_ads_id' )->willReturn( self::TEST_ADS_ID );
 		$link = new ProductLinkInvitation();