[(dZddlmZddlmZddlmZddlZddlZddlmZ ddl m Z ddl m Z dd l m Z ddlmZdd lmZd ZGd d eZy)z.Create iOS test matrices in Firebase Test Lab.)absolute_import)division)unicode_literalsN) exceptions)matrix_creator_common) matrix_ops)util)logczt|||||}|jtjjS)aCreates a new iOS matrix test in Firebase Test Lab from the user's params. Args: args: an argparse namespace. All the arguments that were provided to this gcloud command invocation (i.e. group and command arguments combined). context: {str:obj} dict containing the gcloud command context, which includes the Testing API client+messages libs generated by Apitools. history_id: {str} A history ID to publish Tool Results to. gcs_results_root: the root dir for a matrix within the GCS results bucket. release_track: the release track that the command is invoked from. Returns: A TestMatrix object created from the supplied matrix configuration values. ) MatrixCreatorCreateTestMatrixuuiduuid4hex)argscontext history_idgcs_results_root release_trackcreators >lib/googlecloudsdk/api_lib/firebase/test/ios/matrix_creator.py CreateMatrixr!s7 $5E' )'  ! !$**,"2"2 33cTeZdZdZdZddZdZdZdZdZ dZ d Z d Z d Z d Zy )r zGCreates a single iOS test matrix based on user-supplied test arguments.ctj|_||_||_||_|d|_|d|_||_y)agConstruct an MatrixCreator to be used to create a single test matrix. Args: args: an argparse namespace. All the arguments that were provided to this gcloud command invocation (i.e. group and command arguments combined). context: {str:obj} dict containing the gcloud command context, which includes the Testing API client+messages libs generated by Apitools. history_id: {str} A history ID to publish Tool Results to. gcs_results_root: the root dir for a matrix within the GCS results bucket. release_track: the release track that the command is invoked from. testing_clienttesting_messagesN) r GetProject_project_args _history_id_gcs_results_root_client _messages_release_track)selfrrrrrs r__init__zMatrixCreator.__init__8sKOO%DMDJ!D-D+,DL/0DN'Drc|sy|rtjj|}tjj|j|}|j j |S)z0Build a FileReference pointing to a file in GCS.NgcsPath)ospathbasenamejoinr"r$ FileReference)r&filename use_basenamer,s r_BuildFileReferencez!MatrixCreator._BuildFileReferenceMsQ  !!(+h 77<<.. 9D >> ' ' ' 55rc t|jdgxsgDcgc]0}|jtjj |2}}g}t|jdgxsgD]r}d|vrA|j d\}}|j|jj||H|j|jj|tg}t|jddxsi}|jD]} | jd} | dk7r| d| nd} | dk7r| | d zdn| }|j|jj|jtj|d | | |jjt|jd d|||Scc}w)z&Build an IosTestSetup for an iOS test.additional_ipasdirectories_to_pull:)bundleId devicePath)r8 other_filesNF)r1)contentr7r8network_profile)networkProfileadditionalIpas pushFilespullDirectories)getattrr r2r+r,r-splitappendr$ IosDeviceFilekeysfindr GetRelativeDevicePath IosTestSetup) r&additional_ipar4r5 directorybundler, device_filesr9 device_pathidx bundle_ids r_BuildGenericTestSetupz$MatrixCreator._BuildGenericTestSetupVs&djj2CRHNBNNN   !1!1.!ABNTZZ)>CIrI  s+ "" NN ( (&T ( J L "" NN ( (I ( > @ JL$**mT:@bK"'')    S !c'*by+ds#di&)Ri[q "[d .. & &..,,T2/H  '  * >> & &tzz+> + +"&**"9"99002// 0B0BC..**--djjoo>..tzz/H/HI11$+DJJ,G,1%3 +4 , 5D Krc ||jj|jj |j t j |jj|jj|j|jj|jj}|S)rS)appIpa scenarios)rYrZr[ iosTestLoop) r$r]r r^rQrr_r` IosTestLoopr2appscenario_numbersres r_BuildIosTestLoopTestSpecz'MatrixCreator._BuildIosTestLoopTestSpecs >> + +"&**"9"99002// 0B0BCNN..++DJJNN;jj11/3 , 4D Krc "|jj|jj |j t j |jj|jj|j|jj}t|jddrQ|jtjj|jj|j _|S)z/Build a TestSpecification for an iOS Robo test.)ri)rYrZr[ iosRoboTest robo_scriptN)r$r]r r^rQrr_r` IosRoboTestr2rmrBr+r,r-rrrq roboScriptres r_BuildIosRoboTestSpecz#MatrixCreator._BuildIosRoboTestSpecs >> + +"&**"9"99002// 0B0BCNN..++DJJNN;/= , >D tzz=$/$($<$< ''  4::11 2%4d! Krc|dk(r|jS|dk(r|jS|dk(r|jStjddj |)zBMap a test type into its corresponding TestSpecification message .xctestz game-looprobotypezUnknown test type "{}".)rgrorurInvalidArgumentExceptionformat)r& test_types r_TestSpecFromTypezMatrixCreator._TestSpecFromTypeskH  % % '' k !  + + -- f   ' ' ))  / / +229= ??rc|jjDcgc]}|j|}}|jj |jj |}|jj |j}|jj|j|j}|jj||}tj|jt|jdixsi|j}|jj!|||||jj"xsdScc}w) zBuild just the user-specified parts of an iOS TestMatrix message. Args: spec: a TestSpecification message corresponding to the test type. Returns: A TestMatrix message. ) iosDevices) iosDeviceListr)) projectId historyId)googleCloudStoragetoolResultsHistoryclient_detailsr)testSpecificationenvironmentMatrix clientInfo resultStorageflakyTestAttempts)r device_BuildIosDevicer$EnvironmentMatrix IosDeviceListGoogleCloudStorager"ToolResultsHistoryrr! ResultStoragerBuildClientInforBr% TestMatrixnum_flaky_test_attempts) r&rfddevicesenvironment_matrixgcshistresults client_infos r_BuildTestMatrixzMatrixCreator._BuildTestMatrixs@15 0A0AB0A1t##A&0AGB99nn22g2F:H .. + +D4J4J + KC >> , ,--4+;+; - =Dnn**4+9G(77  ,b17R9L9LNK >> $ $,**<<A % CCCsEcX|jj|d|d|d|dS)Nmodelversionlocale orientation) iosModelId iosVersionIdrr)r$ IosDevice)r& device_maps rrzMatrixCreator._BuildIosDevices> >> # #g& *(#}- $ //rc|j|jj}|jj |j |j ||S)zBuild a TestingProjectsTestMatricesCreateRequest for a test matrix. Args: request_id: {str} a unique ID for the CreateTestMatrixRequest. Returns: A TestingProjectsTestMatricesCreateRequest message. )r testMatrix requestId)r}r ryr$(TestingProjectsTestMatricesCreateRequestrr)r& request_idrfs r_BuildTestMatrixRequestz%MatrixCreator._BuildTestMatrixRequestsQ  ! !$**// 2D >> B B--((. C rc|j|}tjdj| |jj j |}tjdj|tjjdj|j|S#tj$r2}dtj|z}tj|d}~wwxYw)a>Invoke the Testing service to create a test matrix from the user's args. Args: request_id: {str} a unique ID for the CreateTestMatrixRequest. Returns: The TestMatrix response message from the TestMatrices.Create rpc. Raises: HttpException if the test service reports an HttpError. z!TestMatrices.Create request: {0} z"TestMatrices.Create response: {0} z'Http error while creating test matrix: Nz1Test [{id}] has been created in the Google Cloud.)id)rr debugr{r#projects_testMatricesCreateapitools_exceptions HttpErrorr GetErrorr HttpExceptionstatusPrint testMatrixId)r&rrequestresponseerrormsgs rr zMatrixCreator.CreateTestMatrixs**:6GII3::7CD*33::7Ch ii6==hGH JJHOO  P"# O  ( (* 5 e8L Lc  $ $S ))*sA B;;D-C;;DN)T)__name__ __module__ __qualname____doc__r'r2rQrgrorur}rrrr rrr r 5s>O(*6!-F    ?C</ rr )r __future__rrrr+rapitools.base.pyrr$googlecloudsdk.api_lib.firebase.testrrr googlecloudsdk.calliopegooglecloudsdk.corer robjectr rrrrs>5&' >F;5.#4(BFBr