g"ddlmZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSddlTmUZUddlVmWZWddlXmYZYmZZZe r ddl[m\Z\ddl]m^Z^ejZ`GddZay)) annotationsN)Path) TYPE_CHECKINGAnyCallable) MagicMock)parse) source_util)Runtime)MemoryCacheStorageManager)MediaFileManager)MemoryMediaFileStorage) PagesManager)Secrets) TESTING_KEY)SafeSessionState) SessionState).BlockButton ButtonGroupCaption ChatInput ChatMessageCheckboxCode ColorPickerColumn Dataframe DateInputDivider ElementList ElementTreeError ExceptionExpanderHeaderInfoJsonLatexMarkdownMetric MultiselectNode NumberInputRadio Selectbox SelectSliderSliderStatus SubheaderSuccessTabTableTextTextArea TextInput TimeInputTitleToastToggleWarning WidgetListrepr_)LocalScriptRunner)patch_config_options)HASHLIB_KWARGScalc_md5)Sequence) WidgetStatesceZdZdZddd d?dZeddd@dZedddd d@d Zedddd dAd Zedd dBd Z dC dDd Z dd dEdZ dFdZ e dGdZe dGdZe dHdZe dIdZe dJdZe dKdZe dLdZe dMdZe dNdZe dOdZe dPdZe dQdZe dRdZe dSdZe dTdZe dUdZe dVd Ze dWd!Ze dXd"Z e dYd#Z!e dZd$Z"e d[d%Z#e d\d&Z$e d]d'Z%e d^d(Z&e d_d)Z'e d`d*Z(e dad+Z)e dbd,Z*e dcd-Z+e ddd.Z,e ded/Z-e dfd0Z.e dgd1Z/e dhd2Z0e did3Z1e djd4Z2e dkd5Z3e dld6Z4e dmd7Z5e dnd8Z6e dod9Z7dpd:Z8d;Z9dqd<Z:drd=Z;dsd>Zz"AppTest.__init__..sTrQ)str _script_pathdefault_timeoutrrr session_state query_paramssecretsrKrL _page_hashr"_runner_tree)self script_pathrWrKrLrXtrees rR__init__zAppTest.__init__sr ,.$ %' k"-m\J,.')   }  rQrWc(|j||S)a3 Create an instance of ``AppTest`` to simulate an app page defined within a string. This is useful for testing short scripts that fit comfortably as an inline string in the test itself, without having to create a separate file for it. The script must be executable on its own and so must contain all necessary imports. Parameters ---------- script: str The string contents of the script to be run. default_timeout: float Default time in seconds before a script run is timed out. Can be overridden for individual ``.run()`` calls. Returns ------- AppTest A simulated Streamlit app for testing. The simulated app can be executed via ``.run()``. rc) _from_string)clsscriptrWs rR from_stringzAppTest.from_strings6HHrQrWrKrLctjt|dfit}|j }t t j|}tj|}|j|tt||||S)Nzutf-8ri) hashlibmd5bytesrD hexdigestrTMP_DIRnametextwrapdedent write_textrIrU) rfrgrWrKrLhasher script_namepathaligned_scripts rRrezAppTest._from_stringspU673F~F&&( GLL+.!0 ' IT&  rQctj|\}}tjdj |}|d|j dz}|j ||||S)a  Create an instance of ``AppTest`` to simulate an app page defined within a function. This is similar to ``AppTest.from_string()``, but more convenient to write with IDE assistance. The script must be executable on its own and so must contain all necessary imports. Parameters ---------- script: Callable A function whose body will be used as a script. Must be runnable in isolation, so it must include any necessary imports. default_timeout: float Default time in seconds before a script run is timed out. Can be overridden for individual ``.run()`` calls. args: tuple An optional tuple of args to pass to the script function. kwargs: dict An optional dict of kwargs to pass to the script function. Returns ------- AppTest A simulated Streamlit app for testing. The simulated app can be executed via ``.run()``. rT z(*__args, **__kwargs)ri)inspectgetsourcelinesrqrrjoin__name__re) rfrgrWrKrL source_lines_sourcemodules rR from_functionzAppTest.from_functionsjP"008 a!67Bv//DEE O$v   rQc t|}|jr|}nYtjj tj d}t|dj }|j|z }t||S)a. Create an instance of ``AppTest`` to simulate an app page defined within a file. This option is most convenient for CI workflows and testing of published apps. The script must be executable on its own and so must contain all necessary imports. Parameters ---------- script_path: str | Path Path to a script file. The path should be absolute or relative to the file calling ``.from_file``. default_timeout: float Default time in seconds before a script run is timed out. Can be overridden for individual ``.run()`` calls. Returns ------- AppTest A simulated Streamlit app for testing. The simulated app can be executed via ``.run()``. Nrc) ris_file traceback StackSummaryextract walk_stackfilenameparentrI)rfr_rWrvstackfilepaths rR from_filezAppTest.from_file sq8;'    D **2293G3G3MNEE!H--.H??[0Dt_==rQcddl}| |j}tt}t t d|_t|_|t_ t|jd}tj5tj}dt_ddd|j}|jr"t!}|j|_||_t%|j|j&||j(|j*} t-dd i5| j/||j0||j2|_||j4_ddd| j8d d j:} t=j>| |_tj5t_ddd|jrA|jj"$tA|jj"|_||_dt_ |S#1swYxYw#1swYxYw#1swY|xYw) zRun the script, and parse the output messages for querying and interaction. Timeout is in seconds, or None to use the default timeout of the runner. rN)specz /mock/mediaF) setup_watcherrJzglobal.appTestT client_state)! streamlitrWrr r rmedia_file_mgrr cache_storage_manager _instancerrVr _pages_cache_lock _cached_pagesrZr_secretsrBrXrKrLrCrunrYr[r]r\ event_data query_stringr parse_qsdict) r^ widget_statetimeoutst mock_runtime pages_managersaved_cached_pages saved_secrets new_secrets script_runnerrs rR_runz AppTest._run4s  ?**G!g. &6 "= 1' #.G-H *($T%6%6eL  * *!,!:!: (,K %+"$ <<!)K#'<$%>:?%> %>%>R-1 $>)>>  >@.2/.8   " " ! ! ' ' " " % % ' ' # #   ' '""  $ $ % % " "     $ $ # # ! !         # # ! ! & & ' '     ( ( $ $ ! ! $ $ " " ! !    &   $ $ % % % %         ! ! " ",2rQrI)b __future__rrkrztempfilerqrpathlibrtypingrrr unittest.mockrurllibr rr streamlit.runtimer 5streamlit.runtime.caching.storage.dummy_cache_storager $streamlit.runtime.media_file_managerr +streamlit.runtime.memory_media_file_storagerstreamlit.runtime.pages_managerrstreamlit.runtime.secretsrstreamlit.runtime.state.commonr*streamlit.runtime.state.safe_session_stater%streamlit.runtime.state.session_stater!streamlit.testing.v1.element_treerrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rA(streamlit.testing.v1.local_script_runnerrBstreamlit.testing.v1.utilrCstreamlit.utilrDrEcollections.abcrF streamlit.proto.WidgetStates_pb2rGTemporaryDirectoryrorIrPrQrRr?s#//#!%BN8-6G>////////////`G:3(= %( % % '||rQ