@@ -19,10 +19,10 @@ def mock_session_state(mocker):
19
19
def mock_query_params (mocker ):
20
20
"""Add a pytest fixture that returns a function that mocks query params."""
21
21
def func (key : str , value : str ):
22
- get_query_params = mocker .patch (
23
- sut + ".streamlit.experimental_get_query_params"
22
+ mocker .patch .dict (
23
+ sut + ".streamlit.query_params" ,
24
+ {key : value }
24
25
)
25
- get_query_params .return_value = {key : [value ]}
26
26
27
27
return func
28
28
@@ -176,9 +176,9 @@ def test_register_string_parameter(
176
176
assert parameter .default == "G'day!"
177
177
assert parameter .value == "Hello"
178
178
assert repr (parameter ) == "Parameter(default=G'day!,value=Hello,touched=True)"
179
- set_query_params = mocker .patch (sut + ".streamlit.experimental_set_query_params " )
179
+ set_query_params = mocker .patch (sut + ".streamlit.runtime.state.query_params_proxy.QueryParamsProxy.from_dict " )
180
180
Parameters .set_url_fields ()
181
- set_query_params .assert_called_with (foo = " Hello" )
181
+ set_query_params .assert_called_with ({ " foo" : " Hello"} )
182
182
183
183
184
184
def test_register_string_parameter_not_in_url (
@@ -223,14 +223,14 @@ def test_register_string_list_parameter(
223
223
repr (parameter )
224
224
== "Parameter(default=['flying', 'spaghetti', 'monster'],value=['flying', 'spaghetti', 'monster'],touched=True)"
225
225
)
226
- set_query_params = mocker .patch (sut + ".streamlit.experimental_set_query_params " )
226
+ set_query_params = mocker .patch (sut + ".streamlit.runtime.state.query_params_proxy.QueryParamsProxy.from_dict " )
227
227
Parameters .set_url_fields ()
228
- set_query_params .assert_called_with (foo = " ['flying', 'spaghetti', 'monster']" )
228
+ set_query_params .assert_called_with ({ " foo" : " ['flying', 'spaghetti', 'monster']"} )
229
229
230
230
# Now update value and make sure new value properly serialized in query params
231
231
parameters .foo .update (new_value = ["Hello" , "world" ])
232
232
Parameters .set_url_fields ()
233
- set_query_params .assert_called_with (foo = " ['Hello', 'world']" )
233
+ set_query_params .assert_called_with ({ " foo" : " ['Hello', 'world']"} )
234
234
235
235
236
236
def test_register_string_list_parameter_not_in_url (
@@ -278,9 +278,9 @@ def test_register_date_parameter(
278
278
parameter = st .session_state ._parameters ["foo" ]
279
279
assert parameter .default == datetime .date (2021 , 11 , 1 )
280
280
assert parameter .value == datetime .date (2021 , 11 , 1 )
281
- set_query_params = mocker .patch (sut + ".streamlit.experimental_set_query_params " )
281
+ set_query_params = mocker .patch (sut + ".streamlit.runtime.state.query_params_proxy.QueryParamsProxy.from_dict " )
282
282
Parameters .set_url_fields ()
283
- set_query_params .assert_called_with (foo = " 2021-11-01" )
283
+ set_query_params .assert_called_with ({ " foo" : " 2021-11-01"} )
284
284
285
285
286
286
def test_register_date_parameter_not_in_url (
@@ -307,10 +307,10 @@ def test_register_date_range_parameter(
307
307
assert parameter .value == (datetime .date (2021 , 11 , 1 ), datetime .date (2021 , 11 , 3 ))
308
308
309
309
# Now update value and make sure new value properly serialized in query params
310
- set_query_params = mocker .patch (sut + ".streamlit.experimental_set_query_params " )
310
+ set_query_params = mocker .patch (sut + ".streamlit.runtime.state.query_params_proxy.QueryParamsProxy.from_dict " )
311
311
parameters .foo .update (new_value = (datetime .date (2023 , 11 , 1 ), datetime .date (2023 , 11 , 3 )))
312
312
Parameters .set_url_fields ()
313
- set_query_params .assert_called_with (foo = '(2023-11-01,2023-11-03)' )
313
+ set_query_params .assert_called_with ({ " foo" : '(2023-11-01,2023-11-03)' } )
314
314
315
315
316
316
def test_as_dict (
0 commit comments