,%dZddlmZddlmZddlmZddlZddlmZddl mZddl m Z ddl mZ ddlmZdd lmZd Zd Zd Zd ZddZddZddZddZdZdZddZdZy)zIA shared library to support implementation of Firebase Test Lab commands.)absolute_import)division)unicode_literalsN) exceptions)apis)log) propertiesz table[box]( outcome.color(red=Fail, green=Pass, blue=Flaky, yellow=Inconclusive), axis_value:label=TEST_AXIS_VALUE, test_details:label=TEST_DETAILS ) c tj|j}|dd}|dd}dj ||S#t$r|jcYSwxYw)aoReturns a ready-to-print string representation from the http response. Args: error: the Http error response, whose content is a JSON-format string for most cases (e.g. invalid test dimension), but can be just a string other times (e.g. invalid URI for CLOUDSDK_TEST_ENDPOINT). Returns: A ready-to-print string representation of the error. errorcodemessagezResponseError {0}: {1})jsonloadscontent ValueErrorformat)r datar r s 0lib/googlecloudsdk/api_lib/firebase/test/util.pyGetErrorr)se ::emm $D gv $ M) $' ! ( (w 77  ==sAAAc`tj|j}|dd|ddfS)aReturns the individual error code and message from a JSON http response. Prefer using GetError(error) unless you need to examine the error code and take specific action on it. Args: error: the Http error response, whose content is a JSON-format string. Returns: (code, msg) A tuple holding the error code and error message string. Raises: ValueError: if the error is not in JSON format. r r r )rrr)r rs rGetErrorCodeAndMessager>s3 EMM "$ gv W i 8 88ctjjjj }|st j d|S)a(Get the user's project id from the core project properties. Returns: The id of the GCE project to use while running the test. Raises: MissingProjectError: if the user did not specify a project id via the --project flag or via running "gcloud config set project PROJECT_ID". zNo project specified. Please add --project PROJECT_ID to the command line or first run $ gcloud config set project PROJECT_ID)r VALUEScoreprojectGetrMissingProjectError)rs r GetProjectrQsI    " " * * . . 0'   ( ( G HH .rc|r |d}|d}n,tjdd}tjdd}|jjj }t |||jS)aoGets the device IP block catalog from the TestEnvironmentDiscoveryService. Args: context: {str:object}, The current context, which is a set of key-value pairs that can be used for common initialization among commands. Returns: The device IP block catalog Raises: calliope_exceptions.HttpException: If it could not connect to the service. testing_clienttesting_messagestestingv1)rGetClientInstanceGetMessagesModule'TestingTestEnvironmentCatalogGetRequestEnvironmentTypeValueValuesEnumDEVICE_IP_BLOCKS _GetCatalogdeviceIpBlockCatalogcontextclientmessagesenv_types rGetDeviceIpBlocksr1cst % &F)*H  # #It 4F%%i6H66%%&6&6  VXx 0 E EErc|r |d}|d}n,tjdd}tjdd}|jjj }t |||jS)a`Gets the Android catalog from the TestEnvironmentDiscoveryService. Args: context: {str:object}, The current context, which is a set of key-value pairs that can be used for common initialization among commands. Returns: The android catalog. Raises: calliope_exceptions.HttpException: If it could not connect to the service. r!r"r#r$)rr%r&r'r(ANDROIDr*androidDeviceCatalogr,s rGetAndroidCatalogr5}sr % &F)*H  # #It 4F%%i6H66$$WW  VXx 0 E EErc|r |d}|d}n,tjdd}tjdd}|jjj }t |||jS)aXGets the iOS catalog from the TestEnvironmentDiscoveryService. Args: context: {str:object}, The current context, which is a set of key-value pairs that can be used for common initialization among commands. Returns: The iOS catalog. Raises: calliope_exceptions.HttpException: If it could not connect to the service. r!r"r#r$)rr%r&r'r(IOSr*iosDeviceCatalogr,s r GetIosCatalogr9sr % &F)*H  # #It 4F%%i6H66$$SS  VXx 0 A AArc|r |d}|d}n,tjdd}tjdd}|jjj }t |||jS)apGets the network profile catalog from the TestEnvironmentDiscoveryService. Args: context: {str:object}, The current context, which is a set of key-value pairs that can be used for common initialization among commands. Returns: The network profile catalog. Raises: calliope_exceptions.HttpException: If it could not connect to the service. r!r"r#r$)rr%r&r'r(NETWORK_CONFIGURATIONr*networkConfigurationCatalogr,s rGetNetworkProfileCatalogr=st % &F)*H  # #It 4F%%i6H66$$%:%:  VXx 0 L LLrchtjjjj }|j ||} |j j |S#tj$r&}tjdt|zd}~wtjdxYw)azGets a test environment catalog from the TestEnvironmentDiscoveryService. Args: client: The Testing API client object. messages: The Testing API messages object. environment_type: {enum} which EnvironmentType catalog to get. Returns: The test environment catalog. Raises: calliope_exceptions.HttpException: If it could not connect to the service. )environmentType projectIdz/Unable to access the test environment catalog: Nz;Unable to access the Firebase Test Lab environment catalog.)r rrrrr'testEnvironmentCatalogapitools_exceptions HttpErrorcalliope_exceptions HttpExceptionrrr )r.r/environment_type project_idrequestr s rr*r*s  %%--113*  < <& = '   ( ( , ,W 55  & &M  + +9HUOK MM IIKL sA""B15!BB1c H|jd}|d}t|dkDr%tjddj |t|dk(rd}||fS|d}gd }||vr5tjdd j |d j |||fS) aReturns a tuple representing a directive's type and resource name. Args: key: the directive key, which can be ":" or "" Returns: A tuple of the directive's parsed type and resource name. If no type is specified, "text" will be returned as the default type. Raises: InvalidArgException: if the input format is incorrect or if the specified type is unsupported. :robo_directiveszYInvalid format for key [{0}]. Use a colon only to separate action type and resource name.textr)rOclickignorez9Unsupported action type [{0}]. Please choose one of [{1}]z, )splitlenrInvalidArgExceptionrjoin)keyparts resource_name action_typesupported_action_typess rParseRoboDirectiveKeyr[s ))C.%)-Z!^  ( (FFLf G   Z1_K } %%(K800  * *  E L L499%;< > ?? } %%rcN|D] }|jD]}d|vsd|zccS"y)zCReturns a warning string iff any device model is marked deprecated. deprecatedz|Some devices are deprecated. Learn more at https://firebase.google.com/docs/test-lab/%s/available-testing-devices#deprecatedN)tags)modelsplatformmodeltags rGetDeprecatedTagWarningrcs;ezz  79AB C rc2|jdr|ddS|S)zDReturns the relative device path that can be joined with GCS bucket./rNN) startswith) device_paths rGetRelativeDevicePathrhs '2237QRH[Hr)N)android)__doc__ __future__rrrrapitools.base.pyrrB$googlecloudsdk.api_lib.firebase.testgooglecloudsdk.api_lib.utilrgooglecloudsdk.callioperDgooglecloudsdk.corerr OUTCOMES_FORMATrrrr1r5r9r=r*r[rcrhrrrssm P&' >;,E#*8*9&$F4F4B4M4 <%&PIr