&eddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z ddl mZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lm 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-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNddlOmPZPddlQmRZRddlSmTZTejUZVeTGddZWdS)) annotationsN)AnyCallableSequence) MagicMock)parse) source_util) WidgetStates)Runtime)MemoryCacheStorageManager)MediaFileManager)MemoryMediaFileStorage)Secrets)SafeSessionState) SessionState)+BlockButtonCaption ChatInput ChatMessageCheckboxCode ColorPickerColumn Dataframe DateInputDivider ElementList ElementTreeError ExceptionHeaderInfoJsonLatexMarkdownMetric MultiselectNode NumberInputRadio Selectbox SelectSliderSlider SubheaderSuccessTabTableTextTextArea TextInput TimeInputTitleToastToggleWarning WidgetListrepr_)LocalScriptRunner)HASHLIB_KWARGS)_fix_matplotlib_crashc^eZdZdZdndZedddod Zedddpd ZedddqdZ drdsdZ dddtdZ e dudZ e dudZ e dvdZe dwdZe dxdZe dyd!Ze dzd#Ze d{d%Ze d|d'Ze d}d)Ze d~d+Ze dd-Ze dd/Ze dd1Ze dd3Ze dd5Ze dd7Ze dd9Ze dd;Ze dd=Ze dd?Z e ddAZ!e ddCZ"e ddEZ#e ddGZ$e ddIZ%e ddKZ&e ddMZ'e ddOZ(e ddQZ)e ddSZ*e ddUZ+e ddWZ,e ddYZ-e dd[Z.e dd]Z/e dd_Z0e ddaZ1e ddcZ2ddeZ3dfZ4ddiZ5ddlZ6ddmZ7dS)AppTesta# A simulated Streamlit app to check the correctness of displayed elements and outputs. An instance of ``AppTest`` simulates a running Streamlit app. This class provides methods to set up, manipulate, and inspect the app contents via API instead of a browser UI. It can be used to write automated tests of an app in various scenarios. These can then be run using a tool like pytest. ``AppTest`` can be initialized by one of three class methods: * |st.testing.v1.AppTest.from_file|_ (recommended) * |st.testing.v1.AppTest.from_string|_ * |st.testing.v1.AppTest.from_function|_ Once initialized, Session State and widget values can be updated and the script can be run. Unlike an actual live-running Streamlit app, you need to call ``AppTest.run()`` explicitly to re-run the app after changing a widget value. ``AppTest`` enables developers to build tests on their app as-is, in the familiar python test format, without major refactoring or abstracting out logic to be tested separately from the UI. Tests can run quickly with very low overhead. A typical pattern is to build a suite of tests for an app that ensure consistent functionality as the app evolves, and run the tests locally and/or in a CI environment like Github Actions. .. note:: ``AppTest`` only supports testing a single page of an app per instance. For multipage apps, each page will need to be tested separately. No methods exist to programatically switch pages within ``AppTest``. .. |st.testing.v1.AppTest.from_file| replace:: ``st.testing.v1.AppTest.from_file`` .. _st.testing.v1.AppTest.from_file: #apptestfrom_file .. |st.testing.v1.AppTest.from_string| replace:: ``st.testing.v1.AppTest.from_string`` .. _st.testing.v1.AppTest.from_string: #apptestfrom_string .. |st.testing.v1.AppTest.from_function| replace:: ``st.testing.v1.AppTest.from_function`` .. _st.testing.v1.AppTest.from_function: #apptestfrom_function Attributes ---------- secrets: dict[str, Any] Dictionary of secrets to be used the simulated app. Use dict-like syntax to set secret values for the simulated app. session_state: SafeSessionState Session State for the simulated app. SafeSessionState object supports read and write operations as usual for Streamlit apps. query_params: dict[str, Any] Dictionary of query parameters to be used by the simluated app. Use dict-like syntax to set ``query_params`` values for the simulated app. script_pathstrdefault_timeoutfloatc||_||_ttd|_i|_i|_t}||_||_ dS)NcdSNrI=lib/python3.11/site-packages/streamlit/testing/v1/app_test.pyz"AppTest.__init__..sdrJ) _script_pathrDrr session_state query_paramssecretsr_runner_tree)selfrBrDtrees rK__init__zAppTest.__init__sT'.-lnnllKK,.') }}  rJrDscriptreturnc@tjt|dfit}|}t jtj|}tj |}| |tt||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()``. zutf-8rW)hashlibmd5bytesr> hexdigestpathlibPathTMP_DIRnametextwrapdedent write_textrArC)clsrXrDhasher script_namepathaligned_scripts rK from_stringzAppTest.from_strings6U6733FF~FF&&(( |GL+66!00 '''s4yy/BBBBrJCallable[[], None]cZtj|\}}tjd|}t j|}|jd}||jd}tjd|} | | |S)aS 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. Returns ------- AppTest A simulated Streamlit app for testing. The simulated app can be executed via ``.run()``. rNrW) inspectgetsourcelinesrcrdjoinastrbodylinenork) rfrXrD source_lines_sourcemodulefn_def body_linesrss rK from_functionzAppTest.from_functions<"088 a!6!6776""Q!&-//2 rwwz2233t_EEErJcZtjtj|r|}ngtjtjd}tj|dj}t|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 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()``. NrW) r_r`is_file traceback StackSummaryextract walk_stackfilenamerCparentrA)rfrBrDristackfilepaths rK from_filezAppTest.from_files6 <   [ 9 9 : : 6DD *2293G3M3MNNE|E!H$566Hx455Dt_====rJN widget_stateWidgetStates | Nonetimeout float | Noneczddl}||j}tt}t t d|_t|_|t_ tj 5tj }dt_ dddn #1swxYwY|j }|j r"tg}|j |_||_ t!|j|j}|||j||_||j_|jddj} t3j| |_tj 5|t_ dddn #1swxYwY|j r1|j jt7|j j|_ ||_ dt_ |S)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/media client_state) streamlitrDrr r rmedia_file_mgrr cache_storage_manager _instancer _pages_cache_lock _cached_pagesrPr_secretsr=rMrNrunrOrRrQ event_data query_stringrparse_qsdict) rSrrst mock_runtimesaved_cached_pages saved_secrets new_secrets script_runnerrs rK_runz AppTest._runs   ?*G!g... &6 "= 1 1' ' #.G-H-H *(  * - -!,!: (,K % - - - - - - - - - - - - - - -"$ < %!"++K#'#>#>#>#>[#>N-1 $44444l.2//////*   X  " " "X " ! ! !X ! " " "X " % % %X % ' ' 'X ' # # #X #   X  ' ' 'X '"""X"  $ $ $X $ % % %X % " " "X "    X   $ $ $X $ ! ! !X !   X    X     X   # # #X # ! ! !X ! & & &X & ' ' 'X '    X   ( ( (X ( $ $ $X $ ! ! !X ! $ $ $X $ " " "X "    X  X&   X  $ $ $X $ % % %X % % % %X %    X      X   ! ! !X ! " " "X ",,,,2rJrA)X __future__rrrr[ror_tempfilercrtypingrrr unittest.mockrurllibrrr streamlit.proto.WidgetStates_pb2r streamlit.runtimer 5streamlit.runtime.caching.storage.dummy_cache_storager $streamlit.runtime.media_file_managerr +streamlit.runtime.memory_media_file_storagerstreamlit.runtime.secretsr*streamlit.runtime.state.safe_session_stater%streamlit.runtime.state.session_stater!streamlit.testing.v1.element_treerrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<(streamlit.testing.v1.local_script_runnerr=streamlit.utilr>streamlit.web.bootstrapr?TemporaryDirectoryrarArIrJrKrIs#""""" **********######!!!!!!999999%%%%%%BAAAAANNNNNN------GGGGGG>>>>>>,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,ZGFFFFF))))))999999 %( % ' 'J J J J J J J J J J rJ