2SrSSKJr SSKJr SSKJr SSKrSSKrSSKrSSKJ r SSK J r SSK J r SSK J r SSKJr SS KJr SS KJr SSKrS rS r"S S\5rSrSrg)zACommon test matrix operations used by Firebase Test Lab commands.)absolute_import)division)unicode_literalsN) exceptions)util)log) properties) console_iog@z%H:%M:%Sc\rSrSrSr\R R S4SjrSrSr Sr Sr S r S r S rS rS rSrg) MatrixMonitor'zA monitor to follow and possibly cancel a single test matrix invocation. Attributes: matrix_id: {str} the unique ID of the matrix being monitored. completed_matrix_states: the set of TestMatrix.State enums representing all final matrix states. NcRXlX lUSUlUSUlX@l[ R "5UlSUlUbXPl Or[RRRR5=(d [Ul [ R""SS9(dU=RS-sl URR$R&nUR(SUR*S UR,S UR.S UR0S UR2S UR4SUR6SUR8SUR:SUR<S0 Ul[AUR.UR0UR2UR4UR6UR8UR:/5Ul!URRDR&n[AUR.UR0UR8UR:/5Ul#g)a0Construct a MatrixMonitor to monitor a single test matrix instance. Args: matrix_id: {str} the unique ID of the matrix being monitored. test_type: {str} the type of matrix test being run (e.g. 'robo') context: {str:obj} dict containing the gcloud command context, which includes the Testing API client & messages libs generated by Apitools. clock: injected function which returns a current datetime object when called. Used to generate time-stamps on progress messages. status_interval_secs: {float} how long to sleep between status checks. testing_clienttesting_messagesrNT)error ValidatingPendingRunningFinishedError UnsupportedzIncompatible EnvironmentzIncompatible Architecture CancelledInvalidz *Unspecified*)$ matrix_id _test_type_client _messages_clockr GetProject_project_max_status_length_status_interval_secsr VALUEStestmatrix_status_intervalGetInt_DEFAULT_STATUS_INTERVAL_SECSr IsInteractive TestExecutionStateValueValuesEnum VALIDATINGPENDINGRUNNINGFINISHEDERRORUNSUPPORTED_ENVIRONMENTINCOMPATIBLE_ENVIRONMENTINCOMPATIBLE_ARCHITECTURE CANCELLEDINVALIDTEST_STATE_UNSPECIFIED _state_namesset_completed_execution_states TestMatrixcompleted_matrix_states)selfr test_typecontextclockstatus_interval_secs exec_states matrix_statess 6lib/googlecloudsdk/api_lib/firebase/test/matrix_ops.py__init__MatrixMonitor.__init__0s"NO+,DL/0DNKOO%DMD'#7     7 7 > > @ ( '   % %D 1 ""a'"....CCK YYj7++],,.H--/J{Y**O D(+++,,--,(D$NN--BBM#& ($D c*URRRn/n[5nURHTnUR UR :Xa&UR[UR55 MCURU5 MV U(a;[RRSRSRU5S95 [RRSRUR [#U5S95 U$)zReport unsupported device dimensions and return supported test list. Args: matrix: a TestMatrix message. Returns: A list of TestExecution messages which have supported dimensions. zDSome device dimensions are not compatible and will be skipped: {d}z dzFirebase Test Lab will execute your {t} test on {n} device(s). More devices may be added later if flaky test attempts are specified.)tn)rr*r+r8testExecutionsstater1add_FormatInvalidDimension environmentappendrstatusPrintformatjoinrlen)r<matrixstatessupported_testsunsupported_dimensionsr%s rCHandleUnsupportedExecutions)MatrixMonitor.HandleUnsupportedExecutionsus^^ ) ) > >FO U%% v55 5""#:4;K;K#LMt$ &  jj FV[[)?@FACJJ K $//S%9 :< rFcUR5nURHnURU:XdMUs $ [R"XR 5e)aiFetch the TestExecution state of a specific test within a matrix. This method is only intended to be used for a TestMatrix with exactly one supported TestExecution. It would be inefficient to use it iteratively on a larger TestMatrix. Args: test_id: ID of the TestExecution status to find. Returns: The TestExecution message matching the unique test_id. )GetTestMatrixStatusrLidrTestExecutionNotFoundErrorr)r<test_idrWr%s rC_GetTestExecutionStatus%MatrixMonitor._GetTestExecutionStatussJ % % 'F%% G  &  / / HHrFc*URRRnSnSn/nSnURU5nUR 5R [ 5nURn U (a*U R=(d SnU R=(d /nXVSH?n [RRSRXR555 MA [U5nUR UR":Xa[$R&"U5eUR UR(:Xa6[$R*"SR[-UR.5S95eUR U:waGUR n[RRSRXR0U55 UR UR2;aOUR55 GMUR75n U R UR8;a][R:"SU R 5 UR55 UR75n U R UR8;aM]UR=U R 5 g) adMonitor and report the progress of a single running test. This method prints more detailed test progress messages for the case where the matrix has exactly one supported test configuration. Args: test_id: str, the unique id of the single supported test in the matrix. Raises: TestLabInfrastructureError if the Test service reports a backend error. rNz{0} {1}zDevice dimensions are not compatible: {d}. Please use "gcloud firebase test android models list" to determine which device dimensions are compatible.rHz{0} Test is {1}z+Matrix not yet complete, still in state: %s)rr*r+rbrstrftime_TIMESTAMP_FORMAT testDetails errorMessageprogressMessagesrrRrSrTrstriprVrMr0rTestLabInfrastructureErrorr1AllDimensionsIncompatibleErrorrOrPr7r9_SleepForStatusIntervalr^r;debug_LogTestComplete) r<rarX last_staterprogresslast_progress_lenrR timestampdetailsmsgrWs rCMonitorTestExecutionProgress*MatrixMonitor.MonitorTestExecutionProgresss^^ ) ) > >FJ EH ++G4f++-(():;i""g $$*++1r,-# )))ZZ\BC.h-  %33E:: 77 777 @@F)&*<*<=AGA?@ @  #\\  *11 ((46 7 99 9  ""$C J % % 'F ,,d:: : ii=v||L ""$'')f ,,d:: : &,,' rFc URRURURS9nURR R U5$![Ra$n[R"U5nSUl UeSnAff=f)zFetch the response from the GetTestMatrix rpc. Returns: A TestMatrix message holding the current state of the created tests. Raises: HttpException if the Test service reports a backend error.  projectId testMatrixIdz=Http error {status_code} while monitoring test run: {message}N) r%TestingProjectsTestMatricesGetRequestr!rrprojects_testMatricesGetapitools_exceptions HttpErrorcalliope_exceptions HttpException error_formatr<requesteexcs rCr^!MatrixMonitor.GetTestMatrixStatussnnBB--dnnC>G \\ / / 3 3G <<  ( (  - -a 0c I  i s$AB )BB c\UR5n[R"[5nURHnX#R ==S- ss'M UR U5 UR UR;aURUR 5 gUR5 M)zFMonitor and report the progress of multiple running tests in a matrix.N) r^ collections defaultdictintrLrM_UpdateMatrixStatusr;rprn)r<rW state_countsr%s rCMonitorTestMatrixProgress'MatrixMonitor.MonitorTestMatrixProgresss '')f ,,S1l''$ZZ A% ( |, 55 5 fll+  ""$ rFc/nUR5R[5n[R"U5H9upEUS:dM UR SR URUUS95 M; UR5 SR USRU55n[[U5UR5Ul [RRUR!UR55 g)zUpdate the matrix status line with the current test state counts. Example: 'Test matrix status: Finished:5 Running:3 Unsupported:2' Args: state_counts: {state:count} a dict mapping a test state to its frequency. rz{s}:{c})scz {0} Test matrix status: {1}  N)rrfrgsix iteritemsrQrTr7sortrUmaxrVr"rrRwriteljust)r<rrRrtrMcountouts rCr!MatrixMonitor._UpdateMatrixStatus sF &&'89I l3   i&&):):5)AU&KL4 KKM * 1 1)SXXf=M NC"#c(D,C,CDDJJSYYt6678rFc[R"SRU55 [RR SRUR R 555 g)z?Let the user know that their test matrix has completed running.z#Test matrix completed in state: {0}z {0} testing complete.N)rinforTrRrSr capitalize)r< matrix_states rCrpMatrixMonitor._LogTestCompletesJHH 2 9 9, GHJJ.55 ""$&'rFcD[R"UR5 g)N)timesleepr#)r<s rCrn%MatrixMonitor._SleepForStatusInterval$sJJt))*rFc"URRURURS9nURR R U5 g![Ra$n[R"U5nSUl UeSnAff=f)zjCancels an in-progress TestMatrix. Raises: HttpException if the Test service reports a back-end error. rzzCancelTestMatrix: {message}N) r(TestingProjectsTestMatricesCancelRequestr!rrr~Cancelrrrrrrs rCCancelTestMatrixMatrixMonitor.CancelTestMatrix's{ nnEE--dnnF>G ll((//8  ( (  - -a 0c6c is%AB*B  B) rrr9r"rr!r7r#rr;r)__name__ __module__ __qualname____firstlineno____doc__datetimenowrDr[rbrwr^rrrprnr__static_attributes__rFrCr r 'sQ&&**$( CJ:I(> @&% 9*' + rFr c[USS5b0URnSRURURS9$[USS5b0UR nSRUR URS9$g)zHReturn a human-readable string representing an invalid matrix dimension. androidDeviceNz[OS-version {vers} on {model}])modelvers iosDevicez[unknown-environment])getattrrrTandroidModelIdandroidVersionIdr iosModelId iosVersionId)rPdevices rCrOrO7s [/40<  & &F , 3 3##&*A*A 4 CD [+t,8  " "F , 3 3f&9&9 4 ;< !rFc SRUS9$)a%Reformat a Duration arg to work around ApiTools non-support of that type. Duration args are normally converted to an int in seconds (e.g. --timeout 5m becomes args.timeout with int value 300). Duration proto fields are converted to type string during discovery doc creation, so we have to convert the int back into a string-formatted Duration (i.e. append an 's') before passing it to the Testing Service. Args: duration: {int} the number of seconds in the time duration. Returns: String representation of the Duration with units of seconds. z{secs}s)secs)rT)durations rCReformatDurationrEs   x  ((rF)r __future__rrrrrrapitools.base.pyrr$googlecloudsdk.api_lib.firebase.testrgooglecloudsdk.calliopergooglecloudsdk.corerr googlecloudsdk.core.consoler rr(rgobjectr rOrrrFrCrsZH&' >;5E#*2 #MFM` !)rF