fix and testing

Jinming-Hu · May 30, 2024 · 7ffedf6 · 7ffedf6
1 parent 803dcd2
commit 7ffedf6
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/sdk/storage/azure-storage-blobs/test/ut/block_blob_client_test.cpp b/sdk/storage/azure-storage-blobs/test/ut/block_blob_client_test.cpp
@@ -2155,4 +2155,17 @@ namespace Azure { namespace Storage { namespace Test {
         = Blobs::BlockBlobClient(m_blockBlobClient->GetUrl(), keyCredential, clientOptions);
     EXPECT_NO_THROW(blockBlobClient.GetProperties());
   }
+
+  TEST_F(BlockBlobClientTest, SyncCopyFromUriWithRetry_PLAYBACKONLY_)
+  {
+    auto sourceBlobClient = m_blobContainerClient->GetBlockBlobClient("source" + RandomString());
+    sourceBlobClient.UploadFrom(m_blobContent.data(), m_blobContent.size());
+
+    const std::string blobName = "dest" + RandomString();
+    auto destBlobClient = m_blobContainerClient->GetBlockBlobClient(blobName);
+
+    auto res = destBlobClient.CopyFromUri(sourceBlobClient.GetUrl() + GetSas());
+    EXPECT_EQ(res.RawResponse->GetStatusCode(), Azure::Core::Http::HttpStatusCode::Accepted);
+  }
+
 }}} // namespace Azure::Storage::Test
diff --git a/sdk/storage/azure-storage-common/inc/azure/storage/common/internal/storage_retry_policy.hpp b/sdk/storage/azure-storage-common/inc/azure/storage/common/internal/storage_retry_policy.hpp
@@ -15,6 +15,11 @@ namespace Azure { namespace Storage { namespace _internal {
     }
     ~StorageRetryPolicy() override {}
 
+    std::unique_ptr<HttpPolicy> Clone() const override
+    {
+      return std::make_unique<StorageRetryPolicy>(*this);
+    }
+
   protected:
     bool ShouldRetry(
         const std::unique_ptr<Core::Http::RawResponse>& response,