Skip to content

Commit 995eb63

Browse files
author
Hovsep Mkrtchyan
committed
Build/run fixes in CLU.
1 parent 4dd42c3 commit 995eb63

File tree

3 files changed

+23
-20
lines changed

3 files changed

+23
-20
lines changed

src/CLU/Microsoft.Azure.Commands.Resources/Models.ActiveDirectory/ActiveDirectoryClient.cs

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
using Microsoft.Azure.Graph.RBAC;
1818
using Microsoft.Azure.Graph.RBAC.Models;
1919
using Microsoft.Rest.Azure;
20+
using Microsoft.Rest.Azure.OData;
2021
using System;
2122
using System.Collections.Generic;
2223
using System.Diagnostics;
@@ -39,9 +40,10 @@ public ActiveDirectoryClient(IAuthenticationFactory authenticationFactory, IClie
3940
{
4041
AccessTokenCredential creds = (AccessTokenCredential)authenticationFactory.GetSubscriptionCloudCredentials(context);
4142
GraphClient = clientFactory.CreateCustomArmClient<GraphRbacManagementClient>(
42-
creds.TenantID,
43-
creds,
44-
context.Environment.GetEndpointAsUri(AzureEnvironment.Endpoint.Graph));
43+
context.Environment.GetEndpointAsUri(AzureEnvironment.Endpoint.Graph),
44+
creds);
45+
GraphClient.TenantID = creds.TenantID;
46+
GraphClient.SubscriptionId = creds.SubscriptionId;
4547
}
4648

