Match full domain path in src attr

plugins/wpe-headless/includes/replacement/callbacks.php

@@ -56,9 +56,7 @@ function wpe_headless_content_media_replacement( $content ) {
 		$frontend_uri = '/';
 	}
 
-	$content = str_replace( "src=\"{$frontend_uri}", "src=\"{$site_url}", $content );
-
-	return str_replace( 'src="//', 'src="/', $content );
+	return str_replace( "src=\"{$frontend_uri}/", "src=\"{$site_url}/", $content );
 }
 
 add_filter( 'preview_post_link', 'wpe_headless_post_preview_link', 10, 2 );

plugins/wpe-headless/tests/integration/replacement/test-replacement-callbacks.php

@@ -61,9 +61,11 @@ public function test_wpe_headless_content_replacement_filters_content_when_conte
 	 * Tests wpe_headless_content_media_replacement() replaces the frontend_uri value when content replacement is enabled.
 	 */
 	public function test_wpe_headless_content_media_replacement_filters_content_when_content_replacement_enabled() {
-		wpe_headless_update_setting( 'frontend_uri', 'http://foo.com' );
+		wpe_headless_update_setting( 'frontend_uri', 'http://foo.co' );
 		add_filter( 'wpe_headless_domain_replacement_enabled', '__return_true' );
-		$this->assertSame( '<img src="http://example.org">', wpe_headless_content_media_replacement( '<img src="http://foo.com">' ) );
+		# Do not replace partial domain main.
+		$this->assertSame( '<img src="http://foo.com/image.png">', wpe_headless_content_media_replacement( '<img src="http://foo.com/image.png">' ) );
+		$this->assertSame( '<img src="http://example.org/image.png">', wpe_headless_content_media_replacement( '<img src="http://foo.co/image.png">' ) );
 		wpe_headless_update_setting( 'frontend_uri', null );
 		remove_filter( 'wpe_headless_domain_replacement_enabled', '__return_true' );
 	}