Use the new configuration extension methods

dotnet · Mar 30, 2016 · ac67594 · ac67594
1 parent 7a3e125
commit ac67594
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 12 deletions.
diff --git a/test/Tester/GenericGrainsInAzureStorageTests.cs b/test/Tester/GenericGrainsInAzureStorageTests.cs
@@ -1,6 +1,6 @@
 using System;
-using System.Collections.Generic;
 using System.Threading.Tasks;
+using Orleans.Runtime.Configuration;
 using Orleans.TestingHost;
 using UnitTests.GrainInterfaces;
 using UnitTests.Tester;
@@ -21,11 +21,7 @@ protected override TestingSiloHost CreateClusterHost()
                     StartSecondary = false,
                     AdjustConfig = config =>
                     {
-                        const string myProviderFullTypeName = "Orleans.Storage.AzureTableStorage";
-                        const string myProviderName = "AzureStore";
-                        var properties = new Dictionary<string, string>();
-                        properties.Add("DataConnectionString", "UseDevelopmentStorage=true");
-                        config.Globals.RegisterStorageProvider(myProviderFullTypeName, myProviderName, properties);
+                        config.AddAzureTableStorageProvider("AzureStore", StorageTestConstants.DataConnectionString);
                     }
                 });
             }

diff --git a/test/Tester/StreamingTests/EHStreamProviderCheckpointTests.cs b/test/Tester/StreamingTests/EHStreamProviderCheckpointTests.cs
@@ -175,12 +175,7 @@ private static void AdjustConfig(ClusterConfiguration config)
         {
             // register stream provider
             config.Globals.RegisterStreamProvider<EventHubStreamProvider>(StreamProviderName, BuildProviderSettings());
-            config.Globals.RegisterStorageProvider<AzureTableStorage>(
-                ImplicitSubscription_RecoverableStream_CollectorGrain.StorageProviderName,
-                new Dictionary<string, string>
-                {
-                    {"DataConnectionString", StorageTestConstants.DataConnectionString}
-                });
+            config.AddAzureTableStorageProvider(ImplicitSubscription_RecoverableStream_CollectorGrain.StorageProviderName, StorageTestConstants.DataConnectionString);
 
             // Make sure a node config exist for each silo in the cluster.
             // This is required for the DynamicClusterConfigDeploymentBalancer to properly balance queues.