azure_mgmt_security 0.18.2 → 0.19.0

data/lib/2020-07-01-preview/generated/azure_mgmt_security/models/va_rule.rb

@@ -0,0 +1,146 @@
+# encoding: utf-8
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+
+module Azure::Security::Mgmt::V2020_07_01_preview
+  module Models
+    #
+    # vulnerability assessment rule metadata details.
+    #
+    class VaRule
+
+      include MsRestAzure
+
+      # @return [String] The rule Id.
+      attr_accessor :rule_id
+
+      # @return [RuleSeverity] Possible values include: 'High', 'Medium',
+      # 'Low', 'Informational', 'Obsolete'
+      attr_accessor :severity
+
+      # @return [String] The rule category.
+      attr_accessor :category
+
+      # @return [RuleType] Possible values include: 'Binary',
+      # 'BaselineExpected', 'PositiveList', 'NegativeList'
+      attr_accessor :rule_type
+
+      # @return [String] The rule title.
+      attr_accessor :title
+
+      # @return [String] The rule description.
+      attr_accessor :description
+
+      # @return [String] The rule rationale.
+      attr_accessor :rationale
+
+      # @return [QueryCheck]
+      attr_accessor :query_check
+
+      # @return [Array<BenchmarkReference>] The benchmark references.
+      attr_accessor :benchmark_references
+
+
+      #
+      # Mapper for VaRule class as Ruby Hash.
+      # This will be used for serialization/deserialization.
+      #
+      def self.mapper()
+        {
+          client_side_validation: true,
+          required: false,
+          serialized_name: 'VaRule',
+          type: {
+            name: 'Composite',
+            class_name: 'VaRule',
+            model_properties: {
+              rule_id: {
+                client_side_validation: true,
+                required: false,
+                serialized_name: 'ruleId',
+                type: {
+                  name: 'String'
+                }
+              },
+              severity: {
+                client_side_validation: true,
+                required: false,
+                serialized_name: 'severity',
+                type: {
+                  name: 'String'
+                }
+              },
+              category: {
+                client_side_validation: true,
+                required: false,
+                serialized_name: 'category',
+                type: {
+                  name: 'String'
+                }
+              },
+              rule_type: {
+                client_side_validation: true,
+                required: false,
+                serialized_name: 'ruleType',
+                type: {
+                  name: 'String'
+                }
+              },
+              title: {
+                client_side_validation: true,
+                required: false,
+                serialized_name: 'title',
+                type: {
+                  name: 'String'
+                }
+              },
+              description: {
+                client_side_validation: true,
+                required: false,
+                serialized_name: 'description',
+                type: {
+                  name: 'String'
+                }
+              },
+              rationale: {
+                client_side_validation: true,
+                required: false,
+                serialized_name: 'rationale',
+                type: {
+                  name: 'String'
+                }
+              },
+              query_check: {
+                client_side_validation: true,
+                required: false,
+                serialized_name: 'queryCheck',
+                type: {
+                  name: 'Composite',
+                  class_name: 'QueryCheck'
+                }
+              },
+              benchmark_references: {
+                client_side_validation: true,
+                required: false,
+                serialized_name: 'benchmarkReferences',
+                type: {
+                  name: 'Sequence',
+                  element: {
+                      client_side_validation: true,
+                      required: false,
+                      serialized_name: 'BenchmarkReferenceElementType',
+                      type: {
+                        name: 'Composite',
+                        class_name: 'BenchmarkReference'
+                      }
+                  }
+                }
+              }
+            }
+          }
+        }
+      end
+    end
+  end
+end

data/lib/2020-07-01-preview/generated/azure_mgmt_security/module_definition.rb

@@ -0,0 +1,9 @@
+# encoding: utf-8
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+
+module Azure end
+module Azure::Security end
+module Azure::Security::Mgmt end
+module Azure::Security::Mgmt::V2020_07_01_preview end

data/lib/2020-07-01-preview/generated/azure_mgmt_security/security_center.rb