4749
public PSADObject GetADObject(ADObjectFilterOptions options)
@@ -101,7 +103,7 @@ public List<PSADServicePrincipal> FilterServicePrincipals(ADObjectFilterOptions
101103
try
102104
{
103105
servicePrincipal = GraphClient.ServicePrincipal
104-
.List(item => item.ServicePrincipalNames.Contains(options.SPN))
106+
.List(new ODataQuery<ServicePrincipal>(item => item.ServicePrincipalNames.Contains(options.SPN)))
105107
.FirstOrDefault();
106108
}
107109
catch { /* The user does not exist, ignore the exception. */ }
@@ -120,7 +122,7 @@ public List<PSADServicePrincipal> FilterServicePrincipals(ADObjectFilterOptions
120122
if (string.IsNullOrEmpty(options.NextLink))
121123
{
122124
result = GraphClient.ServicePrincipal
123-
.List(item => item.DisplayName.StartsWith(options.SearchString) );
125+
.List(new ODataQuery<ServicePrincipal>(item => item.DisplayName.StartsWith(options.SearchString)));
124126
}
125127
else
126128
{
@@ -132,7 +134,7 @@ public List<PSADServicePrincipal> FilterServicePrincipals(ADObjectFilterOptions
132134
}
133135
else
134136
{
135-
result = GraphClient.ServicePrincipal.List(item => item.DisplayName.StartsWith(options.SearchString));
137+
result = GraphClient.ServicePrincipal.List(new ODataQuery<ServicePrincipal>(item => item.DisplayName.StartsWith(options.SearchString)));
136138
servicePrincipals.AddRange(result.Select(u => u.ToPSADServicePrincipal()));
137139

138140
while (!string.IsNullOrEmpty(result.NextPageLink))
@@ -175,7 +177,7 @@ public List<PSADUser> FilterUsers(ADObjectFilterOptions options)
175177
try
176178
{
177179
user = GraphClient.User
178-
.List(item => item.SignInName == (Normalize(options.Mail) ?? Normalize(options.SignInName)))
180+
.List(new ODataQuery<User>(item => item.SignInName == (Normalize(options.Mail) ?? Normalize(options.SignInName))))
179181
.FirstOrDefault();
180182
}
181183
catch { /* The user does not exist, ignore the exception. */ }
@@ -191,7 +193,7 @@ public List<PSADUser> FilterUsers(ADObjectFilterOptions options)
191193
{
192194
if (string.IsNullOrEmpty(options.NextLink))
193195
{
194-
result = GraphClient.User.List(item => item.DisplayName.StartsWith(options.SearchString));
196+
result = GraphClient.User.List(new ODataQuery<User>(item => item.DisplayName.StartsWith(options.SearchString)));
195197
}
196198
else
197199
{
@@ -203,7 +205,7 @@ public List<PSADUser> FilterUsers(ADObjectFilterOptions options)
203205
}
204206
else
205207
{
206-
result = GraphClient.User.List(item => item.DisplayName.StartsWith(options.SearchString));
208+
result = GraphClient.User.List(new ODataQuery<User>(item => item.DisplayName.StartsWith(options.SearchString)));
207209
users.AddRange(result.Select(u => u.ToPSADUser()));
208210

209211
while (!string.IsNullOrEmpty(result.NextPageLink))
@@ -273,8 +275,8 @@ public List<PSADGroup> FilterGroups(ADObjectFilterOptions options)
273275
{
274276
if (string.IsNullOrEmpty(options.NextLink))
275277
{
276-
result = GraphClient.Group.List(
277-
item => item.Mail == options.Mail && item.DisplayName.StartsWith(options.SearchString));
278+
result = GraphClient.Group.List(new ODataQuery<ADGroup>(
279+
item => item.Mail == options.Mail && item.DisplayName.StartsWith(options.SearchString)));
278280
}
279281
else
280282
{
@@ -286,8 +288,8 @@ public List<PSADGroup> FilterGroups(ADObjectFilterOptions options)
286288
}
287289
else
288290
{
289-
result = GraphClient.Group.List(
290-
item => item.Mail == options.Mail && item.DisplayName.StartsWith(options.SearchString));
291+
result = GraphClient.Group.List(new ODataQuery<ADGroup>(
292+
item => item.Mail == options.Mail && item.DisplayName.StartsWith(options.SearchString)));
291293
groups.AddRange(result.Select(g => g.ToPSADGroup()));
292294

293295
while (!string.IsNullOrEmpty(result.NextPageLink))

src/CLU/Microsoft.Azure.Commands.Resources/Models.Authorization/AuthorizationClient.cs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ static AuthorizationClient()
6161
public AuthorizationClient(IAuthenticationFactory authenticationFactory, IClientFactory clientFactory, AzureContext context)
6262
{
6363
ActiveDirectoryClient = new ActiveDirectoryClient(authenticationFactory, clientFactory, context);
64-
AuthorizationManagementClient = clientFactory.CreateClient<AuthorizationManagementClient>(context, AzureEnvironment.Endpoint.ResourceManager);
64+
AuthorizationManagementClient = clientFactory.CreateArmClient<AuthorizationManagementClient>(context, AzureEnvironment.Endpoint.ResourceManager);
6565
}
6666

6767
/// <summary>
@@ -80,7 +80,7 @@ public PSRoleDefinition GetRoleDefinition(string roleId)
8080
public PSRoleDefinition GetRoleDefinition(Guid roleId)
8181
{
8282
return AuthorizationManagementClient.RoleDefinitions.Get(
83-
"subscription/" + AuthorizationManagementClient.SubscriptionId,
83+
"subscriptions/" + AuthorizationManagementClient.SubscriptionId,
8484
roleId.ToString()).ToPSRoleDefinition();
8585
}
8686

@@ -94,7 +94,7 @@ public List<PSRoleDefinition> FilterRoleDefinitions(string name)
9494
{
9595
List<PSRoleDefinition> result = new List<PSRoleDefinition>();
9696
result.AddRange(AuthorizationManagementClient.RoleDefinitions.List(
97-
"subscription/" + AuthorizationManagementClient.SubscriptionId,
97+
"subscriptions/" + AuthorizationManagementClient.SubscriptionId,
9898
new ODataQuery<RoleDefinition>( item => item.Name == name))
9999
.Select(r => r.ToPSRoleDefinition()));
100100

@@ -109,7 +109,7 @@ public List<PSRoleDefinition> GetRoleDefinitions()
109109
{
110110
List<PSRoleDefinition> result = new List<PSRoleDefinition>();
111111
result.AddRange(AuthorizationManagementClient.RoleDefinitions.List(
112-
"subscription/" + AuthorizationManagementClient.SubscriptionId)
112+
"subscriptions/" + AuthorizationManagementClient.SubscriptionId)
113113
.Select(r => r.ToPSRoleDefinition()));
114114
return result;
115115
}
@@ -122,7 +122,7 @@ public List<PSRoleDefinition> FilterRoleDefinitionsByCustom()
122122
{
123123
List<PSRoleDefinition> result = new List<PSRoleDefinition>();
124124
result.AddRange(AuthorizationManagementClient.RoleDefinitions.List(
125-
"subscription/" + AuthorizationManagementClient.SubscriptionId)
125+
"subscriptions/" + AuthorizationManagementClient.SubscriptionId)
126126
.Where(r => r.Properties.Type == AuthorizationClientExtensions.CustomRole)
127127
.Select(r => r.ToPSRoleDefinition()));
128128
return result;

src/CLU/Microsoft.Azure.Commands.Resources/ResourceGroupDeployments/TestAzureResourceGroupDeploymentCommand.cs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,8 @@ namespace Microsoft.Azure.Commands.Resources.ResourceGroupDeployments
2323
/// <summary>
2424
/// Validate a template to see whether it's using the right syntax, resource providers, resource types, etc.
2525
/// </summary>
26-
[Cmdlet(VerbsDiagnostic.Test, "AzureRmResourceGroupDeployment", DefaultParameterSetName = ParameterlessTemplateFileParameterSetName), OutputType(typeof(List<PSResourceManagerError>))]
26+
[Cmdlet(VerbsDiagnostic.Test, "AzureRmResourceGroupDeployment", DefaultParameterSetName = ParameterlessTemplateFileParameterSetName),
27+
OutputType(typeof(PSResourceManagerError))]
2728
public class TestAzureResourceGroupDeploymentCommand : ResourceWithParameterBaseCmdlet, IDynamicParameters
2829
{
2930
[Parameter(Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "The resource group name.")]
@@ -49,7 +50,7 @@ protected override void ProcessRecord()
4950
ParameterUri = TemplateParameterUri
5051
};
5152

52-
WriteObject(ResourcesClient.ValidatePSResourceGroupDeployment(parameters, Mode));
53+
WriteObject(ResourcesClient.ValidatePSResourceGroupDeployment(parameters, Mode), enumerateCollection: true);
5354
}
5455
}
5556
}

0 commit comments

Comments
 (0)