@@ -185,6 +185,8 @@ private void SetupManagementClients(RestTestFramework.MockContext context)
185
185
NetworkManagementClient = this . GetNetworkManagementClientClient ( context ) ;
186
186
ComputeManagementClient = GetComputeManagementClient ( context ) ;
187
187
AuthorizationManagementClient = GetAuthorizationManagementClient ( ) ;
188
+ var networkManagementClientInternal = this . GetNetworkManagementClientInternal ( context ) ;
189
+ var resourceManagementClientInternal = this . GetResourceManagementClientInternal ( context ) ;
188
190
// GraphClient = GetGraphClient();
189
191
190
192
helper . SetupManagementClients (
@@ -195,7 +197,9 @@ private void SetupManagementClients(RestTestFramework.MockContext context)
195
197
//eventsClient,
196
198
NetworkManagementClient ,
197
199
ComputeManagementClient ,
198
- AuthorizationManagementClient ) ;
200
+ AuthorizationManagementClient ,
201
+ networkManagementClientInternal ,
202
+ resourceManagementClientInternal ) ;
199
203
// GraphClient);
200
204
}
201
205
@@ -237,6 +241,13 @@ private ResourceManagementClient GetResourceManagementClient()
237
241
return TestBase . GetServiceClient < ResourceManagementClient > ( this . csmTestFactory ) ;
238
242
}
239
243
244
+ private Microsoft . Azure . Management . Internal . Resources . ResourceManagementClient GetResourceManagementClientInternal ( RestTestFramework . MockContext context )
245
+ {
246
+ return testViaCsm
247
+ ? context . GetServiceClient < Microsoft . Azure . Management . Internal . Resources . ResourceManagementClient > ( RestTestFramework . TestEnvironmentFactory . GetTestEnvironment ( ) )
248
+ : TestBase . GetServiceClient < Microsoft . Azure . Management . Internal . Resources . ResourceManagementClient > ( new RDFETestEnvironmentFactory ( ) ) ;
249
+ }
250
+
240
251
private SubscriptionClient GetSubscriptionClient ( )
241
252
{
242
253
return TestBase . GetServiceClient < SubscriptionClient > ( this . csmTestFactory ) ;
@@ -266,6 +277,13 @@ private NetworkManagementClient GetNetworkManagementClientClient(RestTestFramewo
266
277
: TestBase . GetServiceClient < NetworkManagementClient > ( new RDFETestEnvironmentFactory ( ) ) ;
267
278
}
268
279
280
+ private Microsoft . Azure . Management . Internal . Network . Version2017_10_01 . NetworkManagementClient GetNetworkManagementClientInternal ( RestTestFramework . MockContext context )
281
+ {
282
+ return testViaCsm
283
+ ? context . GetServiceClient < Microsoft . Azure . Management . Internal . Network . Version2017_10_01 . NetworkManagementClient > ( RestTestFramework . TestEnvironmentFactory . GetTestEnvironment ( ) )
284
+ : TestBase . GetServiceClient < Microsoft . Azure . Management . Internal . Network . Version2017_10_01 . NetworkManagementClient > ( new RDFETestEnvironmentFactory ( ) ) ;
285
+ }
286
+
269
287
private ComputeManagementClient GetComputeManagementClient ( RestTestFramework . MockContext context )
270
288
{
271
289
return testViaCsm
0 commit comments