Tweak cluster conditions

Author: ajcvickers

tests/MongoDB.Driver.Examples/ClientSideEncryption2Examples.cs

@@ -37,7 +37,7 @@ public class ClientSideEncryption2Examples
         [Fact]
         public void FLE2AutomaticEncryption()
         {
-            RequireServer.Check().Supports(Feature.Csfle2).ClusterTypes(ClusterType.ReplicaSet, ClusterType.Sharded, ClusterType.LoadBalanced, ClusterType.Standalone);
+            RequireServer.Check().Supports(Feature.Csfle2);
 
             var unencryptedClient = DriverTestConfiguration.Client;
 

tests/MongoDB.Driver.Examples/TransactionExamplesForDocs/WithTransactionExample1.cs

@@ -16,7 +16,6 @@
 using System.Threading;
 using FluentAssertions;
 using MongoDB.Bson;
-using MongoDB.Driver.Core.Clusters;
 using MongoDB.Driver.Core.Misc;
 using MongoDB.Driver.Core.TestHelpers.XunitExtensions;
 using Xunit;
@@ -29,7 +28,7 @@ public class WithTransactionExample1
         [Fact]
         public void Example1()
         {
-            RequireServer.Check().ClusterTypes(ClusterType.ReplicaSet, ClusterType.Sharded, ClusterType.Standalone).Supports(Feature.Transactions);
+            RequireServer.Check().Supports(Feature.Transactions);
 
             var connectionString = CoreTestConfiguration.ConnectionString.ToString();
             DropCollections(

tests/MongoDB.Driver.TestHelpers/Core/XunitExtensions/RequireServer.cs

@@ -171,9 +171,18 @@ public RequireServer Supports(params Feature[] features)
         }
 
         public RequireServer SupportsCausalConsistency()
-        {
-            return ClusterTypes(Clusters.ClusterType.Sharded, Clusters.ClusterType.ReplicaSet, Clusters.ClusterType.Standalone).SupportsSessions();
-        }
+            // Determine if we are running against Atlas local by determining if the Atlas connection
+            // string is the same as the regular connection string when the cluster is standalone.
+            // This means that if someone has setup Atlas specific tests locally, then these tests will
+            // also run, otherwise we will skip them on a standalone cluster.
+            => CoreTestConfiguration.Cluster.Description.Type == Clusters.ClusterType.Standalone
+               && string.Equals(
+                   CoreTestConfiguration.ConnectionString.ToString(),
+                   Environment.GetEnvironmentVariable("ATLAS_SEARCH_URI"),
+                   StringComparison.Ordinal)
+                ? this
+                // If the server is not standalone, then use the previous logic.
+                : ClusterTypes(Clusters.ClusterType.Sharded, Clusters.ClusterType.ReplicaSet).SupportsSessions();
 
         public RequireServer SupportsSessions()
         {