@@ -61,6 +61,10 @@ public sealed class ComputeTestController : RMTestBase
61
61
62
62
public ComputeManagementClient ComputeManagementClient { get ; private set ; }
63
63
64
+ public Microsoft . Azure . Management . Internal . Network . Version2017_10_01 . NetworkManagementClient InternalNetworkManagementClient { get ; private set ; }
65
+
66
+ public Microsoft . Azure . Management . Internal . Resources . ResourceManagementClient InternalResourceManagementClient { get ; private set ; }
67
+
64
68
public string UserDomain { get ; private set ; }
65
69
66
70
public static ComputeTestController NewInstance
@@ -185,6 +189,8 @@ private void SetupManagementClients(RestTestFramework.MockContext context)
185
189
NetworkManagementClient = this . GetNetworkManagementClientClient ( context ) ;
186
190
ComputeManagementClient = GetComputeManagementClient ( context ) ;
187
191
AuthorizationManagementClient = GetAuthorizationManagementClient ( ) ;
192
+ InternalNetworkManagementClient = this . GetNetworkManagementClientInternal ( context ) ;
193
+ InternalResourceManagementClient = this . GetResourceManagementClientInternal ( context ) ;
188
194
// GraphClient = GetGraphClient();
189
195
190
196
helper . SetupManagementClients (
@@ -195,7 +201,9 @@ private void SetupManagementClients(RestTestFramework.MockContext context)
195
201
//eventsClient,
196
202
NetworkManagementClient ,
197
203
ComputeManagementClient ,
198
- AuthorizationManagementClient ) ;
204
+ AuthorizationManagementClient ,
205
+ InternalNetworkManagementClient ,
206
+ InternalResourceManagementClient ) ;
199
207
// GraphClient);
200
208
}
201
209
@@ -237,6 +245,13 @@ private ResourceManagementClient GetResourceManagementClient()
237
245
return TestBase . GetServiceClient < ResourceManagementClient > ( this . csmTestFactory ) ;
238
246
}
239
247
248
+ private Microsoft . Azure . Management . Internal . Resources . ResourceManagementClient GetResourceManagementClientInternal ( RestTestFramework . MockContext context )
249
+ {
250
+ return testViaCsm
251
+ ? context . GetServiceClient < Microsoft . Azure . Management . Internal . Resources . ResourceManagementClient > ( RestTestFramework . TestEnvironmentFactory . GetTestEnvironment ( ) )
252
+ : TestBase . GetServiceClient < Microsoft . Azure . Management . Internal . Resources . ResourceManagementClient > ( new RDFETestEnvironmentFactory ( ) ) ;
253
+ }
254
+
240
255
private SubscriptionClient GetSubscriptionClient ( )
241
256
{
242
257
return TestBase . GetServiceClient < SubscriptionClient > ( this . csmTestFactory ) ;
@@ -266,6 +281,13 @@ private NetworkManagementClient GetNetworkManagementClientClient(RestTestFramewo
266
281
: TestBase . GetServiceClient < NetworkManagementClient > ( new RDFETestEnvironmentFactory ( ) ) ;
267
282
}
268
283
284
+ private Microsoft . Azure . Management . Internal . Network . Version2017_10_01 . NetworkManagementClient GetNetworkManagementClientInternal ( RestTestFramework . MockContext context )
285
+ {
286
+ return testViaCsm
287
+ ? context . GetServiceClient < Microsoft . Azure . Management . Internal . Network . Version2017_10_01 . NetworkManagementClient > ( RestTestFramework . TestEnvironmentFactory . GetTestEnvironment ( ) )
288
+ : TestBase . GetServiceClient < Microsoft . Azure . Management . Internal . Network . Version2017_10_01 . NetworkManagementClient > ( new RDFETestEnvironmentFactory ( ) ) ;
289
+ }
290
+
269
291
private ComputeManagementClient GetComputeManagementClient ( RestTestFramework . MockContext context )
270
292
{
271
293
return testViaCsm
0 commit comments