23
23
using Microsoft . WindowsAzure ;
24
24
using Microsoft . WindowsAzure . Commands . ScenarioTest ;
25
25
using Microsoft . WindowsAzure . Commands . Utilities . Common ;
26
+ using Microsoft . WindowsAzure . Management . Scheduler ;
26
27
using Microsoft . Azure . Management . RecoveryServices ;
27
28
using Microsoft . Azure . Management . SiteRecovery ;
28
29
using Microsoft . Azure . Test ;
@@ -41,6 +42,7 @@ public abstract class SiteRecoveryTestsBase
41
42
42
43
public SiteRecoveryManagementClient SiteRecoveryMgmtClient { get ; private set ; }
43
44
public RecoveryServicesManagementClient RecoveryServicesMgmtClient { get ; private set ; }
45
+ public CloudServiceManagementClient CloudServiceManagementClient { get ; private set ; }
44
46
45
47
protected SiteRecoveryTestsBase ( )
46
48
{
@@ -82,10 +84,11 @@ protected SiteRecoveryTestsBase()
82
84
83
85
protected void SetupManagementClients ( )
84
86
{
87
+ CloudServiceManagementClient = GetCloudServicesManagementClient ( ) ;
85
88
RecoveryServicesMgmtClient = GetRecoveryServicesManagementClient ( ) ;
86
89
SiteRecoveryMgmtClient = GetSiteRecoveryManagementClient ( ) ;
87
90
88
- helper . SetupManagementClients ( RecoveryServicesMgmtClient , SiteRecoveryMgmtClient ) ;
91
+ helper . SetupManagementClients ( CloudServiceManagementClient , RecoveryServicesMgmtClient , SiteRecoveryMgmtClient ) ;
89
92
}
90
93
91
94
protected void RunPowerShellTest ( params string [ ] scripts )
@@ -98,17 +101,25 @@ protected void RunPowerShellTest(params string[] scripts)
98
101
99
102
SetupManagementClients ( ) ;
100
103
101
- helper . SetupEnvironment ( AzureModule . AzureServiceManagement ) ;
104
+ helper . SetupEnvironment ( AzureModule . AzureResourceManager ) ;
102
105
helper . SetupModules ( AzureModule . AzureResourceManager ,
103
106
"ScenarioTests\\ " + this . GetType ( ) . Name + ".ps1" ) ;
104
107
105
108
helper . RunPowerShellTest ( scripts ) ;
106
109
}
107
110
}
108
111
112
+ private CloudServiceManagementClient GetCloudServicesManagementClient ( )
113
+ {
114
+ return TestBase . GetServiceClient < CloudServiceManagementClient > ( this . armTestFactory ) ;
115
+ }
116
+
109
117
private RecoveryServicesManagementClient GetRecoveryServicesManagementClient ( )
110
118
{
111
- return TestBase . GetServiceClient < RecoveryServicesManagementClient > ( this . armTestFactory ) ;
119
+ return new RecoveryServicesManagementClient (
120
+ "Microsoft.SiteRecovery" ,
121
+ CloudServiceManagementClient . Credentials ,
122
+ CloudServiceManagementClient . BaseUri ) . WithHandler ( HttpMockServer . CreateInstance ( ) ) ;
112
123
}
113
124
114
125
private SiteRecoveryManagementClient GetSiteRecoveryManagementClient ( )
@@ -122,11 +133,11 @@ private SiteRecoveryManagementClient GetSiteRecoveryManagementClient()
122
133
}
123
134
124
135
return new SiteRecoveryManagementClient (
125
- asrVaultCreds . ResourceGroupName ,
126
136
asrVaultCreds . ResourceName ,
127
137
asrVaultCreds . ResourceGroupName ,
128
- RecoveryServicesMgmtClient . Credentials ,
129
- RecoveryServicesMgmtClient . BaseUri ) . WithHandler ( HttpMockServer . CreateInstance ( ) ) ;
138
+ "Microsoft.SiteRecovery" ,
139
+ CloudServiceManagementClient . Credentials ,
140
+ CloudServiceManagementClient . BaseUri ) . WithHandler ( HttpMockServer . CreateInstance ( ) ) ;
130
141
}
131
142
132
143
private static bool IgnoreCertificateErrorHandler
0 commit comments