Skip to content

Commit 0740748

Browse files
committed
Merge pull request #503 from yoavrubin/dev
Timeouts in playback mode to be 0
2 parents 533dfea + 00e1b93 commit 0740748

File tree

1 file changed

+30
-5
lines changed

1 file changed

+30
-5
lines changed

src/ResourceManager/Sql/Commands.Sql.Test/ScenarioTests/SqlTestsBase.cs

Lines changed: 30 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -64,22 +64,47 @@ protected void RunPowerShellTest(params string[] scripts)
6464

6565
protected SqlManagementClient GetSqlClient()
6666
{
67-
return TestBase.GetServiceClient<SqlManagementClient>(new CSMTestEnvironmentFactory());
67+
SqlManagementClient client = TestBase.GetServiceClient<SqlManagementClient>(new CSMTestEnvironmentFactory());
68+
if (HttpMockServer.Mode == HttpRecorderMode.Playback)
69+
{
70+
client.LongRunningOperationInitialTimeout = 0;
71+
client.LongRunningOperationRetryTimeout = 0;
72+
}
73+
return client;
6874
}
6975

7076
protected StorageManagementClient GetStorageClient()
7177
{
72-
return TestBase.GetServiceClient<StorageManagementClient>(new RDFETestEnvironmentFactory());
78+
StorageManagementClient client = TestBase.GetServiceClient<StorageManagementClient>(new RDFETestEnvironmentFactory());
79+
if (HttpMockServer.Mode == HttpRecorderMode.Playback)
80+
{
81+
client.LongRunningOperationInitialTimeout = 0;
82+
client.LongRunningOperationRetryTimeout = 0;
83+
}
84+
return client;
7385
}
7486

7587
protected ResourceManagementClient GetResourcesClient()
7688
{
77-
return TestBase.GetServiceClient<ResourceManagementClient>(new CSMTestEnvironmentFactory());
89+
ResourceManagementClient client = TestBase.GetServiceClient<ResourceManagementClient>(new CSMTestEnvironmentFactory());
90+
if (HttpMockServer.Mode == HttpRecorderMode.Playback)
91+
{
92+
client.LongRunningOperationInitialTimeout = 0;
93+
client.LongRunningOperationRetryTimeout = 0;
94+
}
95+
return client;
7896
}
7997

8098
private AuthorizationManagementClient GetAuthorizationManagementClient()
81-
{
82-
return TestBase.GetServiceClient<AuthorizationManagementClient>(new CSMTestEnvironmentFactory());
99+
{
100+
AuthorizationManagementClient client = TestBase.GetServiceClient<AuthorizationManagementClient>(new CSMTestEnvironmentFactory());
101+
if (HttpMockServer.Mode == HttpRecorderMode.Playback)
102+
{
103+
client.LongRunningOperationInitialTimeout = 0;
104+
client.LongRunningOperationRetryTimeout = 0;
105+
}
106+
return client;
107+
83108
}
84109
}
85110
}

0 commit comments

Comments
 (0)