test_client_storage.py 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648
  1. """Integration tests for client side storage."""
  2. from __future__ import annotations
  3. import time
  4. from typing import Generator
  5. import pytest
  6. from selenium.webdriver import Firefox
  7. from selenium.webdriver.common.by import By
  8. from selenium.webdriver.remote.webdriver import WebDriver
  9. from reflex.testing import AppHarness
  10. from . import utils
  11. def ClientSide():
  12. """App for testing client-side state."""
  13. import reflex as rx
  14. class ClientSideState(rx.State):
  15. state_var: str = ""
  16. input_value: str = ""
  17. class ClientSideSubState(ClientSideState):
  18. # cookies with default settings
  19. c1: str = rx.Cookie()
  20. c2: rx.Cookie = "c2 default" # type: ignore
  21. # cookies with custom settings
  22. c3: str = rx.Cookie(max_age=2) # expires after 2 second
  23. c4: rx.Cookie = rx.Cookie(same_site="strict")
  24. c5: str = rx.Cookie(path="/foo/") # only accessible on `/foo/`
  25. c6: str = rx.Cookie(name="c6")
  26. c7: str = rx.Cookie("c7 default")
  27. # local storage with default settings
  28. l1: str = rx.LocalStorage()
  29. l2: rx.LocalStorage = "l2 default" # type: ignore
  30. # local storage with custom settings
  31. l3: str = rx.LocalStorage(name="l3")
  32. l4: str = rx.LocalStorage("l4 default")
  33. # Sync'd local storage
  34. l5: str = rx.LocalStorage(sync=True)
  35. l6: str = rx.LocalStorage(sync=True, name="l6")
  36. # Session storage
  37. s1: str = rx.SessionStorage()
  38. s2: rx.SessionStorage = "s2 default" # type: ignore
  39. s3: str = rx.SessionStorage(name="s3")
  40. def set_l6(self, my_param: str):
  41. self.l6 = my_param
  42. def set_var(self):
  43. setattr(self, self.state_var, self.input_value)
  44. self.state_var = self.input_value = ""
  45. class ClientSideSubSubState(ClientSideSubState):
  46. c1s: str = rx.Cookie()
  47. l1s: str = rx.LocalStorage()
  48. s1s: str = rx.SessionStorage()
  49. def set_var(self):
  50. setattr(self, self.state_var, self.input_value)
  51. self.state_var = self.input_value = ""
  52. def index():
  53. return rx.fragment(
  54. rx.chakra.input(
  55. value=ClientSideState.router.session.client_token,
  56. is_read_only=True,
  57. id="token",
  58. ),
  59. rx.chakra.input(
  60. placeholder="state var",
  61. value=ClientSideState.state_var,
  62. on_change=ClientSideState.set_state_var, # type: ignore
  63. id="state_var",
  64. ),
  65. rx.chakra.input(
  66. placeholder="input value",
  67. value=ClientSideState.input_value,
  68. on_change=ClientSideState.set_input_value, # type: ignore
  69. id="input_value",
  70. ),
  71. rx.button(
  72. "Set ClientSideSubState",
  73. on_click=ClientSideSubState.set_var,
  74. id="set_sub_state",
  75. ),
  76. rx.button(
  77. "Set ClientSideSubSubState",
  78. on_click=ClientSideSubSubState.set_var,
  79. id="set_sub_sub_state",
  80. ),
  81. rx.box(ClientSideSubState.c1, id="c1"),
  82. rx.box(ClientSideSubState.c2, id="c2"),
  83. rx.box(ClientSideSubState.c3, id="c3"),
  84. rx.box(ClientSideSubState.c4, id="c4"),
  85. rx.box(ClientSideSubState.c5, id="c5"),
  86. rx.box(ClientSideSubState.c6, id="c6"),
  87. rx.box(ClientSideSubState.c7, id="c7"),
  88. rx.box(ClientSideSubState.l1, id="l1"),
  89. rx.box(ClientSideSubState.l2, id="l2"),
  90. rx.box(ClientSideSubState.l3, id="l3"),
  91. rx.box(ClientSideSubState.l4, id="l4"),
  92. rx.box(ClientSideSubState.l5, id="l5"),
  93. rx.box(ClientSideSubState.l6, id="l6"),
  94. rx.box(ClientSideSubState.s1, id="s1"),
  95. rx.box(ClientSideSubState.s2, id="s2"),
  96. rx.box(ClientSideSubState.s3, id="s3"),
  97. rx.box(ClientSideSubSubState.c1s, id="c1s"),
  98. rx.box(ClientSideSubSubState.l1s, id="l1s"),
  99. rx.box(ClientSideSubSubState.s1s, id="s1s"),
  100. )
  101. app = rx.App(state=rx.State)
  102. app.add_page(index)
  103. app.add_page(index, route="/foo")
  104. @pytest.fixture(scope="module")
  105. def client_side(tmp_path_factory) -> Generator[AppHarness, None, None]:
  106. """Start ClientSide app at tmp_path via AppHarness.
  107. Args:
  108. tmp_path_factory: pytest tmp_path_factory fixture
  109. Yields:
  110. running AppHarness instance
  111. """
  112. with AppHarness.create(
  113. root=tmp_path_factory.mktemp("client_side"),
  114. app_source=ClientSide, # type: ignore
  115. ) as harness:
  116. yield harness
  117. @pytest.fixture
  118. def driver(client_side: AppHarness) -> Generator[WebDriver, None, None]:
  119. """Get an instance of the browser open to the client_side app.
  120. Args:
  121. client_side: harness for ClientSide app
  122. Yields:
  123. WebDriver instance.
  124. """
  125. assert client_side.app_instance is not None, "app is not running"
  126. driver = client_side.frontend()
  127. try:
  128. yield driver
  129. finally:
  130. driver.quit()
  131. @pytest.fixture()
  132. def local_storage(driver: WebDriver) -> Generator[utils.LocalStorage, None, None]:
  133. """Get an instance of the local storage helper.
  134. Args:
  135. driver: WebDriver instance.
  136. Yields:
  137. Local storage helper.
  138. """
  139. ls = utils.LocalStorage(driver)
  140. yield ls
  141. ls.clear()
  142. @pytest.fixture()
  143. def session_storage(driver: WebDriver) -> Generator[utils.SessionStorage, None, None]:
  144. """Get an instance of the session storage helper.
  145. Args:
  146. driver: WebDriver instance.
  147. Yields:
  148. Session storage helper.
  149. """
  150. ss = utils.SessionStorage(driver)
  151. yield ss
  152. ss.clear()
  153. @pytest.fixture(autouse=True)
  154. def delete_all_cookies(driver: WebDriver) -> Generator[None, None, None]:
  155. """Delete all cookies after each test.
  156. Args:
  157. driver: WebDriver instance.
  158. Yields:
  159. None
  160. """
  161. yield
  162. driver.delete_all_cookies()
  163. def cookie_info_map(driver: WebDriver) -> dict[str, dict[str, str]]:
  164. """Get a map of cookie names to cookie info.
  165. Args:
  166. driver: WebDriver instance.
  167. Returns:
  168. A map of cookie names to cookie info.
  169. """
  170. return {cookie_info["name"]: cookie_info for cookie_info in driver.get_cookies()}
  171. @pytest.mark.asyncio
  172. async def test_client_side_state(
  173. client_side: AppHarness,
  174. driver: WebDriver,
  175. local_storage: utils.LocalStorage,
  176. session_storage: utils.SessionStorage,
  177. ):
  178. """Test client side state.
  179. Args:
  180. client_side: harness for ClientSide app.
  181. driver: WebDriver instance.
  182. local_storage: Local storage helper.
  183. session_storage: Session storage helper.
  184. """
  185. app = client_side.app_instance
  186. assert app is not None
  187. assert client_side.frontend_url is not None
  188. def poll_for_token():
  189. token_input = driver.find_element(By.ID, "token")
  190. assert token_input
  191. # wait for the backend connection to send the token
  192. token = client_side.poll_for_value(token_input)
  193. assert token is not None
  194. return token
  195. def set_sub(var: str, value: str):
  196. # Get a reference to the cookie manipulation form.
  197. state_var_input = driver.find_element(By.ID, "state_var")
  198. input_value_input = driver.find_element(By.ID, "input_value")
  199. set_sub_state_button = driver.find_element(By.ID, "set_sub_state")
  200. AppHarness._poll_for(lambda: state_var_input.get_attribute("value") == "")
  201. AppHarness._poll_for(lambda: input_value_input.get_attribute("value") == "")
  202. # Set the values.
  203. state_var_input.send_keys(var)
  204. input_value_input.send_keys(value)
  205. set_sub_state_button.click()
  206. def set_sub_sub(var: str, value: str):
  207. # Get a reference to the cookie manipulation form.
  208. state_var_input = driver.find_element(By.ID, "state_var")
  209. input_value_input = driver.find_element(By.ID, "input_value")
  210. set_sub_sub_state_button = driver.find_element(By.ID, "set_sub_sub_state")
  211. AppHarness._poll_for(lambda: state_var_input.get_attribute("value") == "")
  212. AppHarness._poll_for(lambda: input_value_input.get_attribute("value") == "")
  213. # Set the values.
  214. state_var_input.send_keys(var)
  215. input_value_input.send_keys(value)
  216. set_sub_sub_state_button.click()
  217. token = poll_for_token()
  218. # get a reference to all cookie and local storage elements
  219. c1 = driver.find_element(By.ID, "c1")
  220. c2 = driver.find_element(By.ID, "c2")
  221. c3 = driver.find_element(By.ID, "c3")
  222. c4 = driver.find_element(By.ID, "c4")
  223. c5 = driver.find_element(By.ID, "c5")
  224. c6 = driver.find_element(By.ID, "c6")
  225. c7 = driver.find_element(By.ID, "c7")
  226. l1 = driver.find_element(By.ID, "l1")
  227. l2 = driver.find_element(By.ID, "l2")
  228. l3 = driver.find_element(By.ID, "l3")
  229. l4 = driver.find_element(By.ID, "l4")
  230. s1 = driver.find_element(By.ID, "s1")
  231. s2 = driver.find_element(By.ID, "s2")
  232. s3 = driver.find_element(By.ID, "s3")
  233. c1s = driver.find_element(By.ID, "c1s")
  234. l1s = driver.find_element(By.ID, "l1s")
  235. s1s = driver.find_element(By.ID, "s1s")
  236. # assert on defaults where present
  237. assert c1.text == ""
  238. assert c2.text == "c2 default"
  239. assert c3.text == ""
  240. assert c4.text == ""
  241. assert c5.text == ""
  242. assert c6.text == ""
  243. assert c7.text == "c7 default"
  244. assert l1.text == ""
  245. assert l2.text == "l2 default"
  246. assert l3.text == ""
  247. assert l4.text == "l4 default"
  248. assert s1.text == ""
  249. assert s2.text == "s2 default"
  250. assert s3.text == ""
  251. assert c1s.text == ""
  252. assert l1s.text == ""
  253. assert s1s.text == ""
  254. # no cookies should be set yet!
  255. assert not driver.get_cookies()
  256. local_storage_items = local_storage.items()
  257. local_storage_items.pop("chakra-ui-color-mode", None)
  258. local_storage_items.pop("last_compiled_time", None)
  259. assert not local_storage_items
  260. # set some cookies and local storage values
  261. set_sub("c1", "c1 value")
  262. set_sub("c2", "c2 value")
  263. set_sub("c4", "c4 value")
  264. set_sub("c5", "c5 value")
  265. set_sub("c6", "c6 throwaway value")
  266. set_sub("c6", "c6 value")
  267. set_sub("c7", "c7 value")
  268. set_sub("l1", "l1 value")
  269. set_sub("l2", "l2 value")
  270. set_sub("l3", "l3 value")
  271. set_sub("l4", "l4 value")
  272. set_sub("s1", "s1 value")
  273. set_sub("s2", "s2 value")
  274. set_sub("s3", "s3 value")
  275. set_sub_sub("c1s", "c1s value")
  276. set_sub_sub("l1s", "l1s value")
  277. set_sub_sub("s1s", "s1s value")
  278. state_name = client_side.get_full_state_name(["_client_side_state"])
  279. sub_state_name = client_side.get_full_state_name(
  280. ["_client_side_state", "_client_side_sub_state"]
  281. )
  282. sub_sub_state_name = client_side.get_full_state_name(
  283. ["_client_side_state", "_client_side_sub_state", "_client_side_sub_sub_state"]
  284. )
  285. exp_cookies = {
  286. f"{sub_state_name}.c1": {
  287. "domain": "localhost",
  288. "httpOnly": False,
  289. "name": f"{sub_state_name}.c1",
  290. "path": "/",
  291. "sameSite": "Lax",
  292. "secure": False,
  293. "value": "c1%20value",
  294. },
  295. f"{sub_state_name}.c2": {
  296. "domain": "localhost",
  297. "httpOnly": False,
  298. "name": f"{sub_state_name}.c2",
  299. "path": "/",
  300. "sameSite": "Lax",
  301. "secure": False,
  302. "value": "c2%20value",
  303. },
  304. f"{sub_state_name}.c4": {
  305. "domain": "localhost",
  306. "httpOnly": False,
  307. "name": f"{sub_state_name}.c4",
  308. "path": "/",
  309. "sameSite": "Strict",
  310. "secure": False,
  311. "value": "c4%20value",
  312. },
  313. "c6": {
  314. "domain": "localhost",
  315. "httpOnly": False,
  316. "name": "c6",
  317. "path": "/",
  318. "sameSite": "Lax",
  319. "secure": False,
  320. "value": "c6%20value",
  321. },
  322. f"{sub_state_name}.c7": {
  323. "domain": "localhost",
  324. "httpOnly": False,
  325. "name": f"{sub_state_name}.c7",
  326. "path": "/",
  327. "sameSite": "Lax",
  328. "secure": False,
  329. "value": "c7%20value",
  330. },
  331. f"{sub_sub_state_name}.c1s": {
  332. "domain": "localhost",
  333. "httpOnly": False,
  334. "name": f"{sub_sub_state_name}.c1s",
  335. "path": "/",
  336. "sameSite": "Lax",
  337. "secure": False,
  338. "value": "c1s%20value",
  339. },
  340. }
  341. AppHarness._poll_for(
  342. lambda: all(cookie_key in cookie_info_map(driver) for cookie_key in exp_cookies)
  343. )
  344. cookies = cookie_info_map(driver)
  345. for exp_cookie_key, exp_cookie_data in exp_cookies.items():
  346. assert cookies.pop(exp_cookie_key) == exp_cookie_data
  347. # assert all cookies have been popped for this page
  348. assert not cookies
  349. # Test cookie with expiry by itself to avoid timing flakiness
  350. set_sub("c3", "c3 value")
  351. AppHarness._poll_for(lambda: f"{sub_state_name}.c3" in cookie_info_map(driver))
  352. c3_cookie = cookie_info_map(driver)[f"{sub_state_name}.c3"]
  353. assert c3_cookie.pop("expiry") is not None
  354. assert c3_cookie == {
  355. "domain": "localhost",
  356. "httpOnly": False,
  357. "name": f"{sub_state_name}.c3",
  358. "path": "/",
  359. "sameSite": "Lax",
  360. "secure": False,
  361. "value": "c3%20value",
  362. }
  363. time.sleep(2) # wait for c3 to expire
  364. if not isinstance(driver, Firefox):
  365. # Note: Firefox does not remove expired cookies Bug 576347
  366. assert f"{sub_state_name}.c3" not in cookie_info_map(driver)
  367. local_storage_items = local_storage.items()
  368. local_storage_items.pop("chakra-ui-color-mode", None)
  369. local_storage_items.pop("last_compiled_time", None)
  370. assert local_storage_items.pop(f"{sub_state_name}.l1") == "l1 value"
  371. assert local_storage_items.pop(f"{sub_state_name}.l2") == "l2 value"
  372. assert local_storage_items.pop("l3") == "l3 value"
  373. assert local_storage_items.pop(f"{sub_state_name}.l4") == "l4 value"
  374. assert local_storage_items.pop(f"{sub_sub_state_name}.l1s") == "l1s value"
  375. assert not local_storage_items
  376. session_storage_items = session_storage.items()
  377. session_storage_items.pop("token", None)
  378. assert session_storage_items.pop(f"{sub_state_name}.s1") == "s1 value"
  379. assert session_storage_items.pop(f"{sub_state_name}.s2") == "s2 value"
  380. assert session_storage_items.pop("s3") == "s3 value"
  381. assert session_storage_items.pop(f"{sub_sub_state_name}.s1s") == "s1s value"
  382. assert not session_storage_items
  383. assert c1.text == "c1 value"
  384. assert c2.text == "c2 value"
  385. assert c3.text == "c3 value"
  386. assert c4.text == "c4 value"
  387. assert c5.text == "c5 value"
  388. assert c6.text == "c6 value"
  389. assert c7.text == "c7 value"
  390. assert l1.text == "l1 value"
  391. assert l2.text == "l2 value"
  392. assert l3.text == "l3 value"
  393. assert l4.text == "l4 value"
  394. assert s1.text == "s1 value"
  395. assert s2.text == "s2 value"
  396. assert s3.text == "s3 value"
  397. assert c1s.text == "c1s value"
  398. assert l1s.text == "l1s value"
  399. assert s1s.text == "s1s value"
  400. # navigate to the /foo route
  401. with utils.poll_for_navigation(driver):
  402. driver.get(client_side.frontend_url + "/foo")
  403. # get new references to all cookie and local storage elements
  404. c1 = driver.find_element(By.ID, "c1")
  405. c2 = driver.find_element(By.ID, "c2")
  406. c3 = driver.find_element(By.ID, "c3")
  407. c4 = driver.find_element(By.ID, "c4")
  408. c5 = driver.find_element(By.ID, "c5")
  409. c6 = driver.find_element(By.ID, "c6")
  410. c7 = driver.find_element(By.ID, "c7")
  411. l1 = driver.find_element(By.ID, "l1")
  412. l2 = driver.find_element(By.ID, "l2")
  413. l3 = driver.find_element(By.ID, "l3")
  414. l4 = driver.find_element(By.ID, "l4")
  415. s1 = driver.find_element(By.ID, "s1")
  416. s2 = driver.find_element(By.ID, "s2")
  417. s3 = driver.find_element(By.ID, "s3")
  418. c1s = driver.find_element(By.ID, "c1s")
  419. l1s = driver.find_element(By.ID, "l1s")
  420. s1s = driver.find_element(By.ID, "s1s")
  421. assert c1.text == "c1 value"
  422. assert c2.text == "c2 value"
  423. assert c3.text == "" # cookie expired so value removed from state
  424. assert c4.text == "c4 value"
  425. assert c5.text == "c5 value"
  426. assert c6.text == "c6 value"
  427. assert c7.text == "c7 value"
  428. assert l1.text == "l1 value"
  429. assert l2.text == "l2 value"
  430. assert l3.text == "l3 value"
  431. assert l4.text == "l4 value"
  432. assert s1.text == "s1 value"
  433. assert s2.text == "s2 value"
  434. assert s3.text == "s3 value"
  435. assert c1s.text == "c1s value"
  436. assert l1s.text == "l1s value"
  437. assert s1s.text == "s1s value"
  438. # reset the backend state to force refresh from client storage
  439. async with client_side.modify_state(f"{token}_{state_name}") as state:
  440. state.reset()
  441. driver.refresh()
  442. # wait for the backend connection to send the token (again)
  443. token_input = driver.find_element(By.ID, "token")
  444. assert token_input
  445. token = client_side.poll_for_value(token_input)
  446. assert token is not None
  447. # get new references to all cookie and local storage elements (again)
  448. c1 = driver.find_element(By.ID, "c1")
  449. c2 = driver.find_element(By.ID, "c2")
  450. c3 = driver.find_element(By.ID, "c3")
  451. c4 = driver.find_element(By.ID, "c4")
  452. c5 = driver.find_element(By.ID, "c5")
  453. c6 = driver.find_element(By.ID, "c6")
  454. c7 = driver.find_element(By.ID, "c7")
  455. l1 = driver.find_element(By.ID, "l1")
  456. l2 = driver.find_element(By.ID, "l2")
  457. l3 = driver.find_element(By.ID, "l3")
  458. l4 = driver.find_element(By.ID, "l4")
  459. s1 = driver.find_element(By.ID, "s1")
  460. s2 = driver.find_element(By.ID, "s2")
  461. s3 = driver.find_element(By.ID, "s3")
  462. c1s = driver.find_element(By.ID, "c1s")
  463. l1s = driver.find_element(By.ID, "l1s")
  464. s1s = driver.find_element(By.ID, "s1s")
  465. assert c1.text == "c1 value"
  466. assert c2.text == "c2 value"
  467. assert c3.text == "" # temporary cookie expired after reset state!
  468. assert c4.text == "c4 value"
  469. assert c5.text == "c5 value"
  470. assert c6.text == "c6 value"
  471. assert c7.text == "c7 value"
  472. assert l1.text == "l1 value"
  473. assert l2.text == "l2 value"
  474. assert l3.text == "l3 value"
  475. assert l4.text == "l4 value"
  476. assert s1.text == "s1 value"
  477. assert s2.text == "s2 value"
  478. assert s3.text == "s3 value"
  479. assert c1s.text == "c1s value"
  480. assert l1s.text == "l1s value"
  481. assert s1s.text == "s1s value"
  482. # make sure c5 cookie shows up on the `/foo` route
  483. AppHarness._poll_for(lambda: f"{sub_state_name}.c5" in cookie_info_map(driver))
  484. assert cookie_info_map(driver)[f"{sub_state_name}.c5"] == {
  485. "domain": "localhost",
  486. "httpOnly": False,
  487. "name": f"{sub_state_name}.c5",
  488. "path": "/foo/",
  489. "sameSite": "Lax",
  490. "secure": False,
  491. "value": "c5%20value",
  492. }
  493. # Open a new tab to check that sync'd local storage is working
  494. main_tab = driver.window_handles[0]
  495. driver.switch_to.new_window("window")
  496. driver.get(client_side.frontend_url)
  497. # New tab should have a different state token.
  498. assert poll_for_token() != token
  499. # Set values and check them in the new tab.
  500. set_sub("l5", "l5 value")
  501. set_sub("l6", "l6 value")
  502. l5 = driver.find_element(By.ID, "l5")
  503. l6 = driver.find_element(By.ID, "l6")
  504. assert AppHarness._poll_for(lambda: l6.text == "l6 value")
  505. assert l5.text == "l5 value"
  506. # Set session storage values in the new tab
  507. set_sub("s1", "other tab s1")
  508. s1 = driver.find_element(By.ID, "s1")
  509. s2 = driver.find_element(By.ID, "s2")
  510. s3 = driver.find_element(By.ID, "s3")
  511. assert AppHarness._poll_for(lambda: s1.text == "other tab s1")
  512. assert s2.text == "s2 default"
  513. assert s3.text == ""
  514. # Switch back to main window.
  515. driver.switch_to.window(main_tab)
  516. # The values should have updated automatically.
  517. l5 = driver.find_element(By.ID, "l5")
  518. l6 = driver.find_element(By.ID, "l6")
  519. assert AppHarness._poll_for(lambda: l6.text == "l6 value")
  520. assert l5.text == "l5 value"
  521. s1 = driver.find_element(By.ID, "s1")
  522. s2 = driver.find_element(By.ID, "s2")
  523. s3 = driver.find_element(By.ID, "s3")
  524. assert AppHarness._poll_for(lambda: s1.text == "s1 value")
  525. assert s2.text == "s2 value"
  526. assert s3.text == "s3 value"
  527. # clear the cookie jar and local storage, ensure state reset to default
  528. driver.delete_all_cookies()
  529. local_storage.clear()
  530. # refresh the page to trigger re-hydrate
  531. driver.refresh()
  532. # wait for the backend connection to send the token (again)
  533. token_input = driver.find_element(By.ID, "token")
  534. assert token_input
  535. token = client_side.poll_for_value(token_input)
  536. assert token is not None
  537. # all values should be back to their defaults
  538. c1 = driver.find_element(By.ID, "c1")
  539. c2 = driver.find_element(By.ID, "c2")
  540. c3 = driver.find_element(By.ID, "c3")
  541. c4 = driver.find_element(By.ID, "c4")
  542. c5 = driver.find_element(By.ID, "c5")
  543. c6 = driver.find_element(By.ID, "c6")
  544. c7 = driver.find_element(By.ID, "c7")
  545. l1 = driver.find_element(By.ID, "l1")
  546. l2 = driver.find_element(By.ID, "l2")
  547. l3 = driver.find_element(By.ID, "l3")
  548. l4 = driver.find_element(By.ID, "l4")
  549. c1s = driver.find_element(By.ID, "c1s")
  550. l1s = driver.find_element(By.ID, "l1s")
  551. # assert on defaults where present
  552. assert c1.text == ""
  553. assert c2.text == "c2 default"
  554. assert c3.text == ""
  555. assert c4.text == ""
  556. assert c5.text == ""
  557. assert c6.text == ""
  558. assert c7.text == "c7 default"
  559. assert l1.text == ""
  560. assert l2.text == "l2 default"
  561. assert l3.text == ""
  562. assert l4.text == "l4 default"
  563. assert c1s.text == ""
  564. assert l1s.text == ""