@@ -0,0 +1,135 @@
+# encoding: utf-8
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+
+module Azure::Security::Mgmt::V2020_07_01_preview
+  #
+  # A service client - single point of access to the REST API.
+  #
+  class SecurityCenter < MsRestAzure::AzureServiceClient
+    include MsRestAzure
+    include MsRestAzure::Serialization
+
+    # @return [String] the base URI of the service.
+    attr_accessor :base_url
+
+    # @return Credentials needed for the client to connect to Azure.
+    attr_reader :credentials
+
+    # @return [String] The preferred language for the response.
+    attr_accessor :accept_language
+
+    # @return [Integer] The retry timeout in seconds for Long Running
+    # Operations. Default value is 30.
+    attr_accessor :long_running_operation_retry_timeout
+
+    # @return [Boolean] Whether a unique x-ms-client-request-id should be
+    # generated. When set to true a unique x-ms-client-request-id value is
+    # generated and included in each request. Default is true.
+    attr_accessor :generate_client_request_id
+
+    # @return [SqlVulnerabilityAssessmentScans]
+    # sql_vulnerability_assessment_scans
+    attr_reader :sql_vulnerability_assessment_scans
+
+    # @return [SqlVulnerabilityAssessmentScanResults]
+    # sql_vulnerability_assessment_scan_results
+    attr_reader :sql_vulnerability_assessment_scan_results
+
+    # @return [SqlVulnerabilityAssessmentBaselineRules]
+    # sql_vulnerability_assessment_baseline_rules
+    attr_reader :sql_vulnerability_assessment_baseline_rules
+
+    #
+    # Creates initializes a new instance of the SecurityCenter class.
+    # @param credentials [MsRest::ServiceClientCredentials] credentials to authorize HTTP requests made by the service client.
+    # @param base_url [String] the base URI of the service.
+    # @param options [Array] filters to be applied to the HTTP requests.
+    #
+    def initialize(credentials = nil, base_url = nil, options = nil)
+      super(credentials, options)
+      @base_url = base_url || 'https://management.azure.com'
+
+      fail ArgumentError, 'invalid type of credentials input parameter' unless credentials.is_a?(MsRest::ServiceClientCredentials) unless credentials.nil?
+      @credentials = credentials
+
+      @sql_vulnerability_assessment_scans = SqlVulnerabilityAssessmentScans.new(self)
+      @sql_vulnerability_assessment_scan_results = SqlVulnerabilityAssessmentScanResults.new(self)
+      @sql_vulnerability_assessment_baseline_rules = SqlVulnerabilityAssessmentBaselineRules.new(self)
+      @accept_language = 'en-US'
+      @long_running_operation_retry_timeout = 30
+      @generate_client_request_id = true
+      add_telemetry
+    end
+
+    #
+    # Makes a request and returns the body of the response.
+    # @param method [Symbol] with any of the following values :get, :put, :post, :patch, :delete.
+    # @param path [String] the path, relative to {base_url}.
+    # @param options [Hash{String=>String}] specifying any request options like :body.
+    # @return [Hash{String=>String}] containing the body of the response.
+    # Example:
+    #
+    #  request_content = "{'location':'westus','tags':{'tag1':'val1','tag2':'val2'}}"
+    #  path = "/path"
+    #  options = {
+    #    body: request_content,
+    #    query_params: {'api-version' => '2016-02-01'}
+    #  }
+    #  result = @client.make_request(:put, path, options)
+    #
+    def make_request(method, path, options = {})
+      result = make_request_with_http_info(method, path, options)
+      result.body unless result.nil?
+    end
+
+    #
+    # Makes a request and returns the operation response.
+    # @param method [Symbol] with any of the following values :get, :put, :post, :patch, :delete.
+    # @param path [String] the path, relative to {base_url}.
+    # @param options [Hash{String=>String}] specifying any request options like :body.
+    # @return [MsRestAzure::AzureOperationResponse] Operation response containing the request, response and status.
+    #
+    def make_request_with_http_info(method, path, options = {})
+      result = make_request_async(method, path, options).value!
+      result.body = result.response.body.to_s.empty? ? nil : JSON.load(result.response.body)
+      result
+    end
+
+    #
+    # Makes a request asynchronously.
+    # @param method [Symbol] with any of the following values :get, :put, :post, :patch, :delete.
+    # @param path [String] the path, relative to {base_url}.
+    # @param options [Hash{String=>String}] specifying any request options like :body.
+    # @return [Concurrent::Promise] Promise object which holds the HTTP response.
+    #
+    def make_request_async(method, path, options = {})
+      fail ArgumentError, 'method is nil' if method.nil?
+      fail ArgumentError, 'path is nil' if path.nil?
+
+      request_url = options[:base_url] || @base_url
+      if(!options[:headers].nil? && !options[:headers]['Content-Type'].nil?)
+        @request_headers['Content-Type'] = options[:headers]['Content-Type']
+      end
+
+      request_headers = @request_headers
+      request_headers.merge!({'accept-language' => @accept_language}) unless @accept_language.nil?
+      options.merge!({headers: request_headers.merge(options[:headers] || {})})
+      options.merge!({credentials: @credentials}) unless @credentials.nil?
+
+      super(request_url, method, path, options)
+    end
+
+
+    private
+    #
+    # Adds telemetry information.
+    #
+    def add_telemetry
+        sdk_information = 'azure_mgmt_security'
+        sdk_information = "#{sdk_information}/0.19.0"
+        add_user_agent_information(sdk_information)
+    end
+  end
+end

