diff --git a/src/ResourceManager/Sql/Commands.Sql.Test/ScenarioTests/SqlTestsBase.cs b/src/ResourceManager/Sql/Commands.Sql.Test/ScenarioTests/SqlTestsBase.cs index 656bb2b9815a..e22403cf56c6 100644 --- a/src/ResourceManager/Sql/Commands.Sql.Test/ScenarioTests/SqlTestsBase.cs +++ b/src/ResourceManager/Sql/Commands.Sql.Test/ScenarioTests/SqlTestsBase.cs @@ -64,22 +64,47 @@ protected void RunPowerShellTest(params string[] scripts) protected SqlManagementClient GetSqlClient() { - return TestBase.GetServiceClient(new CSMTestEnvironmentFactory()); + SqlManagementClient client = TestBase.GetServiceClient(new CSMTestEnvironmentFactory()); + if (HttpMockServer.Mode == HttpRecorderMode.Playback) + { + client.LongRunningOperationInitialTimeout = 0; + client.LongRunningOperationRetryTimeout = 0; + } + return client; } protected StorageManagementClient GetStorageClient() { - return TestBase.GetServiceClient(new RDFETestEnvironmentFactory()); + StorageManagementClient client = TestBase.GetServiceClient(new RDFETestEnvironmentFactory()); + if (HttpMockServer.Mode == HttpRecorderMode.Playback) + { + client.LongRunningOperationInitialTimeout = 0; + client.LongRunningOperationRetryTimeout = 0; + } + return client; } protected ResourceManagementClient GetResourcesClient() { - return TestBase.GetServiceClient(new CSMTestEnvironmentFactory()); + ResourceManagementClient client = TestBase.GetServiceClient(new CSMTestEnvironmentFactory()); + if (HttpMockServer.Mode == HttpRecorderMode.Playback) + { + client.LongRunningOperationInitialTimeout = 0; + client.LongRunningOperationRetryTimeout = 0; + } + return client; } private AuthorizationManagementClient GetAuthorizationManagementClient() - { - return TestBase.GetServiceClient(new CSMTestEnvironmentFactory()); + { + AuthorizationManagementClient client = TestBase.GetServiceClient(new CSMTestEnvironmentFactory()); + if (HttpMockServer.Mode == HttpRecorderMode.Playback) + { + client.LongRunningOperationInitialTimeout = 0; + client.LongRunningOperationRetryTimeout = 0; + } + return client; + } } }