/// <summary> /// Transforms a data masking rule to its cmdlet model representation /// </summary> private DatabaseDataMaskingRuleModel ModelizeDatabaseDataMaskingRule(DataMaskingRule rule, string resourceGroup, string serverName, string databaseName) { DatabaseDataMaskingRuleModel dbRuleModel = new DatabaseDataMaskingRuleModel(); DataMaskingRuleProperties properties = rule.Properties; dbRuleModel.ResourceGroupName = resourceGroup; dbRuleModel.ServerName = serverName; dbRuleModel.DatabaseName = databaseName; dbRuleModel.ColumnName = properties.ColumnName; dbRuleModel.TableName = properties.TableName; dbRuleModel.SchemaName = properties.SchemaName; dbRuleModel.MaskingFunction = ModelizeMaskingFunction(properties.MaskingFunction); dbRuleModel.PrefixSize = ModelizeNullableUint(properties.PrefixSize); dbRuleModel.ReplacementString = properties.ReplacementString; dbRuleModel.SuffixSize = ModelizeNullableUint(properties.SuffixSize); dbRuleModel.NumberFrom = ModelizeNullableDouble(properties.NumberFrom); dbRuleModel.NumberTo = ModelizeNullableDouble(properties.NumberTo); return(dbRuleModel); }
public async Task <Response <DataMaskingRule> > CreateOrUpdateAsync(string subscriptionId, string resourceGroupName, string serverName, string databaseName, string dataMaskingRuleName, DataMaskingRule parameters, CancellationToken cancellationToken = default) { if (subscriptionId == null) { throw new ArgumentNullException(nameof(subscriptionId)); } if (resourceGroupName == null) { throw new ArgumentNullException(nameof(resourceGroupName)); } if (serverName == null) { throw new ArgumentNullException(nameof(serverName)); } if (databaseName == null) { throw new ArgumentNullException(nameof(databaseName)); } if (dataMaskingRuleName == null) { throw new ArgumentNullException(nameof(dataMaskingRuleName)); } if (parameters == null) { throw new ArgumentNullException(nameof(parameters)); } using var message = CreateCreateOrUpdateRequest(subscriptionId, resourceGroupName, serverName, databaseName, dataMaskingRuleName, parameters); await _pipeline.SendAsync(message, cancellationToken).ConfigureAwait(false); switch (message.Response.Status) { case 200: case 201: { DataMaskingRule value = default; using var document = await JsonDocument.ParseAsync(message.Response.ContentStream, default, cancellationToken).ConfigureAwait(false); value = DataMaskingRule.DeserializeDataMaskingRule(document.RootElement); return(Response.FromValue(value, message.Response)); }
public virtual Response <DataMaskingRule> CreateOrUpdateDataMaskingRule(string dataMaskingRuleName, DataMaskingRule parameters, CancellationToken cancellationToken = default) { Argument.AssertNotNullOrEmpty(dataMaskingRuleName, nameof(dataMaskingRuleName)); if (parameters == null) { throw new ArgumentNullException(nameof(parameters)); } using var scope = _dataMaskingRulesClientDiagnostics.CreateScope("DataMaskingPolicy.CreateOrUpdateDataMaskingRule"); scope.Start(); try { var response = _dataMaskingRulesRestClient.CreateOrUpdate(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, dataMaskingRuleName, parameters, cancellationToken); return(response); } catch (Exception e) { scope.Failed(e); throw; } }
public virtual async Task <Response <DataMaskingRule> > CreateOrUpdateDataMaskingRuleAsync(string dataMaskingRuleName, DataMaskingRule dataMaskingRule, CancellationToken cancellationToken = default) { Argument.AssertNotNullOrEmpty(dataMaskingRuleName, nameof(dataMaskingRuleName)); Argument.AssertNotNull(dataMaskingRule, nameof(dataMaskingRule)); using var scope = _dataMaskingRulesClientDiagnostics.CreateScope("DataMaskingPolicyResource.CreateOrUpdateDataMaskingRule"); scope.Start(); try { var response = await _dataMaskingRulesRestClient.CreateOrUpdateAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, dataMaskingRuleName, dataMaskingRule, cancellationToken).ConfigureAwait(false); return(response); } catch (Exception e) { scope.Failed(e); throw; } }
internal HttpMessage CreateCreateOrUpdateRequest(string subscriptionId, string resourceGroupName, string serverName, string databaseName, string dataMaskingRuleName, DataMaskingRule parameters) { var message = _pipeline.CreateMessage(); var request = message.Request; request.Method = RequestMethod.Put; var uri = new RawRequestUriBuilder(); uri.Reset(_endpoint); uri.AppendPath("/subscriptions/", false); uri.AppendPath(subscriptionId, true); uri.AppendPath("/resourceGroups/", false); uri.AppendPath(resourceGroupName, true); uri.AppendPath("/providers/Microsoft.Sql/servers/", false); uri.AppendPath(serverName, true); uri.AppendPath("/databases/", false); uri.AppendPath(databaseName, true); uri.AppendPath("/dataMaskingPolicies/", false); uri.AppendPath("Default", true); uri.AppendPath("/rules/", false); uri.AppendPath(dataMaskingRuleName, true); uri.AppendQuery("api-version", _apiVersion, true); request.Uri = uri; request.Headers.Add("Accept", "application/json"); request.Headers.Add("Content-Type", "application/json"); var content = new Utf8JsonRequestContent(); content.JsonWriter.WriteObjectValue(parameters); request.Content = content; _userAgent.Apply(message); return(message); }
public virtual Response <DataMaskingRule> CreateOrUpdate(string resourceGroupName, string serverName, string databaseName, string dataMaskingRuleName, DataMaskingRule parameters, CancellationToken cancellationToken = default) { using var scope = _clientDiagnostics.CreateScope("DataMaskingRulesOperations.CreateOrUpdate"); scope.Start(); try { return(RestClient.CreateOrUpdate(resourceGroupName, serverName, databaseName, dataMaskingRuleName, parameters, cancellationToken)); } catch (Exception e) { scope.Failed(e); throw; } }
/// <summary> /// Creates or updates a Sql pool data masking rule. /// </summary> /// <param name='resourceGroupName'> /// The name of the resource group. The name is case insensitive. /// </param> /// <param name='workspaceName'> /// The name of the workspace /// </param> /// <param name='sqlPoolName'> /// SQL pool name /// </param> /// <param name='dataMaskingRuleName'> /// The name of the data masking rule. /// </param> /// <param name='parameters'> /// The required parameters for creating or updating a data masking rule. /// </param> /// <param name='customHeaders'> /// Headers that will be added to request. /// </param> /// <param name='cancellationToken'> /// The cancellation token. /// </param> /// <exception cref="CloudException"> /// Thrown when the operation returned an invalid status code /// </exception> /// <exception cref="SerializationException"> /// Thrown when unable to deserialize the response /// </exception> /// <exception cref="ValidationException"> /// Thrown when a required parameter is null /// </exception> /// <exception cref="System.ArgumentNullException"> /// Thrown when a required parameter is null /// </exception> /// <return> /// A response object containing the response body and response headers. /// </return> public async Task <AzureOperationResponse <DataMaskingRule> > CreateOrUpdateWithHttpMessagesAsync(string resourceGroupName, string workspaceName, string sqlPoolName, string dataMaskingRuleName, DataMaskingRule parameters, Dictionary <string, List <string> > customHeaders = null, CancellationToken cancellationToken = default(CancellationToken)) { if (Client.ApiVersion == null) { throw new ValidationException(ValidationRules.CannotBeNull, "this.Client.ApiVersion"); } if (Client.ApiVersion != null) { if (Client.ApiVersion.Length < 1) { throw new ValidationException(ValidationRules.MinLength, "Client.ApiVersion", 1); } } if (Client.SubscriptionId == null) { throw new ValidationException(ValidationRules.CannotBeNull, "this.Client.SubscriptionId"); } if (Client.SubscriptionId != null) { if (Client.SubscriptionId.Length < 1) { throw new ValidationException(ValidationRules.MinLength, "Client.SubscriptionId", 1); } } if (resourceGroupName == null) { throw new ValidationException(ValidationRules.CannotBeNull, "resourceGroupName"); } if (resourceGroupName != null) { if (resourceGroupName.Length > 90) { throw new ValidationException(ValidationRules.MaxLength, "resourceGroupName", 90); } if (resourceGroupName.Length < 1) { throw new ValidationException(ValidationRules.MinLength, "resourceGroupName", 1); } if (!System.Text.RegularExpressions.Regex.IsMatch(resourceGroupName, "^[-\\w\\._\\(\\)]+$")) { throw new ValidationException(ValidationRules.Pattern, "resourceGroupName", "^[-\\w\\._\\(\\)]+$"); } } if (workspaceName == null) { throw new ValidationException(ValidationRules.CannotBeNull, "workspaceName"); } if (sqlPoolName == null) { throw new ValidationException(ValidationRules.CannotBeNull, "sqlPoolName"); } if (dataMaskingRuleName == null) { throw new ValidationException(ValidationRules.CannotBeNull, "dataMaskingRuleName"); } if (parameters == null) { throw new ValidationException(ValidationRules.CannotBeNull, "parameters"); } if (parameters != null) { parameters.Validate(); } string dataMaskingPolicyName = "Default"; // Tracing bool _shouldTrace = ServiceClientTracing.IsEnabled; string _invocationId = null; if (_shouldTrace) { _invocationId = ServiceClientTracing.NextInvocationId.ToString(); Dictionary <string, object> tracingParameters = new Dictionary <string, object>(); tracingParameters.Add("resourceGroupName", resourceGroupName); tracingParameters.Add("workspaceName", workspaceName); tracingParameters.Add("sqlPoolName", sqlPoolName); tracingParameters.Add("dataMaskingPolicyName", dataMaskingPolicyName); tracingParameters.Add("dataMaskingRuleName", dataMaskingRuleName); tracingParameters.Add("parameters", parameters); tracingParameters.Add("cancellationToken", cancellationToken); ServiceClientTracing.Enter(_invocationId, this, "CreateOrUpdate", tracingParameters); } // Construct URL var _baseUrl = Client.BaseUri.AbsoluteUri; var _url = new System.Uri(new System.Uri(_baseUrl + (_baseUrl.EndsWith("/") ? "" : "/")), "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Synapse/workspaces/{workspaceName}/sqlPools/{sqlPoolName}/dataMaskingPolicies/{dataMaskingPolicyName}/rules/{dataMaskingRuleName}").ToString(); _url = _url.Replace("{subscriptionId}", System.Uri.EscapeDataString(Client.SubscriptionId)); _url = _url.Replace("{resourceGroupName}", System.Uri.EscapeDataString(resourceGroupName)); _url = _url.Replace("{workspaceName}", System.Uri.EscapeDataString(workspaceName)); _url = _url.Replace("{sqlPoolName}", System.Uri.EscapeDataString(sqlPoolName)); _url = _url.Replace("{dataMaskingPolicyName}", System.Uri.EscapeDataString(dataMaskingPolicyName)); _url = _url.Replace("{dataMaskingRuleName}", System.Uri.EscapeDataString(dataMaskingRuleName)); List <string> _queryParameters = new List <string>(); if (Client.ApiVersion != null) { _queryParameters.Add(string.Format("api-version={0}", System.Uri.EscapeDataString(Client.ApiVersion))); } if (_queryParameters.Count > 0) { _url += (_url.Contains("?") ? "&" : "?") + string.Join("&", _queryParameters); } // Create HTTP transport objects var _httpRequest = new HttpRequestMessage(); HttpResponseMessage _httpResponse = null; _httpRequest.Method = new HttpMethod("PUT"); _httpRequest.RequestUri = new System.Uri(_url); // Set Headers if (Client.GenerateClientRequestId != null && Client.GenerateClientRequestId.Value) { _httpRequest.Headers.TryAddWithoutValidation("x-ms-client-request-id", System.Guid.NewGuid().ToString()); } if (Client.AcceptLanguage != null) { if (_httpRequest.Headers.Contains("accept-language")) { _httpRequest.Headers.Remove("accept-language"); } _httpRequest.Headers.TryAddWithoutValidation("accept-language", Client.AcceptLanguage); } if (customHeaders != null) { foreach (var _header in customHeaders) { if (_httpRequest.Headers.Contains(_header.Key)) { _httpRequest.Headers.Remove(_header.Key); } _httpRequest.Headers.TryAddWithoutValidation(_header.Key, _header.Value); } } // Serialize Request string _requestContent = null; if (parameters != null) { _requestContent = Rest.Serialization.SafeJsonConvert.SerializeObject(parameters, Client.SerializationSettings); _httpRequest.Content = new StringContent(_requestContent, System.Text.Encoding.UTF8); _httpRequest.Content.Headers.ContentType = System.Net.Http.Headers.MediaTypeHeaderValue.Parse("application/json; charset=utf-8"); } // Set Credentials if (Client.Credentials != null) { cancellationToken.ThrowIfCancellationRequested(); await Client.Credentials.ProcessHttpRequestAsync(_httpRequest, cancellationToken).ConfigureAwait(false); } // Send Request if (_shouldTrace) { ServiceClientTracing.SendRequest(_invocationId, _httpRequest); } cancellationToken.ThrowIfCancellationRequested(); _httpResponse = await Client.HttpClient.SendAsync(_httpRequest, cancellationToken).ConfigureAwait(false); if (_shouldTrace) { ServiceClientTracing.ReceiveResponse(_invocationId, _httpResponse); } HttpStatusCode _statusCode = _httpResponse.StatusCode; cancellationToken.ThrowIfCancellationRequested(); string _responseContent = null; if ((int)_statusCode != 200 && (int)_statusCode != 201) { var ex = new CloudException(string.Format("Operation returned an invalid status code '{0}'", _statusCode)); try { _responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false); CloudError _errorBody = Rest.Serialization.SafeJsonConvert.DeserializeObject <CloudError>(_responseContent, Client.DeserializationSettings); if (_errorBody != null) { ex = new CloudException(_errorBody.Message); ex.Body = _errorBody; } } catch (JsonException) { // Ignore the exception } ex.Request = new HttpRequestMessageWrapper(_httpRequest, _requestContent); ex.Response = new HttpResponseMessageWrapper(_httpResponse, _responseContent); if (_httpResponse.Headers.Contains("x-ms-request-id")) { ex.RequestId = _httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault(); } if (_shouldTrace) { ServiceClientTracing.Error(_invocationId, ex); } _httpRequest.Dispose(); if (_httpResponse != null) { _httpResponse.Dispose(); } throw ex; } // Create Result var _result = new AzureOperationResponse <DataMaskingRule>(); _result.Request = _httpRequest; _result.Response = _httpResponse; if (_httpResponse.Headers.Contains("x-ms-request-id")) { _result.RequestId = _httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault(); } // Deserialize Response if ((int)_statusCode == 200) { _responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false); try { _result.Body = Rest.Serialization.SafeJsonConvert.DeserializeObject <DataMaskingRule>(_responseContent, Client.DeserializationSettings); } catch (JsonException ex) { _httpRequest.Dispose(); if (_httpResponse != null) { _httpResponse.Dispose(); } throw new SerializationException("Unable to deserialize the response.", _responseContent, ex); } } // Deserialize Response if ((int)_statusCode == 201) { _responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false); try { _result.Body = Rest.Serialization.SafeJsonConvert.DeserializeObject <DataMaskingRule>(_responseContent, Client.DeserializationSettings); } catch (JsonException ex) { _httpRequest.Dispose(); if (_httpResponse != null) { _httpResponse.Dispose(); } throw new SerializationException("Unable to deserialize the response.", _responseContent, ex); } } if (_shouldTrace) { ServiceClientTracing.Exit(_invocationId, _result); } return(_result); }
/// <summary> /// Creates or updates a Sql pool data masking rule. /// </summary> /// <param name='operations'> /// The operations group for this extension method. /// </param> /// <param name='resourceGroupName'> /// The name of the resource group. The name is case insensitive. /// </param> /// <param name='workspaceName'> /// The name of the workspace /// </param> /// <param name='sqlPoolName'> /// SQL pool name /// </param> /// <param name='dataMaskingRuleName'> /// The name of the data masking rule. /// </param> /// <param name='parameters'> /// The required parameters for creating or updating a data masking rule. /// </param> /// <param name='cancellationToken'> /// The cancellation token. /// </param> public static async Task <DataMaskingRule> CreateOrUpdateAsync(this IDataMaskingRulesOperations operations, string resourceGroupName, string workspaceName, string sqlPoolName, string dataMaskingRuleName, DataMaskingRule parameters, CancellationToken cancellationToken = default(CancellationToken)) { using (var _result = await operations.CreateOrUpdateWithHttpMessagesAsync(resourceGroupName, workspaceName, sqlPoolName, dataMaskingRuleName, parameters, null, cancellationToken).ConfigureAwait(false)) { return(_result.Body); } }
/// <summary> /// Creates or updates a Sql pool data masking rule. /// </summary> /// <param name='operations'> /// The operations group for this extension method. /// </param> /// <param name='resourceGroupName'> /// The name of the resource group. The name is case insensitive. /// </param> /// <param name='workspaceName'> /// The name of the workspace /// </param> /// <param name='sqlPoolName'> /// SQL pool name /// </param> /// <param name='dataMaskingRuleName'> /// The name of the data masking rule. /// </param> /// <param name='parameters'> /// The required parameters for creating or updating a data masking rule. /// </param> public static DataMaskingRule CreateOrUpdate(this IDataMaskingRulesOperations operations, string resourceGroupName, string workspaceName, string sqlPoolName, string dataMaskingRuleName, DataMaskingRule parameters) { return(operations.CreateOrUpdateAsync(resourceGroupName, workspaceName, sqlPoolName, dataMaskingRuleName, parameters).GetAwaiter().GetResult()); }
/// <summary> /// The non-boilerplated test code of the APIs for managing the lifecycle data masking rules. /// It is meant to be called with a name of an already exisiting database (and therefore already existing server and resource group). /// This test does not create these resources and does not remove them. /// The flow is: /// 1) Create policy (it's a prereq) /// 2) Create rule1, validate its creation and its content by doing another GET call /// 3) Update rule1, validate the update by doing another GET call /// 4) Create rule2, validate its creation and its content /// 5) Get the list of rules, see that there are two and each one of them has the right content /// 6) Delete rule1, see that we get OK /// 8) List the rules, see that we now have one rule there and it is rule 2 /// </summary> /// <param name="sqlClient">The sqlClient</param> /// <param name="resourceGroupName">The resource group name to use in this test</param> /// <param name="server">The server to use in this test</param> /// <param name="database">The database to use in this test</param> private void TestDataMaskingRuleAPIs(SqlManagementClient sqlClient, string resourceGroupName, Server server, Database database) { DataMaskingPolicyCreateOrUpdateParameters policyParams = new DataMaskingPolicyCreateOrUpdateParameters(); policyParams.Properties = MakeDefaultDataMaskingPolicyProperties(); policyParams.Properties.DataMaskingState = "Enabled"; sqlClient.DataMasking.CreateOrUpdatePolicy(resourceGroupName, server.Name, database.Name, policyParams); int ruleCounter = 1; DataMaskingRuleCreateOrUpdateParameters ruleParams = new DataMaskingRuleCreateOrUpdateParameters(); string serverName = server.Properties.FullyQualifiedDomainName; string uid = server.Properties.AdministratorLogin; string pwd = server.Properties.AdministratorLoginPassword; string dbName = database.Name; string connString = string.Format("Server={0};uid={1}; pwd={2};Database={3};Integrated Security=False;", serverName, uid, pwd, dbName); var conn = new SqlConnection(); conn.ConnectionString = connString; string tableName = "table1", columnName = "column1"; string firewallRuleName = TestUtilities.GenerateName("all"); string startIp1 = "1.1.1.1"; string endIp1 = "255.255.255.255"; sqlClient.FirewallRules.CreateOrUpdate(resourceGroupName, serverName.Split('.').ElementAt(0), firewallRuleName, new FirewallRuleCreateOrUpdateParameters() { Properties = new FirewallRuleCreateOrUpdateProperties() { StartIpAddress = startIp1, EndIpAddress = endIp1, } }); CreateDatabaseContents(conn, tableName, columnName); Func <DataMaskingRuleCreateOrUpdateParameters, Func <DataMaskingRule, bool> > isRuleOnColumn = (DataMaskingRuleCreateOrUpdateParameters parms) => { return((DataMaskingRule r1) => { return parms.Properties.ColumnName == r1.Properties.ColumnName && parms.Properties.TableName == r1.Properties.TableName && parms.Properties.SchemaName == r1.Properties.SchemaName; }); }; ruleParams.Properties = MakeRuleProperties(ruleCounter++, tableName, columnName); string rule1Name = ruleParams.Properties.Id; var createRuleResponse = sqlClient.DataMasking.CreateOrUpdateRule(resourceGroupName, server.Name, database.Name, rule1Name, ruleParams); TestUtilities.ValidateOperationResponse(createRuleResponse, HttpStatusCode.OK); var listAfterCreateResponse = sqlClient.DataMasking.List(resourceGroupName, server.Name, database.Name); TestUtilities.ValidateOperationResponse(listAfterCreateResponse, HttpStatusCode.OK); Assert.Equal(1, listAfterCreateResponse.DataMaskingRules.Count); DataMaskingRule receivedRule = listAfterCreateResponse.DataMaskingRules.FirstOrDefault(isRuleOnColumn(ruleParams)); VerifyDataMaskingRuleInformation(receivedRule.Properties, ruleParams.Properties); // Modify the policy properties, send and receive, see it its still ok ruleParams.Properties.PrefixSize = "2"; ruleParams.Properties.ReplacementString = "ABC"; ruleParams.Properties.SuffixSize = "1"; var updateRuleResponse = sqlClient.DataMasking.CreateOrUpdateRule(resourceGroupName, server.Name, database.Name, rule1Name, ruleParams); TestUtilities.ValidateOperationResponse(updateRuleResponse, HttpStatusCode.OK); var listUpdateResponse = sqlClient.DataMasking.List(resourceGroupName, server.Name, database.Name); TestUtilities.ValidateOperationResponse(listUpdateResponse, HttpStatusCode.OK); Assert.Equal(1, listUpdateResponse.DataMaskingRules.Count); var updatedRule = listUpdateResponse.DataMaskingRules.FirstOrDefault(isRuleOnColumn(ruleParams)); VerifyDataMaskingRuleInformation(updatedRule.Properties, ruleParams.Properties); DataMaskingRuleCreateOrUpdateParameters ruleParams2 = new DataMaskingRuleCreateOrUpdateParameters(); tableName = "table2"; columnName = "column2"; CreateDatabaseContents(conn, tableName, columnName); ruleParams2.Properties = MakeRuleProperties(ruleCounter++, tableName, columnName); string rule2Name = ruleParams2.Properties.Id; var createSecondRuleResponse = sqlClient.DataMasking.CreateOrUpdateRule(resourceGroupName, server.Name, database.Name, rule2Name, ruleParams2); TestUtilities.ValidateOperationResponse(createSecondRuleResponse, HttpStatusCode.OK); DataMaskingRuleListResponse listAfterSecondCreateResponse = sqlClient.DataMasking.List(resourceGroupName, server.Name, database.Name); TestUtilities.ValidateOperationResponse(listAfterSecondCreateResponse, HttpStatusCode.OK); Assert.Equal(2, listAfterSecondCreateResponse.DataMaskingRules.Count); updatedRule = listUpdateResponse.DataMaskingRules.FirstOrDefault(isRuleOnColumn(ruleParams)); VerifyDataMaskingRuleInformation(updatedRule.Properties, ruleParams.Properties); var receivedSecondRule = listAfterSecondCreateResponse.DataMaskingRules.FirstOrDefault(isRuleOnColumn(ruleParams2)); VerifyDataMaskingRuleInformation(receivedSecondRule.Properties, ruleParams2.Properties); AzureOperationResponse deleteResponse = sqlClient.DataMasking.Delete(resourceGroupName, server.Name, database.Name, rule1Name); TestUtilities.ValidateOperationResponse(deleteResponse, HttpStatusCode.OK); DataMaskingRuleListResponse listAfterDeleteResponse = sqlClient.DataMasking.List(resourceGroupName, server.Name, database.Name); TestUtilities.ValidateOperationResponse(listAfterDeleteResponse, HttpStatusCode.OK); Assert.Equal(listAfterDeleteResponse.DataMaskingRules.Count, 1); var receivedAfterDelete = listAfterSecondCreateResponse.DataMaskingRules.FirstOrDefault(isRuleOnColumn(ruleParams2)); VerifyDataMaskingRuleInformation(receivedAfterDelete.Properties, ruleParams2.Properties); }
public async virtual Task <Response <DataMaskingRule> > CreateOrUpdateDataMaskingRuleAsync(string dataMaskingRuleName, DataMaskingRule parameters, CancellationToken cancellationToken = default) { if (dataMaskingRuleName == null) { throw new ArgumentNullException(nameof(dataMaskingRuleName)); } if (parameters == null) { throw new ArgumentNullException(nameof(parameters)); } using var scope = _clientDiagnostics.CreateScope("DataMaskingPolicy.CreateOrUpdateDataMaskingRule"); scope.Start(); try { var response = await _dataMaskingRulesRestClient.CreateOrUpdateAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, dataMaskingRuleName, parameters, cancellationToken).ConfigureAwait(false); return(response); } catch (Exception e) { scope.Failed(e); throw; } }
public void TestCreateUpdateGetDataMaskingRules() { string testPrefix = "sqldatamaskingcrudtest-"; using (SqlManagementTestContext context = new SqlManagementTestContext(this)) { ResourceGroup resourceGroup = context.CreateResourceGroup(); SqlManagementClient sqlClient = context.GetClient <SqlManagementClient>(); Server server = context.CreateServer(resourceGroup); // Create database // string dbName = SqlManagementTestUtilities.GenerateName(testPrefix); var db1 = sqlClient.Databases.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, new Database() { Location = server.Location, }); Assert.NotNull(db1); // Create server firewall rule sqlClient.FirewallRules.CreateOrUpdate(resourceGroup.Name, server.Name, "sqltestrule", new FirewallRule() { StartIpAddress = "0.0.0.0", EndIpAddress = "255.255.255.255" }); // Create test table with columns // This is not needed in playback because in playback, there is no actual database to execute against HttpRecorderMode testMode = HttpMockServer.GetCurrentMode(); if (testMode != HttpRecorderMode.Playback) { SqlConnectionStringBuilder builder = new SqlConnectionStringBuilder() { DataSource = string.Format(server.FullyQualifiedDomainName, server.Name), UserID = SqlManagementTestUtilities.DefaultLogin, Password = SqlManagementTestUtilities.DefaultPassword, InitialCatalog = dbName }; using (SqlConnection conn = new SqlConnection(builder.ConnectionString)) { conn.Open(); SqlCommand command = new SqlCommand("create table table1 (column1 int, column2 nvarchar(max))", conn); command.ExecuteNonQuery(); } } // Verify Policy is disabled to begin with DataMaskingPolicy policy = sqlClient.DataMaskingPolicies.Get(resourceGroup.Name, server.Name, dbName); Assert.Equal(DataMaskingState.Disabled, policy.DataMaskingState); // Create a Number data masking rule (enables the data masking policy) DataMaskingRule numberRule = new DataMaskingRule() { SchemaName = "dbo", TableName = "table1", ColumnName = "column1", MaskingFunction = DataMaskingFunction.Number, NumberFrom = "0", NumberTo = "10" }; // Create a Text data masking rule DataMaskingRule textRule = new DataMaskingRule() { SchemaName = "dbo", TableName = "table1", ColumnName = "column2", MaskingFunction = DataMaskingFunction.Text, PrefixSize = "1", SuffixSize = "1", ReplacementString = "teststring" }; // Not creating datamasking rule names because name is ignored when creating the rules anyway sqlClient.DataMaskingRules.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, "name", numberRule); sqlClient.DataMaskingRules.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, "name2", textRule); // Verify Policy is now enabled policy = sqlClient.DataMaskingPolicies.Get(resourceGroup.Name, server.Name, dbName); Assert.Equal(DataMaskingState.Enabled, policy.DataMaskingState); // List data masking rules IEnumerable <DataMaskingRule> rules = sqlClient.DataMaskingRules.ListByDatabase(resourceGroup.Name, server.Name, dbName); Assert.Equal(2, rules.Count()); // Verify number rule numberRule = rules.FirstOrDefault(rule => rule.MaskingFunction == DataMaskingFunction.Number); Assert.Equal("dbo", numberRule.SchemaName); Assert.Equal("table1", numberRule.TableName); Assert.Equal("column1", numberRule.ColumnName); Assert.Equal("0", numberRule.NumberFrom); Assert.Equal("10", numberRule.NumberTo); // Verify text rule textRule = rules.FirstOrDefault(rule => rule.MaskingFunction == DataMaskingFunction.Text); Assert.Equal("dbo", textRule.SchemaName); Assert.Equal("table1", textRule.TableName); Assert.Equal("column2", textRule.ColumnName); Assert.Equal("1", textRule.PrefixSize); Assert.Equal("1", textRule.SuffixSize); Assert.Equal("teststring", textRule.ReplacementString); // Delete one rule through PUT numberRule.RuleState = DataMaskingRuleState.Disabled; sqlClient.DataMaskingRules.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, "name", numberRule); // List data masking rules rules = sqlClient.DataMaskingRules.ListByDatabase(resourceGroup.Name, server.Name, dbName); Assert.Equal(1, rules.Count()); // Verify Policy now enabled policy = sqlClient.DataMaskingPolicies.Get(resourceGroup.Name, server.Name, dbName); Assert.Equal(DataMaskingState.Enabled, policy.DataMaskingState); // Disable data masking policy (this deletes data masking rules) sqlClient.DataMaskingPolicies.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, new DataMaskingPolicy() { DataMaskingState = DataMaskingState.Disabled }); // Verify policy is disabled policy = sqlClient.DataMaskingPolicies.Get(resourceGroup.Name, server.Name, dbName); Assert.Equal(DataMaskingState.Disabled, policy.DataMaskingState); // Verify no rules are returned rules = sqlClient.DataMaskingRules.ListByDatabase(resourceGroup.Name, server.Name, dbName); Assert.Equal(0, rules.Count()); }; }