public async virtual Task <PolicyAssignmentCreateOrUpdateOperation> CreateOrUpdateAsync(bool waitForCompletion, string policyAssignmentName, PolicyAssignmentData parameters, CancellationToken cancellationToken = default) { Argument.AssertNotNullOrEmpty(policyAssignmentName, nameof(policyAssignmentName)); if (parameters == null) { throw new ArgumentNullException(nameof(parameters)); } using var scope = _policyAssignmentClientDiagnostics.CreateScope("PolicyAssignmentCollection.CreateOrUpdate"); scope.Start(); try { var response = await _policyAssignmentRestClient.CreateAsync(Id, policyAssignmentName, parameters, cancellationToken).ConfigureAwait(false); var operation = new PolicyAssignmentCreateOrUpdateOperation(ArmClient, response); if (waitForCompletion) { await operation.WaitForCompletionAsync(cancellationToken).ConfigureAwait(false); } return(operation); } catch (Exception e) { scope.Failed(e); throw; } }
protected async Task <PolicyAssignment> CreatePolicyAssignment(ArmResource armResource, string policyAssignmentName, string PolicyDefinitionId = "/providers/Microsoft.Authorization/policyDefinitions/06a78e20-9358-41c9-923c-fb736d382a4d") { PolicyAssignmentData input = new PolicyAssignmentData { DisplayName = $"Test ${policyAssignmentName}", PolicyDefinitionId = PolicyDefinitionId }; PolicyAssignmentCreateOrUpdateOperation lro = await armResource.GetPolicyAssignments().CreateOrUpdateAsync(true, policyAssignmentName, input); return(lro.Value); }