Ignore other providers for specific test for a provider

snowflakedb · Jul 12, 2024 · ae012ae · ae012ae
1 parent a8be169
commit ae012ae
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 5 deletions.
diff --git a/Snowflake.Data.Tests/UnitTests/SFAzureClientTest.cs b/Snowflake.Data.Tests/UnitTests/SFAzureClientTest.cs
@@ -22,6 +22,8 @@ namespace Snowflake.Data.Tests.UnitTests
     using Azure.Storage.Blobs.Models;
 
     [TestFixture]
+    [IgnoreOnEnvIs("snowflake_cloud_env",
+        new string[] { "GCP", "AWS" })]
     class SFAzureClientTest : SFBaseTest
     {
         // Mock data for file metadata
@@ -63,7 +65,7 @@ class SFAzureClientTest : SFBaseTest
         public void BeforeTest()
         {
             t_downloadFileName = TestNameWithWorker + "_mockFileName.txt";
-            
+
             _fileMetadata = new SFFileMetadata()
             {
                 stageInfo = new PutGetStageInfo()

diff --git a/Snowflake.Data.Tests/UnitTests/SFGCSClientTest.cs b/Snowflake.Data.Tests/UnitTests/SFGCSClientTest.cs
@@ -19,6 +19,8 @@ namespace Snowflake.Data.Tests.UnitTests
     using Moq;
 
     [TestFixture]
+    [IgnoreOnEnvIs("snowflake_cloud_env",
+        new string[] { "AWS", "AZURE" })]
     class SFGCSClientTest : SFBaseTest
     {
         // Mock data for file metadata
@@ -56,7 +58,7 @@ class SFGCSClientTest : SFBaseTest
         public void BeforeTest()
         {
             t_downloadFileName = TestNameWithWorker + "_mockFileName.txt";
-            
+
             _fileMetadata = new SFFileMetadata()
             {
                 stageInfo = new PutGetStageInfo()

diff --git a/Snowflake.Data.Tests/UnitTests/SFRemoteStorageClientTest.cs b/Snowflake.Data.Tests/UnitTests/SFRemoteStorageClientTest.cs
@@ -115,6 +115,7 @@ public void BeforeTest()
             };
 
             // Set the mock GCS client to use
+            // TODO: ISSUE WITH SPLIT ALWAYS USING GCP SPECIFIC CLIENT IN TESTS
             _client = new SFGCSClient(_fileMetadata.stageInfo);
             _fileMetadata.client = _client;
 

diff --git a/Snowflake.Data.Tests/UnitTests/SFS3ClientTest.cs b/Snowflake.Data.Tests/UnitTests/SFS3ClientTest.cs
@@ -22,6 +22,8 @@ namespace Snowflake.Data.Tests.UnitTests
     using Amazon.S3.Model;
 
     [TestFixture]
+    [IgnoreOnEnvIs("snowflake_cloud_env",
+        new string[] { "GCP", "AZURE" })]
     class SFS3ClientTest : SFBaseTest
     {
         // Mock data for file metadata
@@ -81,7 +83,7 @@ class SFS3ClientTest : SFBaseTest
         public void BeforeTest()
         {
             t_downloadFileName = TestNameWithWorker + "_mockFileName.txt";
-            
+
             _fileMetadata = new SFFileMetadata()
             {
                 stageInfo = new PutGetStageInfo()
@@ -274,7 +276,7 @@ public async Task TestUploadFileAsync(string requestKey, ResultStatus expectedRe
                 iv = MockS3Client.AmzIV,
                 key = MockS3Client.AmzKey,
                 matDesc = MockS3Client.AmzMatdesc
-            }, 
+            },
             _cancellationToken).ConfigureAwait(false);
 
             // Assert
@@ -331,7 +333,7 @@ public async Task TestDownloadFileAsync(string requestKey, ResultStatus expected
             _client = new SFS3Client(_fileMetadata.stageInfo, MaxRetry, Parallel, _proxyCredentials, mockAmazonS3Client.Object);
             _fileMetadata.client = _client;
             _fileMetadata.stageInfo.location = requestKey;
-            
+
             // Act
             await _client.DownloadFileAsync(_fileMetadata, t_downloadFileName, Parallel, _cancellationToken).ConfigureAwait(false);