2dZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z ddl m ZddlmZdd lmZdd lmZddlZd Zd ZGd deZdZdZy)zACommon test matrix operations used by Firebase Test Lab commands.)absolute_import)division)unicode_literalsN) exceptions)util)log) properties) console_iog@z%H:%M:%SczeZdZdZejj dfdZdZdZdZ dZ dZ d Z d Z d Zd Zy) MatrixMonitoraA 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. NcF||_||_|d|_|d|_||_t j |_d|_|||_ njtjjjjxst|_ t!j"ds|xjdzc_ |jj$j&}|j(d|j*d |j,d |j.d |j0d |j2d |j4d|j6d|j8d|j:d|j<di |_tA|j.|j0|j2|j4|j6|j8|j:g|_!|jjDj&}tA|j.|j0|j8|j:g|_#y)aTConstruct 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__zMatrixMonitor.__init__0s"DNDO+,DL/0DNDKOO%DMD'#7d     7 7 > > @ ( '   % %D 1 ""a'"....CCK YYj7++],,.H--/J{Y**O D(+++,,--,(D$NN--BBM#& ($D c$|jjj}g}t}|jD]Q}|j |j k(r%|jt|jA|j|S|r>tjjdjdj|tjjdj|j t#||S)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*r7testExecutionsstater0add_FormatInvalidDimension environmentappendrstatusPrintformatjoinrlen)r;matrixstatessupported_testsunsupported_dimensionsr$s rBHandleUnsupportedExecutionsz)MatrixMonitor.HandleUnsupportedExecutionsus^^ ) ) > >FO U%% v55 5""#:4;K;K#LMt$ &  jj FV[[)?@FACJJ K $//S%9 :< rDc|j}|jD]}|j|k(s|cStj||j )aFetch 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. )GetTestMatrixStatusrJidrTestExecutionNotFoundErrorr)r;test_idrUr$s rB_GetTestExecutionStatusz%MatrixMonitor._GetTestExecutionStatussL % % 'F%% G  &  / / HHrDc|jjj}d}d}g}d} |j|}|j j t }|j} | r | jxsd}| jxsg}||dD]?} tjjdj|| jAt|}|j |j"k(rt%j&||j |j(k(r8t%j*djt-|j.|j |k7rH|j }tjjdj||j0||j |j2vrn|j5|j7} | j |j8vrYtj:d| j |j5|j7} | j |j8vrY|j=| j y) aMonitor 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.rFz{0} Test is {1}z+Matrix not yet complete, still in state: %s)rr)r*r_rstrftime_TIMESTAMP_FORMAT testDetails errorMessageprogressMessagesrrPrQrRrstriprTrKr/rTestLabInfrastructureErrorr0AllDimensionsIncompatibleErrorrMrNr6r8_SleepForStatusIntervalr[r:debug_LogTestComplete) r;r^rV last_staterprogresslast_progress_lenrP timestampdetailsmsgrUs rBMonitorTestExecutionProgressz*MatrixMonitor.MonitorTestExecutionProgresss#^^ ) ) > >FJ EH ++G4f++-(():;i""g $$*++1r+,-# )))SZZ\BC.h-  %33E:: 77 777 @@F)&*<*<=AGA?@ @  #\\  *11 t((46 7 99 9  ""$C J % % 'F ,,d:: : ii=v||L ""$'')f ,,d:: : &,,' rDc"|jj|j|j} |jj j |S#tj$r#}tj|}d|_ |d}~wwxYw)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 rBr[z!MatrixMonitor.GetTestMatrixStatussnnBB--dnnC>G \\ / / 3 3G <<  ( (  - -a 0c I  i s$AB+B  BcR |j}tjt}|jD]}||j xxdz cc<|j ||j |jvr|j|j y|j)zFMonitor and report the progress of multiple running tests in a matrix.N) r[ collections defaultdictintrJrK_UpdateMatrixStatusr:rlrj)r;rU state_countsr$s rBMonitorTestMatrixProgressz'MatrixMonitor.MonitorTestMatrixProgresss '')f ,,S1l''$TZZ A% ( |, 55 5 fll+  ""$ rDc g}|jjt}tj|D]:\}}|dkDs |j dj |j||<|jdj |dj|}tt||j|_ tjj|j!|jy)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)rrbrcsix iteritemsrOrRr6sortrSmaxrTr!rrPwriteljust)r;rrPrprKcountouts rBrz!MatrixMonitor._UpdateMatrixStatus sF &&'89I l3 u  i&&):):5)AU&KL4 KKM * 1 1)SXXf=M NC"#c(D,C,CDDJJSYYt6678rDctjdj|tjj dj|j j y)z?Let the user know that their test matrix has completed running.z#Test matrix completed in state: {0}z {0} testing complete.N)rinforRrPrQr capitalize)r; matrix_states rBrlzMatrixMonitor._LogTestCompletesJHH 2 9 9, GHJJ.55 ""$&'rDcBtj|jy)N)timesleepr")r;s rBrjz%MatrixMonitor._SleepForStatusInterval$sJJt))*rDc$|jj|j|j} |jj j |y#tj$r#}tj|}d|_ |d}~wwxYw)zvCancels an in-progress TestMatrix. Raises: HttpException if the Test service reports a back-end error. ruzCancelTestMatrix: {message}N) r(TestingProjectsTestMatricesCancelRequestr rrryCancelr{r|r}r~rrs rBCancelTestMatrixzMatrixMonitor.CancelTestMatrix's{ nnEE--dnnF>G ll((//8  ( (  - -a 0c6c is%AB,B  B)__name__ __module__ __qualname____doc__datetimenowrCrYr_rsr[rrrlrjrrDrBr r 'sQ&&**$( CJ:I(> @&% 9*' + rDr ct|dd3|j}dj|j|jSt|dd3|j }dj|j |jSy)zHReturn a human-readable string representing an invalid matrix dimension. androidDeviceNz[OS-version {vers} on {model}])modelvers iosDevicez[unknown-environment])getattrrrRandroidModelIdandroidVersionIdr iosModelId iosVersionId)rNdevices rBrMrM7s [/40<  & &F , 3 3##&*A*A 4 CD [+t,8  " "F , 3 3f&9&9 4 ;< !rDc&dj|S)a9Reformat 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)rR)durations rBReformatDurationrEs   x  ((rD)r __future__rrrrrrapitools.base.pyrr{$googlecloudsdk.api_lib.firebase.testrgooglecloudsdk.callioper}googlecloudsdk.corerr googlecloudsdk.core.consoler rr'rcobjectr rMrrrDrBrsZH&' >;5E#*2 #MFM` !)rD