From ed8137a47ec1d66e4b775c92e3ebc8e20a62a49d Mon Sep 17 00:00:00 2001 From: Christopher Wilcox Date: Wed, 28 Nov 2018 14:28:56 -0800 Subject: [PATCH] Run Black on Generated libraries (#6666) * blacken appveyor * Blacken all gen'd libraries not under PR --- .../google-cloud-securitycenter/docs/conf.py | 168 ++-- .../google/__init__.py | 2 + .../google/cloud/__init__.py | 2 + .../google/cloud/securitycenter.py | 6 +- .../cloud/securitycenter_v1beta1/__init__.py | 6 +- .../securitycenter_v1beta1/gapic/enums.py | 4 + .../gapic/security_center_client.py | 819 +++++++++--------- .../gapic/security_center_client_config.py | 44 +- .../security_center_grpc_transport.py | 78 +- .../cloud/securitycenter_v1beta1/types.py | 2 +- .../test_security_center_client_v1beta1.py | 387 +++++---- 11 files changed, 790 insertions(+), 728 deletions(-) diff --git a/packages/google-cloud-securitycenter/docs/conf.py b/packages/google-cloud-securitycenter/docs/conf.py index f0f50ac8cfbd..60b537aac388 100644 --- a/packages/google-cloud-securitycenter/docs/conf.py +++ b/packages/google-cloud-securitycenter/docs/conf.py @@ -18,50 +18,50 @@ # If extensions (or modules to document with autodoc) are in another directory, # add these directories to sys.path here. If the directory is relative to the # documentation root, use os.path.abspath to make it absolute, like shown here. -sys.path.insert(0, os.path.abspath('..')) +sys.path.insert(0, os.path.abspath("..")) -__version__ = '0.1.0' +__version__ = "0.1.0" # -- General configuration ------------------------------------------------ # If your documentation needs a minimal Sphinx version, state it here. -#needs_sphinx = '1.0' +# needs_sphinx = '1.0' # Add any Sphinx extension module names here, as strings. They can be # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom # ones. extensions = [ - 'sphinx.ext.autodoc', - 'sphinx.ext.autosummary', - 'sphinx.ext.intersphinx', - 'sphinx.ext.coverage', - 'sphinx.ext.napoleon', - 'sphinx.ext.viewcode', + "sphinx.ext.autodoc", + "sphinx.ext.autosummary", + "sphinx.ext.intersphinx", + "sphinx.ext.coverage", + "sphinx.ext.napoleon", + "sphinx.ext.viewcode", ] # autodoc/autosummary flags -autoclass_content = 'both' -autodoc_default_flags = ['members'] +autoclass_content = "both" +autodoc_default_flags = ["members"] autosummary_generate = True # Add any paths that contain templates here, relative to this directory. -templates_path = ['_templates'] +templates_path = ["_templates"] # The suffix(es) of source filenames. # You can specify multiple suffix as a list of string: # source_suffix = ['.rst', '.md'] -source_suffix = '.rst' +source_suffix = ".rst" # The encoding of source files. -#source_encoding = 'utf-8-sig' +# source_encoding = 'utf-8-sig' # The master toctree document. -master_doc = 'index' +master_doc = "index" # General information about the project. -project = u'google-cloud-securitycenter' -copyright = u'2017, Google' -author = u'Google APIs' +project = u"google-cloud-securitycenter" +copyright = u"2017, Google" +author = u"Google APIs" # The version info for the project you're documenting, acts as replacement for # |version| and |release|, also used in various other places throughout the @@ -70,7 +70,7 @@ # The full version, including alpha/beta/rc tags. release = __version__ # The short X.Y version. -version = '.'.join(release.split('.')[0:2]) +version = ".".join(release.split(".")[0:2]) # The language for content autogenerated by Sphinx. Refer to documentation # for a list of supported languages. @@ -81,37 +81,37 @@ # There are two options for replacing |today|: either, you set today to some # non-false value, then it is used: -#today = '' +# today = '' # Else, today_fmt is used as the format for a strftime call. -#today_fmt = '%B %d, %Y' +# today_fmt = '%B %d, %Y' # List of patterns, relative to source directory, that match files and # directories to ignore when looking for source files. -exclude_patterns = ['_build'] +exclude_patterns = ["_build"] # The reST default role (used for this markup: `text`) to use for all # documents. -#default_role = None +# default_role = None # If true, '()' will be appended to :func: etc. cross-reference text. -#add_function_parentheses = True +# add_function_parentheses = True # If true, the current module name will be prepended to all description # unit titles (such as .. function::). -#add_module_names = True +# add_module_names = True # If true, sectionauthor and moduleauthor directives will be shown in the # output. They are ignored by default. -#show_authors = False +# show_authors = False # The name of the Pygments (syntax highlighting) style to use. -pygments_style = 'sphinx' +pygments_style = "sphinx" # A list of ignored prefixes for module index sorting. -#modindex_common_prefix = [] +# modindex_common_prefix = [] # If true, keep warnings as "system message" paragraphs in the built documents. -#keep_warnings = False +# keep_warnings = False # If true, `todo` and `todoList` produce output, else they produce nothing. todo_include_todos = True @@ -120,31 +120,31 @@ # The theme to use for HTML and HTML Help pages. See the documentation for # a list of builtin themes. -html_theme = 'sphinx_rtd_theme' +html_theme = "sphinx_rtd_theme" # Theme options are theme-specific and customize the look and feel of a theme # further. For a list of options available for each theme, see the # documentation. -#html_theme_options = {} +# html_theme_options = {} # Add any paths that contain custom themes here, relative to this directory. -#html_theme_path = [] +# html_theme_path = [] # The name for this set of Sphinx documents. If None, it defaults to # " v documentation". -#html_title = None +# html_title = None # A shorter title for the navigation bar. Default is the same as html_title. -#html_short_title = None +# html_short_title = None # The name of an image file (relative to this directory) to place at the top # of the sidebar. -#html_logo = None +# html_logo = None # The name of an image file (within the static path) to use as favicon of the # docs. This file should be a Windows icon file (.ico) being 16x16 or 32x32 # pixels large. -#html_favicon = None +# html_favicon = None # Add any paths that contain custom static files (such as style sheets) here, # relative to this directory. They are copied after the builtin static files, @@ -154,78 +154,75 @@ # Add any extra paths that contain custom files (such as robots.txt or # .htaccess) here, relative to this directory. These files are copied # directly to the root of the documentation. -#html_extra_path = [] +# html_extra_path = [] # If not '', a 'Last updated on:' timestamp is inserted at every page bottom, # using the given strftime format. -#html_last_updated_fmt = '%b %d, %Y' +# html_last_updated_fmt = '%b %d, %Y' # If true, SmartyPants will be used to convert quotes and dashes to # typographically correct entities. -#html_use_smartypants = True +# html_use_smartypants = True # Custom sidebar templates, maps document names to template names. -#html_sidebars = {} +# html_sidebars = {} # Additional templates that should be rendered to pages, maps page names to # template names. -#html_additional_pages = {} +# html_additional_pages = {} # If false, no module index is generated. -#html_domain_indices = True +# html_domain_indices = True # If false, no index is generated. -#html_use_index = True +# html_use_index = True # If true, the index is split into individual pages for each letter. -#html_split_index = False +# html_split_index = False # If true, links to the reST sources are added to the pages. -#html_show_sourcelink = True +# html_show_sourcelink = True # If true, "Created using Sphinx" is shown in the HTML footer. Default is True. -#html_show_sphinx = True +# html_show_sphinx = True # If true, "(C) Copyright ..." is shown in the HTML footer. Default is True. -#html_show_copyright = True +# html_show_copyright = True # If true, an OpenSearch description file will be output, and all pages will # contain a tag referring to it. The value of this option must be the # base URL from which the finished HTML is served. -#html_use_opensearch = '' +# html_use_opensearch = '' # This is the file name suffix for HTML files (e.g. ".xhtml"). -#html_file_suffix = None +# html_file_suffix = None # Language to be used for generating the HTML full-text search index. # Sphinx supports the following languages: # 'da', 'de', 'en', 'es', 'fi', 'fr', 'hu', 'it', 'ja' # 'nl', 'no', 'pt', 'ro', 'ru', 'sv', 'tr' -#html_search_language = 'en' +# html_search_language = 'en' # A dictionary with options for the search language support, empty by default. # Now only 'ja' uses this config value -#html_search_options = {'type': 'default'} +# html_search_options = {'type': 'default'} # The name of a javascript file (relative to the configuration directory) that # implements a search results scorer. If empty, the default will be used. -#html_search_scorer = 'scorer.js' +# html_search_scorer = 'scorer.js' # Output file base name for HTML help builder. -htmlhelp_basename = 'google-cloud-securitycenter-doc' +htmlhelp_basename = "google-cloud-securitycenter-doc" # -- Options for LaTeX output --------------------------------------------- latex_elements = { # The paper size ('letterpaper' or 'a4paper'). #'papersize': 'letterpaper', - # The font size ('10pt', '11pt' or '12pt'). #'pointsize': '10pt', - # Additional stuff for the LaTeX preamble. #'preamble': '', - # Latex figure (float) alignment #'figure_align': 'htbp', } @@ -234,39 +231,51 @@ # (source start file, target name, title, # author, documentclass [howto, manual, or own class]). latex_documents = [ - (master_doc, 'google-cloud-securitycenter.tex', - u'google-cloud-securitycenter Documentation', author, 'manual'), + ( + master_doc, + "google-cloud-securitycenter.tex", + u"google-cloud-securitycenter Documentation", + author, + "manual", + ) ] # The name of an image file (relative to this directory) to place at the top of # the title page. -#latex_logo = None +# latex_logo = None # For "manual" documents, if this is true, then toplevel headings are parts, # not chapters. -#latex_use_parts = False +# latex_use_parts = False # If true, show page references after internal links. -#latex_show_pagerefs = False +# latex_show_pagerefs = False # If true, show URL addresses after external links. -#latex_show_urls = False +# latex_show_urls = False # Documents to append as an appendix to all manuals. -#latex_appendices = [] +# latex_appendices = [] # If false, no module index is generated. -#latex_domain_indices = True +# latex_domain_indices = True # -- Options for manual page output --------------------------------------- # One entry per manual page. List of tuples # (source start file, name, description, authors, manual section). -man_pages = [(master_doc, 'google-cloud-securitycenter', - u'google-cloud-securitycenter Documentation', [author], 1)] +man_pages = [ + ( + master_doc, + "google-cloud-securitycenter", + u"google-cloud-securitycenter Documentation", + [author], + 1, + ) +] # If true, show URL addresses after external links. -#man_show_urls = False +# man_show_urls = False # -- Options for Texinfo output ------------------------------------------- @@ -274,28 +283,33 @@ # (source start file, target name, title, author, # dir menu entry, description, category) texinfo_documents = [ - (master_doc, 'google-cloud-securitycenter', - u'google-cloud-securitycenter Documentation', author, - 'google-cloud-securitycenter', - 'GAPIC library for the {metadata.shortName} v1beta1 service', 'APIs'), + ( + master_doc, + "google-cloud-securitycenter", + u"google-cloud-securitycenter Documentation", + author, + "google-cloud-securitycenter", + "GAPIC library for the {metadata.shortName} v1beta1 service", + "APIs", + ) ] # Documents to append as an appendix to all manuals. -#texinfo_appendices = [] +# texinfo_appendices = [] # If false, no module index is generated. -#texinfo_domain_indices = True +# texinfo_domain_indices = True # How to display URL addresses: 'footnote', 'no', or 'inline'. -#texinfo_show_urls = 'footnote' +# texinfo_show_urls = 'footnote' # If true, do not generate a @detailmenu in the "Top" node's menu. -#texinfo_no_detailmenu = False +# texinfo_no_detailmenu = False # Example configuration for intersphinx: refer to the Python standard library. intersphinx_mapping = { - 'python': ('http://python.readthedocs.org/en/latest/', None), - 'gax': ('https://gax-python.readthedocs.org/en/latest/', None), + "python": ("http://python.readthedocs.org/en/latest/", None), + "gax": ("https://gax-python.readthedocs.org/en/latest/", None), } # Napoleon settings diff --git a/packages/google-cloud-securitycenter/google/__init__.py b/packages/google-cloud-securitycenter/google/__init__.py index f65701dd143f..7aa71bb4e2f1 100644 --- a/packages/google-cloud-securitycenter/google/__init__.py +++ b/packages/google-cloud-securitycenter/google/__init__.py @@ -16,7 +16,9 @@ try: import pkg_resources + pkg_resources.declare_namespace(__name__) except ImportError: import pkgutil + __path__ = pkgutil.extend_path(__path__, __name__) diff --git a/packages/google-cloud-securitycenter/google/cloud/__init__.py b/packages/google-cloud-securitycenter/google/cloud/__init__.py index f65701dd143f..7aa71bb4e2f1 100644 --- a/packages/google-cloud-securitycenter/google/cloud/__init__.py +++ b/packages/google-cloud-securitycenter/google/cloud/__init__.py @@ -16,7 +16,9 @@ try: import pkg_resources + pkg_resources.declare_namespace(__name__) except ImportError: import pkgutil + __path__ = pkgutil.extend_path(__path__, __name__) diff --git a/packages/google-cloud-securitycenter/google/cloud/securitycenter.py b/packages/google-cloud-securitycenter/google/cloud/securitycenter.py index 0cdbfc23938d..714a4febb337 100644 --- a/packages/google-cloud-securitycenter/google/cloud/securitycenter.py +++ b/packages/google-cloud-securitycenter/google/cloud/securitycenter.py @@ -20,8 +20,4 @@ from google.cloud.securitycenter_v1beta1 import enums from google.cloud.securitycenter_v1beta1 import types -__all__ = ( - 'enums', - 'types', - 'SecurityCenterClient', -) +__all__ = ("enums", "types", "SecurityCenterClient") diff --git a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/__init__.py b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/__init__.py index b93174ba6ef1..688612f67132 100644 --- a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/__init__.py +++ b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/__init__.py @@ -26,8 +26,4 @@ class SecurityCenterClient(security_center_client.SecurityCenterClient): enums = enums -__all__ = ( - 'enums', - 'types', - 'SecurityCenterClient', -) +__all__ = ("enums", "types", "SecurityCenterClient") diff --git a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/enums.py b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/enums.py index 61b213ad97f9..bfe0629cde25 100644 --- a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/enums.py +++ b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/enums.py @@ -28,6 +28,7 @@ class NullValue(enum.IntEnum): Attributes: NULL_VALUE (int): Null value. """ + NULL_VALUE = 0 @@ -42,6 +43,7 @@ class State(enum.IntEnum): INACTIVE (int): The finding has been fixed, triaged as a non-issue or otherwise addressed and is no longer active. """ + STATE_UNSPECIFIED = 0 ACTIVE = 1 INACTIVE = 2 @@ -67,6 +69,7 @@ class InclusionMode(enum.IntEnum): EXCLUDE (int): Asset Discovery will ignore all resources under the projects specified. All other resources will be retrieved. """ + INCLUSION_MODE_UNSPECIFIED = 0 INCLUDE_ONLY = 1 EXCLUDE = 2 @@ -89,6 +92,7 @@ class State(enum.IntEnum): REMOVED (int): Asset was removed between the points in time. ACTIVE (int): Asset was active at both point(s) in time. """ + STATE_UNSPECIFIED = 0 UNUSED = 1 ADDED = 2 diff --git a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/security_center_client.py b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/security_center_client.py index 6ce62bd4d9b9..82a5d5278e7d 100644 --- a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/security_center_client.py +++ b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/security_center_client.py @@ -32,7 +32,9 @@ from google.cloud.securitycenter_v1beta1.gapic import enums from google.cloud.securitycenter_v1beta1.gapic import security_center_client_config -from google.cloud.securitycenter_v1beta1.gapic.transports import security_center_grpc_transport +from google.cloud.securitycenter_v1beta1.gapic.transports import ( + security_center_grpc_transport, +) from google.cloud.securitycenter_v1beta1.proto import finding_pb2 from google.cloud.securitycenter_v1beta1.proto import organization_settings_pb2 from google.cloud.securitycenter_v1beta1.proto import security_marks_pb2 @@ -48,18 +50,19 @@ from google.protobuf import timestamp_pb2 _GAPIC_LIBRARY_VERSION = pkg_resources.get_distribution( - 'google-cloud-securitycenter', ).version + "google-cloud-securitycenter" +).version class SecurityCenterClient(object): """V1 Beta APIs for Security Center service.""" - SERVICE_ADDRESS = 'securitycenter.googleapis.com:443' + SERVICE_ADDRESS = "securitycenter.googleapis.com:443" """The default address of the service.""" # The name of the interface for this client. This is the key used to # find the method configuration in the client_config dictionary. - _INTERFACE_NAME = 'google.cloud.securitycenter.v1beta1.SecurityCenter' + _INTERFACE_NAME = "google.cloud.securitycenter.v1beta1.SecurityCenter" @classmethod def from_service_account_file(cls, filename, *args, **kwargs): @@ -75,9 +78,8 @@ def from_service_account_file(cls, filename, *args, **kwargs): Returns: SecurityCenterClient: The constructed client. """ - credentials = service_account.Credentials.from_service_account_file( - filename) - kwargs['credentials'] = credentials + credentials = service_account.Credentials.from_service_account_file(filename) + kwargs["credentials"] = credentials return cls(*args, **kwargs) from_service_account_json = from_service_account_file @@ -86,15 +88,14 @@ def from_service_account_file(cls, filename, *args, **kwargs): def organization_path(cls, organization): """Return a fully-qualified organization string.""" return google.api_core.path_template.expand( - 'organizations/{organization}', - organization=organization, + "organizations/{organization}", organization=organization ) @classmethod def source_path(cls, organization, source): """Return a fully-qualified source string.""" return google.api_core.path_template.expand( - 'organizations/{organization}/sources/{source}', + "organizations/{organization}/sources/{source}", organization=organization, source=source, ) @@ -103,7 +104,7 @@ def source_path(cls, organization, source): def organization_settings_path(cls, organization): """Return a fully-qualified organization_settings string.""" return google.api_core.path_template.expand( - 'organizations/{organization}/organizationSettings', + "organizations/{organization}/organizationSettings", organization=organization, ) @@ -111,7 +112,7 @@ def organization_settings_path(cls, organization): def finding_path(cls, organization, source, finding): """Return a fully-qualified finding string.""" return google.api_core.path_template.expand( - 'organizations/{organization}/sources/{source}/findings/{finding}', + "organizations/{organization}/sources/{source}/findings/{finding}", organization=organization, source=source, finding=finding, @@ -121,17 +122,19 @@ def finding_path(cls, organization, source, finding): def asset_security_marks_path(cls, organization, asset): """Return a fully-qualified asset_security_marks string.""" return google.api_core.path_template.expand( - 'organizations/{organization}/assets/{asset}/securityMarks', + "organizations/{organization}/assets/{asset}/securityMarks", organization=organization, asset=asset, ) - def __init__(self, - transport=None, - channel=None, - credentials=None, - client_config=None, - client_info=None): + def __init__( + self, + transport=None, + channel=None, + credentials=None, + client_config=None, + client_info=None, + ): """Constructor. Args: @@ -165,18 +168,19 @@ def __init__(self, # Raise deprecation warnings for things we want to go away. if client_config is not None: warnings.warn( - 'The `client_config` argument is deprecated.', + "The `client_config` argument is deprecated.", PendingDeprecationWarning, - stacklevel=2) + stacklevel=2, + ) else: client_config = security_center_client_config.config if channel: warnings.warn( - 'The `channel` argument is deprecated; use ' - '`transport` instead.', + "The `channel` argument is deprecated; use " "`transport` instead.", PendingDeprecationWarning, - stacklevel=2) + stacklevel=2, + ) # Instantiate the transport. # The transport is responsible for handling serialization and @@ -185,25 +189,24 @@ def __init__(self, if callable(transport): self.transport = transport( credentials=credentials, - default_class=security_center_grpc_transport. - SecurityCenterGrpcTransport, + default_class=security_center_grpc_transport.SecurityCenterGrpcTransport, ) else: if credentials: raise ValueError( - 'Received both a transport instance and ' - 'credentials; these are mutually exclusive.') + "Received both a transport instance and " + "credentials; these are mutually exclusive." + ) self.transport = transport else: self.transport = security_center_grpc_transport.SecurityCenterGrpcTransport( - address=self.SERVICE_ADDRESS, - channel=channel, - credentials=credentials, + address=self.SERVICE_ADDRESS, channel=channel, credentials=credentials ) if client_info is None: client_info = google.api_core.gapic_v1.client_info.ClientInfo( - gapic_version=_GAPIC_LIBRARY_VERSION, ) + gapic_version=_GAPIC_LIBRARY_VERSION + ) else: client_info.gapic_version = _GAPIC_LIBRARY_VERSION self._client_info = client_info @@ -213,7 +216,8 @@ def __init__(self, # (Ordinarily, these are the defaults specified in the `*_config.py` # file next to this one.) self._method_configs = google.api_core.gapic_v1.config.parse_method_configs( - client_config['interfaces'][self._INTERFACE_NAME], ) + client_config["interfaces"][self._INTERFACE_NAME] + ) # Save a dictionary of cached API call functions. # These are the actual callables which invoke the proper @@ -222,12 +226,14 @@ def __init__(self, self._inner_api_calls = {} # Service calls - def create_source(self, - parent, - source, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + def create_source( + self, + parent, + source, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Creates a source. @@ -271,30 +277,32 @@ def create_source(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'create_source' not in self._inner_api_calls: + if "create_source" not in self._inner_api_calls: self._inner_api_calls[ - 'create_source'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.create_source, - default_retry=self._method_configs['CreateSource'].retry, - default_timeout=self._method_configs['CreateSource']. - timeout, - client_info=self._client_info, - ) + "create_source" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.create_source, + default_retry=self._method_configs["CreateSource"].retry, + default_timeout=self._method_configs["CreateSource"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.CreateSourceRequest( - parent=parent, - source=source, + parent=parent, source=source + ) + return self._inner_api_calls["create_source"]( + request, retry=retry, timeout=timeout, metadata=metadata ) - return self._inner_api_calls['create_source']( - request, retry=retry, timeout=timeout, metadata=metadata) - - def create_finding(self, - parent, - finding_id, - finding, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + + def create_finding( + self, + parent, + finding_id, + finding, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Creates a finding. The corresponding source must exist for finding creation to succeed. @@ -345,29 +353,30 @@ def create_finding(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'create_finding' not in self._inner_api_calls: + if "create_finding" not in self._inner_api_calls: self._inner_api_calls[ - 'create_finding'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.create_finding, - default_retry=self._method_configs['CreateFinding'].retry, - default_timeout=self._method_configs['CreateFinding']. - timeout, - client_info=self._client_info, - ) + "create_finding" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.create_finding, + default_retry=self._method_configs["CreateFinding"].retry, + default_timeout=self._method_configs["CreateFinding"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.CreateFindingRequest( - parent=parent, - finding_id=finding_id, - finding=finding, + parent=parent, finding_id=finding_id, finding=finding ) - return self._inner_api_calls['create_finding']( - request, retry=retry, timeout=timeout, metadata=metadata) - - def get_iam_policy(self, - resource, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + return self._inner_api_calls["create_finding"]( + request, retry=retry, timeout=timeout, metadata=metadata + ) + + def get_iam_policy( + self, + resource, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Gets the access control policy on the specified Source. @@ -404,26 +413,28 @@ def get_iam_policy(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'get_iam_policy' not in self._inner_api_calls: + if "get_iam_policy" not in self._inner_api_calls: self._inner_api_calls[ - 'get_iam_policy'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.get_iam_policy, - default_retry=self._method_configs['GetIamPolicy'].retry, - default_timeout=self._method_configs['GetIamPolicy']. - timeout, - client_info=self._client_info, - ) + "get_iam_policy" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.get_iam_policy, + default_retry=self._method_configs["GetIamPolicy"].retry, + default_timeout=self._method_configs["GetIamPolicy"].timeout, + client_info=self._client_info, + ) - request = iam_policy_pb2.GetIamPolicyRequest(resource=resource, ) - return self._inner_api_calls['get_iam_policy']( - request, retry=retry, timeout=timeout, metadata=metadata) + request = iam_policy_pb2.GetIamPolicyRequest(resource=resource) + return self._inner_api_calls["get_iam_policy"]( + request, retry=retry, timeout=timeout, metadata=metadata + ) def get_organization_settings( - self, - name, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + self, + name, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Gets the settings for an organization. @@ -459,27 +470,28 @@ def get_organization_settings( ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'get_organization_settings' not in self._inner_api_calls: + if "get_organization_settings" not in self._inner_api_calls: self._inner_api_calls[ - 'get_organization_settings'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.get_organization_settings, - default_retry=self. - _method_configs['GetOrganizationSettings'].retry, - default_timeout=self. - _method_configs['GetOrganizationSettings'].timeout, - client_info=self._client_info, - ) + "get_organization_settings" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.get_organization_settings, + default_retry=self._method_configs["GetOrganizationSettings"].retry, + default_timeout=self._method_configs["GetOrganizationSettings"].timeout, + client_info=self._client_info, + ) - request = securitycenter_service_pb2.GetOrganizationSettingsRequest( - name=name, ) - return self._inner_api_calls['get_organization_settings']( - request, retry=retry, timeout=timeout, metadata=metadata) + request = securitycenter_service_pb2.GetOrganizationSettingsRequest(name=name) + return self._inner_api_calls["get_organization_settings"]( + request, retry=retry, timeout=timeout, metadata=metadata + ) - def get_source(self, - name, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + def get_source( + self, + name, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Gets a source. @@ -515,29 +527,33 @@ def get_source(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'get_source' not in self._inner_api_calls: + if "get_source" not in self._inner_api_calls: self._inner_api_calls[ - 'get_source'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.get_source, - default_retry=self._method_configs['GetSource'].retry, - default_timeout=self._method_configs['GetSource'].timeout, - client_info=self._client_info, - ) + "get_source" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.get_source, + default_retry=self._method_configs["GetSource"].retry, + default_timeout=self._method_configs["GetSource"].timeout, + client_info=self._client_info, + ) + + request = securitycenter_service_pb2.GetSourceRequest(name=name) + return self._inner_api_calls["get_source"]( + request, retry=retry, timeout=timeout, metadata=metadata + ) - request = securitycenter_service_pb2.GetSourceRequest(name=name, ) - return self._inner_api_calls['get_source']( - request, retry=retry, timeout=timeout, metadata=metadata) - - def group_assets(self, - parent, - group_by, - filter_=None, - compare_duration=None, - read_time=None, - page_size=None, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + def group_assets( + self, + parent, + group_by, + filter_=None, + compare_duration=None, + read_time=None, + page_size=None, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Filters an organization's assets and groups them by their specified properties. @@ -671,15 +687,15 @@ def group_assets(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'group_assets' not in self._inner_api_calls: + if "group_assets" not in self._inner_api_calls: self._inner_api_calls[ - 'group_assets'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.group_assets, - default_retry=self._method_configs['GroupAssets'].retry, - default_timeout=self._method_configs['GroupAssets']. - timeout, - client_info=self._client_info, - ) + "group_assets" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.group_assets, + default_retry=self._method_configs["GroupAssets"].retry, + default_timeout=self._method_configs["GroupAssets"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.GroupAssetsRequest( parent=parent, @@ -692,26 +708,29 @@ def group_assets(self, iterator = google.api_core.page_iterator.GRPCIterator( client=None, method=functools.partial( - self._inner_api_calls['group_assets'], + self._inner_api_calls["group_assets"], retry=retry, timeout=timeout, - metadata=metadata), + metadata=metadata, + ), request=request, - items_field='group_by_results', - request_token_field='page_token', - response_token_field='next_page_token', + items_field="group_by_results", + request_token_field="page_token", + response_token_field="next_page_token", ) return iterator - def group_findings(self, - parent, - group_by, - filter_=None, - read_time=None, - page_size=None, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + def group_findings( + self, + parent, + group_by, + filter_=None, + read_time=None, + page_size=None, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Filters an organization or source's findings and groups them by their specified properties. @@ -818,15 +837,15 @@ def group_findings(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'group_findings' not in self._inner_api_calls: + if "group_findings" not in self._inner_api_calls: self._inner_api_calls[ - 'group_findings'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.group_findings, - default_retry=self._method_configs['GroupFindings'].retry, - default_timeout=self._method_configs['GroupFindings']. - timeout, - client_info=self._client_info, - ) + "group_findings" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.group_findings, + default_retry=self._method_configs["GroupFindings"].retry, + default_timeout=self._method_configs["GroupFindings"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.GroupFindingsRequest( parent=parent, @@ -838,28 +857,31 @@ def group_findings(self, iterator = google.api_core.page_iterator.GRPCIterator( client=None, method=functools.partial( - self._inner_api_calls['group_findings'], + self._inner_api_calls["group_findings"], retry=retry, timeout=timeout, - metadata=metadata), + metadata=metadata, + ), request=request, - items_field='group_by_results', - request_token_field='page_token', - response_token_field='next_page_token', + items_field="group_by_results", + request_token_field="page_token", + response_token_field="next_page_token", ) return iterator - def list_assets(self, - parent, - filter_=None, - order_by=None, - read_time=None, - compare_duration=None, - field_mask=None, - page_size=None, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + def list_assets( + self, + parent, + filter_=None, + order_by=None, + read_time=None, + compare_duration=None, + field_mask=None, + page_size=None, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Lists an organization's assets. @@ -993,14 +1015,15 @@ def list_assets(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'list_assets' not in self._inner_api_calls: + if "list_assets" not in self._inner_api_calls: self._inner_api_calls[ - 'list_assets'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.list_assets, - default_retry=self._method_configs['ListAssets'].retry, - default_timeout=self._method_configs['ListAssets'].timeout, - client_info=self._client_info, - ) + "list_assets" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.list_assets, + default_retry=self._method_configs["ListAssets"].retry, + default_timeout=self._method_configs["ListAssets"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.ListAssetsRequest( parent=parent, @@ -1014,27 +1037,30 @@ def list_assets(self, iterator = google.api_core.page_iterator.GRPCIterator( client=None, method=functools.partial( - self._inner_api_calls['list_assets'], + self._inner_api_calls["list_assets"], retry=retry, timeout=timeout, - metadata=metadata), + metadata=metadata, + ), request=request, - items_field='list_assets_results', - request_token_field='page_token', - response_token_field='next_page_token', + items_field="list_assets_results", + request_token_field="page_token", + response_token_field="next_page_token", ) return iterator - def list_findings(self, - parent, - filter_=None, - order_by=None, - read_time=None, - field_mask=None, - page_size=None, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + def list_findings( + self, + parent, + filter_=None, + order_by=None, + read_time=None, + field_mask=None, + page_size=None, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Lists an organization or source's findings. @@ -1142,15 +1168,15 @@ def list_findings(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'list_findings' not in self._inner_api_calls: + if "list_findings" not in self._inner_api_calls: self._inner_api_calls[ - 'list_findings'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.list_findings, - default_retry=self._method_configs['ListFindings'].retry, - default_timeout=self._method_configs['ListFindings']. - timeout, - client_info=self._client_info, - ) + "list_findings" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.list_findings, + default_retry=self._method_configs["ListFindings"].retry, + default_timeout=self._method_configs["ListFindings"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.ListFindingsRequest( parent=parent, @@ -1163,23 +1189,26 @@ def list_findings(self, iterator = google.api_core.page_iterator.GRPCIterator( client=None, method=functools.partial( - self._inner_api_calls['list_findings'], + self._inner_api_calls["list_findings"], retry=retry, timeout=timeout, - metadata=metadata), + metadata=metadata, + ), request=request, - items_field='findings', - request_token_field='page_token', - response_token_field='next_page_token', + items_field="findings", + request_token_field="page_token", + response_token_field="next_page_token", ) return iterator - def list_sources(self, - parent, - page_size=None, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + def list_sources( + self, + parent, + page_size=None, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Lists all sources belonging to an organization. @@ -1235,39 +1264,41 @@ def list_sources(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'list_sources' not in self._inner_api_calls: + if "list_sources" not in self._inner_api_calls: self._inner_api_calls[ - 'list_sources'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.list_sources, - default_retry=self._method_configs['ListSources'].retry, - default_timeout=self._method_configs['ListSources']. - timeout, - client_info=self._client_info, - ) + "list_sources" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.list_sources, + default_retry=self._method_configs["ListSources"].retry, + default_timeout=self._method_configs["ListSources"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.ListSourcesRequest( - parent=parent, - page_size=page_size, + parent=parent, page_size=page_size ) iterator = google.api_core.page_iterator.GRPCIterator( client=None, method=functools.partial( - self._inner_api_calls['list_sources'], + self._inner_api_calls["list_sources"], retry=retry, timeout=timeout, - metadata=metadata), + metadata=metadata, + ), request=request, - items_field='sources', - request_token_field='page_token', - response_token_field='next_page_token', + items_field="sources", + request_token_field="page_token", + response_token_field="next_page_token", ) return iterator - def run_asset_discovery(self, - parent, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + def run_asset_discovery( + self, + parent, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Runs asset discovery. The discovery is tracked with a long-running operation. @@ -1317,21 +1348,20 @@ def run_asset_discovery(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'run_asset_discovery' not in self._inner_api_calls: + if "run_asset_discovery" not in self._inner_api_calls: self._inner_api_calls[ - 'run_asset_discovery'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.run_asset_discovery, - default_retry=self._method_configs['RunAssetDiscovery']. - retry, - default_timeout=self._method_configs['RunAssetDiscovery']. - timeout, - client_info=self._client_info, - ) + "run_asset_discovery" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.run_asset_discovery, + default_retry=self._method_configs["RunAssetDiscovery"].retry, + default_timeout=self._method_configs["RunAssetDiscovery"].timeout, + client_info=self._client_info, + ) - request = securitycenter_service_pb2.RunAssetDiscoveryRequest( - parent=parent, ) - operation = self._inner_api_calls['run_asset_discovery']( - request, retry=retry, timeout=timeout, metadata=metadata) + request = securitycenter_service_pb2.RunAssetDiscoveryRequest(parent=parent) + operation = self._inner_api_calls["run_asset_discovery"]( + request, retry=retry, timeout=timeout, metadata=metadata + ) return google.api_core.operation.from_gapic( operation, self.transport._operations_client, @@ -1339,13 +1369,15 @@ def run_asset_discovery(self, metadata_type=empty_pb2.Empty, ) - def set_finding_state(self, - name, - state, - start_time, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + def set_finding_state( + self, + name, + state, + start_time, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Updates the state of a finding. @@ -1394,31 +1426,31 @@ def set_finding_state(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'set_finding_state' not in self._inner_api_calls: + if "set_finding_state" not in self._inner_api_calls: self._inner_api_calls[ - 'set_finding_state'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.set_finding_state, - default_retry=self._method_configs['SetFindingState']. - retry, - default_timeout=self._method_configs['SetFindingState']. - timeout, - client_info=self._client_info, - ) + "set_finding_state" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.set_finding_state, + default_retry=self._method_configs["SetFindingState"].retry, + default_timeout=self._method_configs["SetFindingState"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.SetFindingStateRequest( - name=name, - state=state, - start_time=start_time, + name=name, state=state, start_time=start_time + ) + return self._inner_api_calls["set_finding_state"]( + request, retry=retry, timeout=timeout, metadata=metadata ) - return self._inner_api_calls['set_finding_state']( - request, retry=retry, timeout=timeout, metadata=metadata) - - def set_iam_policy(self, - resource, - policy, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + + def set_iam_policy( + self, + resource, + policy, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Sets the access control policy on the specified Source. @@ -1465,29 +1497,29 @@ def set_iam_policy(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'set_iam_policy' not in self._inner_api_calls: + if "set_iam_policy" not in self._inner_api_calls: self._inner_api_calls[ - 'set_iam_policy'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.set_iam_policy, - default_retry=self._method_configs['SetIamPolicy'].retry, - default_timeout=self._method_configs['SetIamPolicy']. - timeout, - client_info=self._client_info, - ) + "set_iam_policy" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.set_iam_policy, + default_retry=self._method_configs["SetIamPolicy"].retry, + default_timeout=self._method_configs["SetIamPolicy"].timeout, + client_info=self._client_info, + ) - request = iam_policy_pb2.SetIamPolicyRequest( - resource=resource, - policy=policy, + request = iam_policy_pb2.SetIamPolicyRequest(resource=resource, policy=policy) + return self._inner_api_calls["set_iam_policy"]( + request, retry=retry, timeout=timeout, metadata=metadata ) - return self._inner_api_calls['set_iam_policy']( - request, retry=retry, timeout=timeout, metadata=metadata) - - def test_iam_permissions(self, - resource, - permissions, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + + def test_iam_permissions( + self, + resource, + permissions, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Returns the permissions that a caller has on the specified source. @@ -1531,30 +1563,31 @@ def test_iam_permissions(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'test_iam_permissions' not in self._inner_api_calls: + if "test_iam_permissions" not in self._inner_api_calls: self._inner_api_calls[ - 'test_iam_permissions'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.test_iam_permissions, - default_retry=self._method_configs['TestIamPermissions']. - retry, - default_timeout=self._method_configs['TestIamPermissions']. - timeout, - client_info=self._client_info, - ) + "test_iam_permissions" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.test_iam_permissions, + default_retry=self._method_configs["TestIamPermissions"].retry, + default_timeout=self._method_configs["TestIamPermissions"].timeout, + client_info=self._client_info, + ) request = iam_policy_pb2.TestIamPermissionsRequest( - resource=resource, - permissions=permissions, + resource=resource, permissions=permissions + ) + return self._inner_api_calls["test_iam_permissions"]( + request, retry=retry, timeout=timeout, metadata=metadata ) - return self._inner_api_calls['test_iam_permissions']( - request, retry=retry, timeout=timeout, metadata=metadata) - - def update_finding(self, - finding, - update_mask=None, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + + def update_finding( + self, + finding, + update_mask=None, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Creates or updates a finding. The corresponding source must exist for a finding creation to succeed. @@ -1604,30 +1637,31 @@ def update_finding(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'update_finding' not in self._inner_api_calls: + if "update_finding" not in self._inner_api_calls: self._inner_api_calls[ - 'update_finding'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.update_finding, - default_retry=self._method_configs['UpdateFinding'].retry, - default_timeout=self._method_configs['UpdateFinding']. - timeout, - client_info=self._client_info, - ) + "update_finding" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.update_finding, + default_retry=self._method_configs["UpdateFinding"].retry, + default_timeout=self._method_configs["UpdateFinding"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.UpdateFindingRequest( - finding=finding, - update_mask=update_mask, + finding=finding, update_mask=update_mask + ) + return self._inner_api_calls["update_finding"]( + request, retry=retry, timeout=timeout, metadata=metadata ) - return self._inner_api_calls['update_finding']( - request, retry=retry, timeout=timeout, metadata=metadata) def update_organization_settings( - self, - organization_settings, - update_mask=None, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + self, + organization_settings, + update_mask=None, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Updates an organization's settings. @@ -1670,30 +1704,33 @@ def update_organization_settings( ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'update_organization_settings' not in self._inner_api_calls: + if "update_organization_settings" not in self._inner_api_calls: self._inner_api_calls[ - 'update_organization_settings'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.update_organization_settings, - default_retry=self. - _method_configs['UpdateOrganizationSettings'].retry, - default_timeout=self. - _method_configs['UpdateOrganizationSettings'].timeout, - client_info=self._client_info, - ) + "update_organization_settings" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.update_organization_settings, + default_retry=self._method_configs["UpdateOrganizationSettings"].retry, + default_timeout=self._method_configs[ + "UpdateOrganizationSettings" + ].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.UpdateOrganizationSettingsRequest( - organization_settings=organization_settings, - update_mask=update_mask, + organization_settings=organization_settings, update_mask=update_mask ) - return self._inner_api_calls['update_organization_settings']( - request, retry=retry, timeout=timeout, metadata=metadata) - - def update_source(self, - source, - update_mask=None, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + return self._inner_api_calls["update_organization_settings"]( + request, retry=retry, timeout=timeout, metadata=metadata + ) + + def update_source( + self, + source, + update_mask=None, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Updates a source. @@ -1736,30 +1773,32 @@ def update_source(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'update_source' not in self._inner_api_calls: + if "update_source" not in self._inner_api_calls: self._inner_api_calls[ - 'update_source'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.update_source, - default_retry=self._method_configs['UpdateSource'].retry, - default_timeout=self._method_configs['UpdateSource']. - timeout, - client_info=self._client_info, - ) + "update_source" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.update_source, + default_retry=self._method_configs["UpdateSource"].retry, + default_timeout=self._method_configs["UpdateSource"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.UpdateSourceRequest( - source=source, - update_mask=update_mask, + source=source, update_mask=update_mask + ) + return self._inner_api_calls["update_source"]( + request, retry=retry, timeout=timeout, metadata=metadata ) - return self._inner_api_calls['update_source']( - request, retry=retry, timeout=timeout, metadata=metadata) - - def update_security_marks(self, - security_marks, - update_mask=None, - start_time=None, - retry=google.api_core.gapic_v1.method.DEFAULT, - timeout=google.api_core.gapic_v1.method.DEFAULT, - metadata=None): + + def update_security_marks( + self, + security_marks, + update_mask=None, + start_time=None, + retry=google.api_core.gapic_v1.method.DEFAULT, + timeout=google.api_core.gapic_v1.method.DEFAULT, + metadata=None, + ): """ Updates security marks. @@ -1806,21 +1845,21 @@ def update_security_marks(self, ValueError: If the parameters are invalid. """ # Wrap the transport method to add retry and timeout logic. - if 'update_security_marks' not in self._inner_api_calls: + if "update_security_marks" not in self._inner_api_calls: self._inner_api_calls[ - 'update_security_marks'] = google.api_core.gapic_v1.method.wrap_method( - self.transport.update_security_marks, - default_retry=self._method_configs['UpdateSecurityMarks']. - retry, - default_timeout=self. - _method_configs['UpdateSecurityMarks'].timeout, - client_info=self._client_info, - ) + "update_security_marks" + ] = google.api_core.gapic_v1.method.wrap_method( + self.transport.update_security_marks, + default_retry=self._method_configs["UpdateSecurityMarks"].retry, + default_timeout=self._method_configs["UpdateSecurityMarks"].timeout, + client_info=self._client_info, + ) request = securitycenter_service_pb2.UpdateSecurityMarksRequest( security_marks=security_marks, update_mask=update_mask, start_time=start_time, ) - return self._inner_api_calls['update_security_marks']( - request, retry=retry, timeout=timeout, metadata=metadata) + return self._inner_api_calls["update_security_marks"]( + request, retry=retry, timeout=timeout, metadata=metadata + ) diff --git a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/security_center_client_config.py b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/security_center_client_config.py index e9444324e9b4..9b5c01a985a6 100644 --- a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/security_center_client_config.py +++ b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/security_center_client_config.py @@ -3,7 +3,7 @@ "google.cloud.securitycenter.v1beta1.SecurityCenter": { "retry_codes": { "idempotent": ["DEADLINE_EXCEEDED", "UNAVAILABLE"], - "non_idempotent": [] + "non_idempotent": [], }, "retry_params": { "default": { @@ -13,101 +13,101 @@ "initial_rpc_timeout_millis": 20000, "rpc_timeout_multiplier": 1.0, "max_rpc_timeout_millis": 20000, - "total_timeout_millis": 600000 + "total_timeout_millis": 600000, } }, "methods": { "CreateSource": { "timeout_millis": 60000, "retry_codes_name": "non_idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "CreateFinding": { "timeout_millis": 60000, "retry_codes_name": "non_idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "GetIamPolicy": { "timeout_millis": 60000, "retry_codes_name": "idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "GetOrganizationSettings": { "timeout_millis": 60000, "retry_codes_name": "idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "GetSource": { "timeout_millis": 60000, "retry_codes_name": "idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "GroupAssets": { "timeout_millis": 60000, "retry_codes_name": "idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "GroupFindings": { "timeout_millis": 60000, "retry_codes_name": "idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "ListAssets": { "timeout_millis": 60000, "retry_codes_name": "idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "ListFindings": { "timeout_millis": 60000, "retry_codes_name": "idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "ListSources": { "timeout_millis": 60000, "retry_codes_name": "idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "RunAssetDiscovery": { "timeout_millis": 60000, "retry_codes_name": "non_idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "SetFindingState": { "timeout_millis": 60000, "retry_codes_name": "non_idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "SetIamPolicy": { "timeout_millis": 60000, "retry_codes_name": "non_idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "TestIamPermissions": { "timeout_millis": 60000, "retry_codes_name": "idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "UpdateFinding": { "timeout_millis": 60000, "retry_codes_name": "non_idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "UpdateOrganizationSettings": { "timeout_millis": 60000, "retry_codes_name": "non_idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "UpdateSource": { "timeout_millis": 60000, "retry_codes_name": "non_idempotent", - "retry_params_name": "default" + "retry_params_name": "default", }, "UpdateSecurityMarks": { "timeout_millis": 60000, "retry_codes_name": "non_idempotent", - "retry_params_name": "default" - } - } + "retry_params_name": "default", + }, + }, } } } diff --git a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/transports/security_center_grpc_transport.py b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/transports/security_center_grpc_transport.py index f6e1fe349122..9005190edaf4 100644 --- a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/transports/security_center_grpc_transport.py +++ b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/gapic/transports/security_center_grpc_transport.py @@ -28,14 +28,17 @@ class SecurityCenterGrpcTransport(object): which can be used to take advantage of advanced features of gRPC. """ + # The scopes needed to make gRPC calls to all of the methods defined # in this service. - _OAUTH_SCOPES = ('https://www.googleapis.com/auth/cloud-platform', ) - - def __init__(self, - channel=None, - credentials=None, - address='securitycenter.googleapis.com:443'): + _OAUTH_SCOPES = ("https://www.googleapis.com/auth/cloud-platform",) + + def __init__( + self, + channel=None, + credentials=None, + address="securitycenter.googleapis.com:443", + ): """Instantiate the transport class. Args: @@ -53,35 +56,34 @@ def __init__(self, # exception (channels come with credentials baked in already). if channel is not None and credentials is not None: raise ValueError( - 'The `channel` and `credentials` arguments are mutually ' - 'exclusive.', ) + "The `channel` and `credentials` arguments are mutually " "exclusive." + ) # Create the channel. if channel is None: - channel = self.create_channel( - address=address, - credentials=credentials, - ) + channel = self.create_channel(address=address, credentials=credentials) self._channel = channel # gRPC uses objects called "stubs" that are bound to the # channel and provide a basic method for each RPC. self._stubs = { - 'security_center_stub': - securitycenter_service_pb2_grpc.SecurityCenterStub(channel), + "security_center_stub": securitycenter_service_pb2_grpc.SecurityCenterStub( + channel + ) } # Because this API includes a method that returns a # long-running operation (proto: google.longrunning.Operation), # instantiate an LRO client. self._operations_client = google.api_core.operations_v1.OperationsClient( - channel) + channel + ) @classmethod - def create_channel(cls, - address='securitycenter.googleapis.com:443', - credentials=None): + def create_channel( + cls, address="securitycenter.googleapis.com:443", credentials=None + ): """Create and return a gRPC channel object. Args: @@ -96,9 +98,7 @@ def create_channel(cls, grpc.Channel: A gRPC channel object. """ return google.api_core.grpc_helpers.create_channel( - address, - credentials=credentials, - scopes=cls._OAUTH_SCOPES, + address, credentials=credentials, scopes=cls._OAUTH_SCOPES ) @property @@ -121,7 +121,7 @@ def create_source(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].CreateSource + return self._stubs["security_center_stub"].CreateSource @property def create_finding(self): @@ -135,7 +135,7 @@ def create_finding(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].CreateFinding + return self._stubs["security_center_stub"].CreateFinding @property def get_iam_policy(self): @@ -148,7 +148,7 @@ def get_iam_policy(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].GetIamPolicy + return self._stubs["security_center_stub"].GetIamPolicy @property def get_organization_settings(self): @@ -161,7 +161,7 @@ def get_organization_settings(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].GetOrganizationSettings + return self._stubs["security_center_stub"].GetOrganizationSettings @property def get_source(self): @@ -174,7 +174,7 @@ def get_source(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].GetSource + return self._stubs["security_center_stub"].GetSource @property def group_assets(self): @@ -188,7 +188,7 @@ def group_assets(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].GroupAssets + return self._stubs["security_center_stub"].GroupAssets @property def group_findings(self): @@ -205,7 +205,7 @@ def group_findings(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].GroupFindings + return self._stubs["security_center_stub"].GroupFindings @property def list_assets(self): @@ -218,7 +218,7 @@ def list_assets(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].ListAssets + return self._stubs["security_center_stub"].ListAssets @property def list_findings(self): @@ -234,7 +234,7 @@ def list_findings(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].ListFindings + return self._stubs["security_center_stub"].ListFindings @property def list_sources(self): @@ -247,7 +247,7 @@ def list_sources(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].ListSources + return self._stubs["security_center_stub"].ListSources @property def run_asset_discovery(self): @@ -265,7 +265,7 @@ def run_asset_discovery(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].RunAssetDiscovery + return self._stubs["security_center_stub"].RunAssetDiscovery @property def set_finding_state(self): @@ -278,7 +278,7 @@ def set_finding_state(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].SetFindingState + return self._stubs["security_center_stub"].SetFindingState @property def set_iam_policy(self): @@ -291,7 +291,7 @@ def set_iam_policy(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].SetIamPolicy + return self._stubs["security_center_stub"].SetIamPolicy @property def test_iam_permissions(self): @@ -304,7 +304,7 @@ def test_iam_permissions(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].TestIamPermissions + return self._stubs["security_center_stub"].TestIamPermissions @property def update_finding(self): @@ -318,7 +318,7 @@ def update_finding(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].UpdateFinding + return self._stubs["security_center_stub"].UpdateFinding @property def update_organization_settings(self): @@ -331,7 +331,7 @@ def update_organization_settings(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].UpdateOrganizationSettings + return self._stubs["security_center_stub"].UpdateOrganizationSettings @property def update_source(self): @@ -344,7 +344,7 @@ def update_source(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].UpdateSource + return self._stubs["security_center_stub"].UpdateSource @property def update_security_marks(self): @@ -357,4 +357,4 @@ def update_security_marks(self): deserialized request object and returns a deserialized response object. """ - return self._stubs['security_center_stub'].UpdateSecurityMarks + return self._stubs["security_center_stub"].UpdateSecurityMarks diff --git a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/types.py b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/types.py index 3a249f2b9e61..9e9b4898a39c 100644 --- a/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/types.py +++ b/packages/google-cloud-securitycenter/google/cloud/securitycenter_v1beta1/types.py @@ -70,7 +70,7 @@ names.append(name) for module in _local_modules: for name, message in get_messages(module).items(): - message.__module__ = 'google.cloud.securitycenter_v1beta1.types' + message.__module__ = "google.cloud.securitycenter_v1beta1.types" setattr(sys.modules[__name__], name, message) names.append(name) diff --git a/packages/google-cloud-securitycenter/tests/unit/gapic/v1beta1/test_security_center_client_v1beta1.py b/packages/google-cloud-securitycenter/tests/unit/gapic/v1beta1/test_security_center_client_v1beta1.py index 0e8e8c7241f6..d68eaf8e54c4 100644 --- a/packages/google-cloud-securitycenter/tests/unit/gapic/v1beta1/test_security_center_client_v1beta1.py +++ b/packages/google-cloud-securitycenter/tests/unit/gapic/v1beta1/test_security_center_client_v1beta1.py @@ -62,10 +62,7 @@ def __init__(self, responses=[]): self.responses = responses self.requests = [] - def unary_unary(self, - method, - request_serializer=None, - response_deserializer=None): + def unary_unary(self, method, request_serializer=None, response_deserializer=None): return MultiCallableStub(method, self) @@ -76,25 +73,25 @@ class CustomException(Exception): class TestSecurityCenterClient(object): def test_create_source(self): # Setup Expected Response - name = 'name3373707' - display_name = 'displayName1615086568' - description = 'description-1724546052' + name = "name3373707" + display_name = "displayName1615086568" + description = "description-1724546052" expected_response = { - 'name': name, - 'display_name': display_name, - 'description': description + "name": name, + "display_name": display_name, + "description": description, } expected_response = source_pb2.Source(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - parent = client.organization_path('[ORGANIZATION]') + parent = client.organization_path("[ORGANIZATION]") source = {} response = client.create_source(parent, source) @@ -102,20 +99,21 @@ def test_create_source(self): assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.CreateSourceRequest( - parent=parent, source=source) + parent=parent, source=source + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_create_source_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - parent = client.organization_path('[ORGANIZATION]') + parent = client.organization_path("[ORGANIZATION]") source = {} with pytest.raises(CustomException): @@ -123,30 +121,30 @@ def test_create_source_exception(self): def test_create_finding(self): # Setup Expected Response - name = 'name3373707' - parent_2 = 'parent21175163357' - resource_name = 'resourceName979421212' - category = 'category50511102' - external_uri = 'externalUri-1385596168' + name = "name3373707" + parent_2 = "parent21175163357" + resource_name = "resourceName979421212" + category = "category50511102" + external_uri = "externalUri-1385596168" expected_response = { - 'name': name, - 'parent': parent_2, - 'resource_name': resource_name, - 'category': category, - 'external_uri': external_uri + "name": name, + "parent": parent_2, + "resource_name": resource_name, + "category": category, + "external_uri": external_uri, } expected_response = finding_pb2.Finding(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - parent = client.source_path('[ORGANIZATION]', '[SOURCE]') - finding_id = 'findingId728776081' + parent = client.source_path("[ORGANIZATION]", "[SOURCE]") + finding_id = "findingId728776081" finding = {} response = client.create_finding(parent, finding_id, finding) @@ -154,21 +152,22 @@ def test_create_finding(self): assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.CreateFindingRequest( - parent=parent, finding_id=finding_id, finding=finding) + parent=parent, finding_id=finding_id, finding=finding + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_create_finding_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - parent = client.source_path('[ORGANIZATION]', '[SOURCE]') - finding_id = 'findingId728776081' + parent = client.source_path("[ORGANIZATION]", "[SOURCE]") + finding_id = "findingId728776081" finding = {} with pytest.raises(CustomException): @@ -177,154 +176,155 @@ def test_create_finding_exception(self): def test_get_iam_policy(self): # Setup Expected Response version = 351608024 - etag = b'21' - expected_response = {'version': version, 'etag': etag} + etag = b"21" + expected_response = {"version": version, "etag": etag} expected_response = policy_pb2.Policy(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - resource = client.source_path('[ORGANIZATION]', '[SOURCE]') + resource = client.source_path("[ORGANIZATION]", "[SOURCE]") response = client.get_iam_policy(resource) assert expected_response == response assert len(channel.requests) == 1 - expected_request = iam_policy_pb2.GetIamPolicyRequest( - resource=resource) + expected_request = iam_policy_pb2.GetIamPolicyRequest(resource=resource) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_get_iam_policy_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - resource = client.source_path('[ORGANIZATION]', '[SOURCE]') + resource = client.source_path("[ORGANIZATION]", "[SOURCE]") with pytest.raises(CustomException): client.get_iam_policy(resource) def test_get_organization_settings(self): # Setup Expected Response - name_2 = 'name2-1052831874' + name_2 = "name2-1052831874" enable_asset_discovery = False expected_response = { - 'name': name_2, - 'enable_asset_discovery': enable_asset_discovery + "name": name_2, + "enable_asset_discovery": enable_asset_discovery, } expected_response = organization_settings_pb2.OrganizationSettings( - **expected_response) + **expected_response + ) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - name = client.organization_settings_path('[ORGANIZATION]') + name = client.organization_settings_path("[ORGANIZATION]") response = client.get_organization_settings(name) assert expected_response == response assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.GetOrganizationSettingsRequest( - name=name) + name=name + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_get_organization_settings_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - name = client.organization_settings_path('[ORGANIZATION]') + name = client.organization_settings_path("[ORGANIZATION]") with pytest.raises(CustomException): client.get_organization_settings(name) def test_get_source(self): # Setup Expected Response - name_2 = 'name2-1052831874' - display_name = 'displayName1615086568' - description = 'description-1724546052' + name_2 = "name2-1052831874" + display_name = "displayName1615086568" + description = "description-1724546052" expected_response = { - 'name': name_2, - 'display_name': display_name, - 'description': description + "name": name_2, + "display_name": display_name, + "description": description, } expected_response = source_pb2.Source(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - name = client.source_path('[ORGANIZATION]', '[SOURCE]') + name = client.source_path("[ORGANIZATION]", "[SOURCE]") response = client.get_source(name) assert expected_response == response assert len(channel.requests) == 1 - expected_request = securitycenter_service_pb2.GetSourceRequest( - name=name) + expected_request = securitycenter_service_pb2.GetSourceRequest(name=name) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_get_source_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - name = client.source_path('[ORGANIZATION]', '[SOURCE]') + name = client.source_path("[ORGANIZATION]", "[SOURCE]") with pytest.raises(CustomException): client.get_source(name) def test_group_assets(self): # Setup Expected Response - next_page_token = '' + next_page_token = "" group_by_results_element = {} group_by_results = [group_by_results_element] expected_response = { - 'next_page_token': next_page_token, - 'group_by_results': group_by_results + "next_page_token": next_page_token, + "group_by_results": group_by_results, } expected_response = securitycenter_service_pb2.GroupAssetsResponse( - **expected_response) + **expected_response + ) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - parent = client.organization_path('[ORGANIZATION]') - group_by = 'groupBy506361367' + parent = client.organization_path("[ORGANIZATION]") + group_by = "groupBy506361367" paged_list_response = client.group_assets(parent, group_by) resources = list(paged_list_response) @@ -334,20 +334,21 @@ def test_group_assets(self): assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.GroupAssetsRequest( - parent=parent, group_by=group_by) + parent=parent, group_by=group_by + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_group_assets_exception(self): channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - parent = client.organization_path('[ORGANIZATION]') - group_by = 'groupBy506361367' + parent = client.organization_path("[ORGANIZATION]") + group_by = "groupBy506361367" paged_list_response = client.group_assets(parent, group_by) with pytest.raises(CustomException): @@ -355,26 +356,27 @@ def test_group_assets_exception(self): def test_group_findings(self): # Setup Expected Response - next_page_token = '' + next_page_token = "" group_by_results_element = {} group_by_results = [group_by_results_element] expected_response = { - 'next_page_token': next_page_token, - 'group_by_results': group_by_results + "next_page_token": next_page_token, + "group_by_results": group_by_results, } expected_response = securitycenter_service_pb2.GroupFindingsResponse( - **expected_response) + **expected_response + ) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - parent = client.source_path('[ORGANIZATION]', '[SOURCE]') - group_by = 'groupBy506361367' + parent = client.source_path("[ORGANIZATION]", "[SOURCE]") + group_by = "groupBy506361367" paged_list_response = client.group_findings(parent, group_by) resources = list(paged_list_response) @@ -384,20 +386,21 @@ def test_group_findings(self): assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.GroupFindingsRequest( - parent=parent, group_by=group_by) + parent=parent, group_by=group_by + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_group_findings_exception(self): channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - parent = client.source_path('[ORGANIZATION]', '[SOURCE]') - group_by = 'groupBy506361367' + parent = client.source_path("[ORGANIZATION]", "[SOURCE]") + group_by = "groupBy506361367" paged_list_response = client.group_findings(parent, group_by) with pytest.raises(CustomException): @@ -405,27 +408,28 @@ def test_group_findings_exception(self): def test_list_assets(self): # Setup Expected Response - next_page_token = '' + next_page_token = "" total_size = 705419236 list_assets_results_element = {} list_assets_results = [list_assets_results_element] expected_response = { - 'next_page_token': next_page_token, - 'total_size': total_size, - 'list_assets_results': list_assets_results + "next_page_token": next_page_token, + "total_size": total_size, + "list_assets_results": list_assets_results, } expected_response = securitycenter_service_pb2.ListAssetsResponse( - **expected_response) + **expected_response + ) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - parent = client.organization_path('[ORGANIZATION]') + parent = client.organization_path("[ORGANIZATION]") paged_list_response = client.list_assets(parent) resources = list(paged_list_response) @@ -434,20 +438,19 @@ def test_list_assets(self): assert expected_response.list_assets_results[0] == resources[0] assert len(channel.requests) == 1 - expected_request = securitycenter_service_pb2.ListAssetsRequest( - parent=parent) + expected_request = securitycenter_service_pb2.ListAssetsRequest(parent=parent) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_list_assets_exception(self): channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - parent = client.organization_path('[ORGANIZATION]') + parent = client.organization_path("[ORGANIZATION]") paged_list_response = client.list_assets(parent) with pytest.raises(CustomException): @@ -455,27 +458,28 @@ def test_list_assets_exception(self): def test_list_findings(self): # Setup Expected Response - next_page_token = '' + next_page_token = "" total_size = 705419236 findings_element = {} findings = [findings_element] expected_response = { - 'next_page_token': next_page_token, - 'total_size': total_size, - 'findings': findings + "next_page_token": next_page_token, + "total_size": total_size, + "findings": findings, } expected_response = securitycenter_service_pb2.ListFindingsResponse( - **expected_response) + **expected_response + ) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - parent = client.source_path('[ORGANIZATION]', '[SOURCE]') + parent = client.source_path("[ORGANIZATION]", "[SOURCE]") paged_list_response = client.list_findings(parent) resources = list(paged_list_response) @@ -484,20 +488,19 @@ def test_list_findings(self): assert expected_response.findings[0] == resources[0] assert len(channel.requests) == 1 - expected_request = securitycenter_service_pb2.ListFindingsRequest( - parent=parent) + expected_request = securitycenter_service_pb2.ListFindingsRequest(parent=parent) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_list_findings_exception(self): channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - parent = client.source_path('[ORGANIZATION]', '[SOURCE]') + parent = client.source_path("[ORGANIZATION]", "[SOURCE]") paged_list_response = client.list_findings(parent) with pytest.raises(CustomException): @@ -505,25 +508,23 @@ def test_list_findings_exception(self): def test_list_sources(self): # Setup Expected Response - next_page_token = '' + next_page_token = "" sources_element = {} sources = [sources_element] - expected_response = { - 'next_page_token': next_page_token, - 'sources': sources - } + expected_response = {"next_page_token": next_page_token, "sources": sources} expected_response = securitycenter_service_pb2.ListSourcesResponse( - **expected_response) + **expected_response + ) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - parent = client.organization_path('[ORGANIZATION]') + parent = client.organization_path("[ORGANIZATION]") paged_list_response = client.list_sources(parent) resources = list(paged_list_response) @@ -532,20 +533,19 @@ def test_list_sources(self): assert expected_response.sources[0] == resources[0] assert len(channel.requests) == 1 - expected_request = securitycenter_service_pb2.ListSourcesRequest( - parent=parent) + expected_request = securitycenter_service_pb2.ListSourcesRequest(parent=parent) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_list_sources_exception(self): channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - parent = client.organization_path('[ORGANIZATION]') + parent = client.organization_path("[ORGANIZATION]") paged_list_response = client.list_sources(parent) with pytest.raises(CustomException): @@ -556,18 +556,19 @@ def test_run_asset_discovery(self): expected_response = {} expected_response = empty_pb2.Empty(**expected_response) operation = operations_pb2.Operation( - name='operations/test_run_asset_discovery', done=True) + name="operations/test_run_asset_discovery", done=True + ) operation.response.Pack(expected_response) # Mock the API response channel = ChannelStub(responses=[operation]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - parent = client.organization_path('[ORGANIZATION]') + parent = client.organization_path("[ORGANIZATION]") response = client.run_asset_discovery(parent) result = response.result() @@ -575,7 +576,8 @@ def test_run_asset_discovery(self): assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.RunAssetDiscoveryRequest( - parent=parent) + parent=parent + ) actual_request = channel.requests[0][1] assert expected_request == actual_request @@ -583,18 +585,19 @@ def test_run_asset_discovery_exception(self): # Setup Response error = status_pb2.Status() operation = operations_pb2.Operation( - name='operations/test_run_asset_discovery_exception', done=True) + name="operations/test_run_asset_discovery_exception", done=True + ) operation.error.CopyFrom(error) # Mock the API response channel = ChannelStub(responses=[operation]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - parent = client.organization_path('[ORGANIZATION]') + parent = client.organization_path("[ORGANIZATION]") response = client.run_asset_discovery(parent) exception = response.exception() @@ -602,29 +605,29 @@ def test_run_asset_discovery_exception(self): def test_set_finding_state(self): # Setup Expected Response - name_2 = 'name2-1052831874' - parent = 'parent-995424086' - resource_name = 'resourceName979421212' - category = 'category50511102' - external_uri = 'externalUri-1385596168' + name_2 = "name2-1052831874" + parent = "parent-995424086" + resource_name = "resourceName979421212" + category = "category50511102" + external_uri = "externalUri-1385596168" expected_response = { - 'name': name_2, - 'parent': parent, - 'resource_name': resource_name, - 'category': category, - 'external_uri': external_uri + "name": name_2, + "parent": parent, + "resource_name": resource_name, + "category": category, + "external_uri": external_uri, } expected_response = finding_pb2.Finding(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - name = client.finding_path('[ORGANIZATION]', '[SOURCE]', '[FINDING]') + name = client.finding_path("[ORGANIZATION]", "[SOURCE]", "[FINDING]") state = enums.Finding.State.STATE_UNSPECIFIED start_time = {} @@ -633,20 +636,21 @@ def test_set_finding_state(self): assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.SetFindingStateRequest( - name=name, state=state, start_time=start_time) + name=name, state=state, start_time=start_time + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_set_finding_state_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - name = client.finding_path('[ORGANIZATION]', '[SOURCE]', '[FINDING]') + name = client.finding_path("[ORGANIZATION]", "[SOURCE]", "[FINDING]") state = enums.Finding.State.STATE_UNSPECIFIED start_time = {} @@ -656,19 +660,19 @@ def test_set_finding_state_exception(self): def test_set_iam_policy(self): # Setup Expected Response version = 351608024 - etag = b'21' - expected_response = {'version': version, 'etag': etag} + etag = b"21" + expected_response = {"version": version, "etag": etag} expected_response = policy_pb2.Policy(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - resource = client.source_path('[ORGANIZATION]', '[SOURCE]') + resource = client.source_path("[ORGANIZATION]", "[SOURCE]") policy = {} response = client.set_iam_policy(resource, policy) @@ -676,20 +680,21 @@ def test_set_iam_policy(self): assert len(channel.requests) == 1 expected_request = iam_policy_pb2.SetIamPolicyRequest( - resource=resource, policy=policy) + resource=resource, policy=policy + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_set_iam_policy_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - resource = client.source_path('[ORGANIZATION]', '[SOURCE]') + resource = client.source_path("[ORGANIZATION]", "[SOURCE]") policy = {} with pytest.raises(CustomException): @@ -699,17 +704,18 @@ def test_test_iam_permissions(self): # Setup Expected Response expected_response = {} expected_response = iam_policy_pb2.TestIamPermissionsResponse( - **expected_response) + **expected_response + ) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup Request - resource = client.source_path('[ORGANIZATION]', '[SOURCE]') + resource = client.source_path("[ORGANIZATION]", "[SOURCE]") permissions = [] response = client.test_iam_permissions(resource, permissions) @@ -717,20 +723,21 @@ def test_test_iam_permissions(self): assert len(channel.requests) == 1 expected_request = iam_policy_pb2.TestIamPermissionsRequest( - resource=resource, permissions=permissions) + resource=resource, permissions=permissions + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_test_iam_permissions_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() # Setup request - resource = client.source_path('[ORGANIZATION]', '[SOURCE]') + resource = client.source_path("[ORGANIZATION]", "[SOURCE]") permissions = [] with pytest.raises(CustomException): @@ -738,23 +745,23 @@ def test_test_iam_permissions_exception(self): def test_update_finding(self): # Setup Expected Response - name = 'name3373707' - parent = 'parent-995424086' - resource_name = 'resourceName979421212' - category = 'category50511102' - external_uri = 'externalUri-1385596168' + name = "name3373707" + parent = "parent-995424086" + resource_name = "resourceName979421212" + category = "category50511102" + external_uri = "externalUri-1385596168" expected_response = { - 'name': name, - 'parent': parent, - 'resource_name': resource_name, - 'category': category, - 'external_uri': external_uri + "name": name, + "parent": parent, + "resource_name": resource_name, + "category": category, + "external_uri": external_uri, } expected_response = finding_pb2.Finding(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() @@ -767,14 +774,15 @@ def test_update_finding(self): assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.UpdateFindingRequest( - finding=finding) + finding=finding + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_update_finding_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() @@ -787,18 +795,19 @@ def test_update_finding_exception(self): def test_update_organization_settings(self): # Setup Expected Response - name = 'name3373707' + name = "name3373707" enable_asset_discovery = False expected_response = { - 'name': name, - 'enable_asset_discovery': enable_asset_discovery + "name": name, + "enable_asset_discovery": enable_asset_discovery, } expected_response = organization_settings_pb2.OrganizationSettings( - **expected_response) + **expected_response + ) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() @@ -811,14 +820,15 @@ def test_update_organization_settings(self): assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.UpdateOrganizationSettingsRequest( - organization_settings=organization_settings) + organization_settings=organization_settings + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_update_organization_settings_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() @@ -831,19 +841,19 @@ def test_update_organization_settings_exception(self): def test_update_source(self): # Setup Expected Response - name = 'name3373707' - display_name = 'displayName1615086568' - description = 'description-1724546052' + name = "name3373707" + display_name = "displayName1615086568" + description = "description-1724546052" expected_response = { - 'name': name, - 'display_name': display_name, - 'description': description + "name": name, + "display_name": display_name, + "description": description, } expected_response = source_pb2.Source(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() @@ -855,15 +865,14 @@ def test_update_source(self): assert expected_response == response assert len(channel.requests) == 1 - expected_request = securitycenter_service_pb2.UpdateSourceRequest( - source=source) + expected_request = securitycenter_service_pb2.UpdateSourceRequest(source=source) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_update_source_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() @@ -876,14 +885,13 @@ def test_update_source_exception(self): def test_update_security_marks(self): # Setup Expected Response - name = 'name3373707' - expected_response = {'name': name} - expected_response = security_marks_pb2.SecurityMarks( - **expected_response) + name = "name3373707" + expected_response = {"name": name} + expected_response = security_marks_pb2.SecurityMarks(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient() @@ -896,14 +904,15 @@ def test_update_security_marks(self): assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.UpdateSecurityMarksRequest( - security_marks=security_marks) + security_marks=security_marks + ) actual_request = channel.requests[0][1] assert expected_request == actual_request def test_update_security_marks_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) - patch = mock.patch('google.api_core.grpc_helpers.create_channel') + patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = securitycenter_v1beta1.SecurityCenterClient()