data/lib/2020-07-01-preview/generated/azure_mgmt_security/sql_vulnerability_assessment_baseline_rules.rb

@@ -0,0 +1,535 @@
+# encoding: utf-8
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+
+module Azure::Security::Mgmt::V2020_07_01_preview
+  #
+  # API spec for Microsoft.Security (Azure Security Center) resource provider
+  #
+  class SqlVulnerabilityAssessmentBaselineRules
+    include MsRestAzure
+
+    #
+    # Creates and initializes a new instance of the SqlVulnerabilityAssessmentBaselineRules class.
+    # @param client service class for accessing basic functionality.
+    #
+    def initialize(client)
+      @client = client
+    end
+
+    # @return [SecurityCenter] reference to the SecurityCenter
+    attr_reader :client
+
+    #
+    # Creates a Baseline for a rule in a database. Will overwrite any previously
+    # existing results.
+    #
+    # @param rule_id [String] The rule Id.
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param body [RuleResultsInput] The baseline results for this rule.
+    # @param custom_headers [Hash{String => String}] A hash of custom headers that
+    # will be added to the HTTP request.
+    #
+    # @return [RuleResults] operation results.
+    #
+    def create_or_update(rule_id, workspace_id, api_version, resource_id, body:nil, custom_headers:nil)
+      response = create_or_update_async(rule_id, workspace_id, api_version, resource_id, body:body, custom_headers:custom_headers).value!
+      response.body unless response.nil?
+    end
+
+    #
+    # Creates a Baseline for a rule in a database. Will overwrite any previously
+    # existing results.
+    #
+    # @param rule_id [String] The rule Id.
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param body [RuleResultsInput] The baseline results for this rule.
+    # @param custom_headers [Hash{String => String}] A hash of custom headers that
+    # will be added to the HTTP request.
+    #
+    # @return [MsRestAzure::AzureOperationResponse] HTTP response information.
+    #
+    def create_or_update_with_http_info(rule_id, workspace_id, api_version, resource_id, body:nil, custom_headers:nil)
+      create_or_update_async(rule_id, workspace_id, api_version, resource_id, body:body, custom_headers:custom_headers).value!
+    end
+
+    #
+    # Creates a Baseline for a rule in a database. Will overwrite any previously
+    # existing results.
+    #
+    # @param rule_id [String] The rule Id.
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param body [RuleResultsInput] The baseline results for this rule.
+    # @param [Hash{String => String}] A hash of custom headers that will be added
+    # to the HTTP request.
+    #
+    # @return [Concurrent::Promise] Promise object which holds the HTTP response.
+    #
+    def create_or_update_async(rule_id, workspace_id, api_version, resource_id, body:nil, custom_headers:nil)
+      fail ArgumentError, 'rule_id is nil' if rule_id.nil?
+      fail ArgumentError, 'workspace_id is nil' if workspace_id.nil?
+      fail ArgumentError, 'api_version is nil' if api_version.nil?
+      fail ArgumentError, 'resource_id is nil' if resource_id.nil?
+
+
+      request_headers = {}
+      request_headers['Content-Type'] = 'application/json; charset=utf-8'
+
+      # Set Headers
+      request_headers['x-ms-client-request-id'] = SecureRandom.uuid
+      request_headers['accept-language'] = @client.accept_language unless @client.accept_language.nil?
+
+      # Serialize Request
+      request_mapper = Azure::Security::Mgmt::V2020_07_01_preview::Models::RuleResultsInput.mapper()
+      request_content = @client.serialize(request_mapper,  body)
+      request_content = request_content != nil ? JSON.generate(request_content, quirks_mode: true) : nil
+
+      path_template = '{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}'
+
+      request_url = @base_url || @client.base_url
+
+      options = {
+          middlewares: [[MsRest::RetryPolicyMiddleware, times: 3, retry: 0.02], [:cookie_jar]],
+          path_params: {'ruleId' => rule_id},
+          skip_encoding_path_params: {'resourceId' => resource_id},
+          query_params: {'workspaceId' => workspace_id,'api-version' => api_version},
+          body: request_content,
+          headers: request_headers.merge(custom_headers || {}),
+          base_url: request_url
+      }
+      promise = @client.make_request_async(:put, path_template, options)
+
+      promise = promise.then do |result|
+        http_response = result.response
+        status_code = http_response.status
+        response_content = http_response.body
+        unless status_code == 200
+          error_model = JSON.load(response_content)
+          fail MsRestAzure::AzureOperationError.new(result.request, http_response, error_model)
+        end
+
+        result.request_id = http_response['x-ms-request-id'] unless http_response['x-ms-request-id'].nil?
+        result.correlation_request_id = http_response['x-ms-correlation-request-id'] unless http_response['x-ms-correlation-request-id'].nil?
+        result.client_request_id = http_response['x-ms-client-request-id'] unless http_response['x-ms-client-request-id'].nil?
+        # Deserialize Response
+        if status_code == 200
+          begin
+            parsed_response = response_content.to_s.empty? ? nil : JSON.load(response_content)
+            result_mapper = Azure::Security::Mgmt::V2020_07_01_preview::Models::RuleResults.mapper()
+            result.body = @client.deserialize(result_mapper, parsed_response)
+          rescue Exception => e
+            fail MsRest::DeserializationError.new('Error occurred in deserializing the response', e.message, e.backtrace, result)
+          end
+        end
+
+        result
+      end
+
+      promise.execute
+    end
+
+    #
+    # Gets the results for a given rule in the Baseline.
+    #
+    # @param rule_id [String] The rule Id.
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param custom_headers [Hash{String => String}] A hash of custom headers that
+    # will be added to the HTTP request.
+    #
+    # @return [RuleResults] operation results.
+    #
+    def get(rule_id, workspace_id, api_version, resource_id, custom_headers:nil)
+      response = get_async(rule_id, workspace_id, api_version, resource_id, custom_headers:custom_headers).value!
+      response.body unless response.nil?
+    end
+
+    #
+    # Gets the results for a given rule in the Baseline.
+    #
+    # @param rule_id [String] The rule Id.
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param custom_headers [Hash{String => String}] A hash of custom headers that
+    # will be added to the HTTP request.
+    #
+    # @return [MsRestAzure::AzureOperationResponse] HTTP response information.
+    #
+    def get_with_http_info(rule_id, workspace_id, api_version, resource_id, custom_headers:nil)
+      get_async(rule_id, workspace_id, api_version, resource_id, custom_headers:custom_headers).value!
+    end
+
+    #
+    # Gets the results for a given rule in the Baseline.
+    #
+    # @param rule_id [String] The rule Id.
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param [Hash{String => String}] A hash of custom headers that will be added
+    # to the HTTP request.
+    #
+    # @return [Concurrent::Promise] Promise object which holds the HTTP response.
+    #
+    def get_async(rule_id, workspace_id, api_version, resource_id, custom_headers:nil)
+      fail ArgumentError, 'rule_id is nil' if rule_id.nil?
+      fail ArgumentError, 'workspace_id is nil' if workspace_id.nil?
+      fail ArgumentError, 'api_version is nil' if api_version.nil?
+      fail ArgumentError, 'resource_id is nil' if resource_id.nil?
+
+
+      request_headers = {}
+      request_headers['Content-Type'] = 'application/json; charset=utf-8'
+
+      # Set Headers
+      request_headers['x-ms-client-request-id'] = SecureRandom.uuid
+      request_headers['accept-language'] = @client.accept_language unless @client.accept_language.nil?
+      path_template = '{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}'
+
+      request_url = @base_url || @client.base_url
+
+      options = {
+          middlewares: [[MsRest::RetryPolicyMiddleware, times: 3, retry: 0.02], [:cookie_jar]],
+          path_params: {'ruleId' => rule_id},
+          skip_encoding_path_params: {'resourceId' => resource_id},
+          query_params: {'workspaceId' => workspace_id,'api-version' => api_version},
+          headers: request_headers.merge(custom_headers || {}),
+          base_url: request_url
+      }
+      promise = @client.make_request_async(:get, path_template, options)
+
+      promise = promise.then do |result|
+        http_response = result.response
+        status_code = http_response.status
+        response_content = http_response.body
+        unless status_code == 200
+          error_model = JSON.load(response_content)
+          fail MsRestAzure::AzureOperationError.new(result.request, http_response, error_model)
+        end
+
+        result.request_id = http_response['x-ms-request-id'] unless http_response['x-ms-request-id'].nil?
+        result.correlation_request_id = http_response['x-ms-correlation-request-id'] unless http_response['x-ms-correlation-request-id'].nil?
+        result.client_request_id = http_response['x-ms-client-request-id'] unless http_response['x-ms-client-request-id'].nil?
+        # Deserialize Response
+        if status_code == 200
+          begin
+            parsed_response = response_content.to_s.empty? ? nil : JSON.load(response_content)
+            result_mapper = Azure::Security::Mgmt::V2020_07_01_preview::Models::RuleResults.mapper()
+            result.body = @client.deserialize(result_mapper, parsed_response)
+          rescue Exception => e
+            fail MsRest::DeserializationError.new('Error occurred in deserializing the response', e.message, e.backtrace, result)
+          end
+        end
+
+        result
+      end
+
+      promise.execute
+    end
+
+    #
+    # Deletes a rule from the Baseline of a given database.
+    #
+    # @param rule_id [String] The rule Id.
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param custom_headers [Hash{String => String}] A hash of custom headers that
+    # will be added to the HTTP request.
+    #
+    #
+    def delete(rule_id, workspace_id, api_version, resource_id, custom_headers:nil)
+      response = delete_async(rule_id, workspace_id, api_version, resource_id, custom_headers:custom_headers).value!
+      nil
+    end
+
+    #
+    # Deletes a rule from the Baseline of a given database.
+    #
+    # @param rule_id [String] The rule Id.
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param custom_headers [Hash{String => String}] A hash of custom headers that
+    # will be added to the HTTP request.
+    #
+    # @return [MsRestAzure::AzureOperationResponse] HTTP response information.
+    #
+    def delete_with_http_info(rule_id, workspace_id, api_version, resource_id, custom_headers:nil)
+      delete_async(rule_id, workspace_id, api_version, resource_id, custom_headers:custom_headers).value!
+    end
+
+    #
+    # Deletes a rule from the Baseline of a given database.
+    #
+    # @param rule_id [String] The rule Id.
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param [Hash{String => String}] A hash of custom headers that will be added
+    # to the HTTP request.
+    #
+    # @return [Concurrent::Promise] Promise object which holds the HTTP response.
+    #
+    def delete_async(rule_id, workspace_id, api_version, resource_id, custom_headers:nil)
+      fail ArgumentError, 'rule_id is nil' if rule_id.nil?
+      fail ArgumentError, 'workspace_id is nil' if workspace_id.nil?
+      fail ArgumentError, 'api_version is nil' if api_version.nil?
+      fail ArgumentError, 'resource_id is nil' if resource_id.nil?
+
+
+      request_headers = {}
+      request_headers['Content-Type'] = 'application/json; charset=utf-8'
+
+      # Set Headers
+      request_headers['x-ms-client-request-id'] = SecureRandom.uuid
+      request_headers['accept-language'] = @client.accept_language unless @client.accept_language.nil?
+      path_template = '{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}'
+
+      request_url = @base_url || @client.base_url
+
+      options = {
+          middlewares: [[MsRest::RetryPolicyMiddleware, times: 3, retry: 0.02], [:cookie_jar]],
+          path_params: {'ruleId' => rule_id},
+          skip_encoding_path_params: {'resourceId' => resource_id},
+          query_params: {'workspaceId' => workspace_id,'api-version' => api_version},
+          headers: request_headers.merge(custom_headers || {}),
+          base_url: request_url
+      }
+      promise = @client.make_request_async(:delete, path_template, options)
+
+      promise = promise.then do |result|
+        http_response = result.response
+        status_code = http_response.status
+        response_content = http_response.body
+        unless status_code == 200 || status_code == 204
+          error_model = JSON.load(response_content)
+          fail MsRestAzure::AzureOperationError.new(result.request, http_response, error_model)
+        end
+
+        result.request_id = http_response['x-ms-request-id'] unless http_response['x-ms-request-id'].nil?
+        result.correlation_request_id = http_response['x-ms-correlation-request-id'] unless http_response['x-ms-correlation-request-id'].nil?
+        result.client_request_id = http_response['x-ms-client-request-id'] unless http_response['x-ms-client-request-id'].nil?
+
+        result
+      end
+
+      promise.execute
+    end
+
+    #
+    # Gets the results for all rules in the Baseline.
+    #
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param custom_headers [Hash{String => String}] A hash of custom headers that
+    # will be added to the HTTP request.
+    #
+    # @return [RulesResults] operation results.
+    #
+    def list(workspace_id, api_version, resource_id, custom_headers:nil)
+      response = list_async(workspace_id, api_version, resource_id, custom_headers:custom_headers).value!
+      response.body unless response.nil?
+    end
+
+    #
+    # Gets the results for all rules in the Baseline.
+    #
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param custom_headers [Hash{String => String}] A hash of custom headers that
+    # will be added to the HTTP request.
+    #
+    # @return [MsRestAzure::AzureOperationResponse] HTTP response information.
+    #
+    def list_with_http_info(workspace_id, api_version, resource_id, custom_headers:nil)
+      list_async(workspace_id, api_version, resource_id, custom_headers:custom_headers).value!
+    end
+
+    #
+    # Gets the results for all rules in the Baseline.
+    #
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param [Hash{String => String}] A hash of custom headers that will be added
+    # to the HTTP request.
+    #
+    # @return [Concurrent::Promise] Promise object which holds the HTTP response.
+    #
+    def list_async(workspace_id, api_version, resource_id, custom_headers:nil)
+      fail ArgumentError, 'workspace_id is nil' if workspace_id.nil?
+      fail ArgumentError, 'api_version is nil' if api_version.nil?
+      fail ArgumentError, 'resource_id is nil' if resource_id.nil?
+
+
+      request_headers = {}
+      request_headers['Content-Type'] = 'application/json; charset=utf-8'
+
+      # Set Headers
+      request_headers['x-ms-client-request-id'] = SecureRandom.uuid
+      request_headers['accept-language'] = @client.accept_language unless @client.accept_language.nil?
+      path_template = '{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules'
+
+      request_url = @base_url || @client.base_url
+
+      options = {
+          middlewares: [[MsRest::RetryPolicyMiddleware, times: 3, retry: 0.02], [:cookie_jar]],
+          skip_encoding_path_params: {'resourceId' => resource_id},
+          query_params: {'workspaceId' => workspace_id,'api-version' => api_version},
+          headers: request_headers.merge(custom_headers || {}),
+          base_url: request_url
+      }
+      promise = @client.make_request_async(:get, path_template, options)
+
+      promise = promise.then do |result|
+        http_response = result.response
+        status_code = http_response.status
+        response_content = http_response.body
+        unless status_code == 200
+          error_model = JSON.load(response_content)
+          fail MsRestAzure::AzureOperationError.new(result.request, http_response, error_model)
+        end
+
+        result.request_id = http_response['x-ms-request-id'] unless http_response['x-ms-request-id'].nil?
+        result.correlation_request_id = http_response['x-ms-correlation-request-id'] unless http_response['x-ms-correlation-request-id'].nil?
+        result.client_request_id = http_response['x-ms-client-request-id'] unless http_response['x-ms-client-request-id'].nil?
+        # Deserialize Response
+        if status_code == 200
+          begin
+            parsed_response = response_content.to_s.empty? ? nil : JSON.load(response_content)
+            result_mapper = Azure::Security::Mgmt::V2020_07_01_preview::Models::RulesResults.mapper()
+            result.body = @client.deserialize(result_mapper, parsed_response)
+          rescue Exception => e
+            fail MsRest::DeserializationError.new('Error occurred in deserializing the response', e.message, e.backtrace, result)
+          end
+        end
+
+        result
+      end
+
+      promise.execute
+    end
+
+    #
+    # Add a list of baseline rules. Will overwrite any previously existing results
+    # (for all rules).
+    #
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param body [RulesResultsInput] The baseline rules.
+    # @param custom_headers [Hash{String => String}] A hash of custom headers that
+    # will be added to the HTTP request.
+    #
+    # @return [RulesResults] operation results.
+    #
+    def add(workspace_id, api_version, resource_id, body:nil, custom_headers:nil)
+      response = add_async(workspace_id, api_version, resource_id, body:body, custom_headers:custom_headers).value!
+      response.body unless response.nil?
+    end
+
+    #
+    # Add a list of baseline rules. Will overwrite any previously existing results
+    # (for all rules).
+    #
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param body [RulesResultsInput] The baseline rules.
+    # @param custom_headers [Hash{String => String}] A hash of custom headers that
+    # will be added to the HTTP request.
+    #
+    # @return [MsRestAzure::AzureOperationResponse] HTTP response information.
+    #
+    def add_with_http_info(workspace_id, api_version, resource_id, body:nil, custom_headers:nil)
+      add_async(workspace_id, api_version, resource_id, body:body, custom_headers:custom_headers).value!
+    end
+
+    #
+    # Add a list of baseline rules. Will overwrite any previously existing results
+    # (for all rules).
+    #
+    # @param workspace_id [String] The workspace Id.
+    # @param api_version [String] The api version.
+    # @param resource_id [String] The identifier of the resource.
+    # @param body [RulesResultsInput] The baseline rules.
+    # @param [Hash{String => String}] A hash of custom headers that will be added
+    # to the HTTP request.
+    #
+    # @return [Concurrent::Promise] Promise object which holds the HTTP response.
+    #
+    def add_async(workspace_id, api_version, resource_id, body:nil, custom_headers:nil)
+      fail ArgumentError, 'workspace_id is nil' if workspace_id.nil?
+      fail ArgumentError, 'api_version is nil' if api_version.nil?
+      fail ArgumentError, 'resource_id is nil' if resource_id.nil?
+
+
+      request_headers = {}
+      request_headers['Content-Type'] = 'application/json; charset=utf-8'
+
+      # Set Headers
+      request_headers['x-ms-client-request-id'] = SecureRandom.uuid
+      request_headers['accept-language'] = @client.accept_language unless @client.accept_language.nil?
+
+      # Serialize Request
+      request_mapper = Azure::Security::Mgmt::V2020_07_01_preview::Models::RulesResultsInput.mapper()
+      request_content = @client.serialize(request_mapper,  body)
+      request_content = request_content != nil ? JSON.generate(request_content, quirks_mode: true) : nil
+
+      path_template = '{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules'
+
+      request_url = @base_url || @client.base_url
+
+      options = {
+          middlewares: [[MsRest::RetryPolicyMiddleware, times: 3, retry: 0.02], [:cookie_jar]],
+          skip_encoding_path_params: {'resourceId' => resource_id},
+          query_params: {'workspaceId' => workspace_id,'api-version' => api_version},
+          body: request_content,
+          headers: request_headers.merge(custom_headers || {}),
+          base_url: request_url
+      }
+      promise = @client.make_request_async(:post, path_template, options)
+
+      promise = promise.then do |result|
+        http_response = result.response
+        status_code = http_response.status
+        response_content = http_response.body
+        unless status_code == 200
+          error_model = JSON.load(response_content)
+          fail MsRestAzure::AzureOperationError.new(result.request, http_response, error_model)
+        end
+
+        result.request_id = http_response['x-ms-request-id'] unless http_response['x-ms-request-id'].nil?
+        result.correlation_request_id = http_response['x-ms-correlation-request-id'] unless http_response['x-ms-correlation-request-id'].nil?
+        result.client_request_id = http_response['x-ms-client-request-id'] unless http_response['x-ms-client-request-id'].nil?
+        # Deserialize Response
+        if status_code == 200
+          begin
+            parsed_response = response_content.to_s.empty? ? nil : JSON.load(response_content)
+            result_mapper = Azure::Security::Mgmt::V2020_07_01_preview::Models::RulesResults.mapper()
+            result.body = @client.deserialize(result_mapper, parsed_response)
+          rescue Exception => e
+            fail MsRest::DeserializationError.new('Error occurred in deserializing the response', e.message, e.backtrace, result)
+          end
+        end
+
+        result
+      end
+
+      promise.execute
+    end
+
+  end
+end