173 lines
5.0 KiB
Python
173 lines
5.0 KiB
Python
"""Test states for mutable vars."""
|
|
|
|
from typing import Dict, List, Set, Union
|
|
|
|
import reflex as rx
|
|
|
|
|
|
class DictMutationTestState(rx.State):
|
|
"""A state for testing ReflexDict mutation."""
|
|
|
|
# plain dict
|
|
details = {"name": "Tommy"}
|
|
|
|
def add_age(self):
|
|
"""Add an age to the dict."""
|
|
self.details.update({"age": 20}) # type: ignore
|
|
|
|
def change_name(self):
|
|
"""Change the name in the dict."""
|
|
self.details["name"] = "Jenny"
|
|
|
|
def remove_last_detail(self):
|
|
"""Remove the last item in the dict."""
|
|
self.details.popitem()
|
|
|
|
def clear_details(self):
|
|
"""Clear the dict."""
|
|
self.details.clear()
|
|
|
|
def remove_name(self):
|
|
"""Remove the name from the dict."""
|
|
del self.details["name"]
|
|
|
|
def pop_out_age(self):
|
|
"""Pop out the age from the dict."""
|
|
self.details.pop("age")
|
|
|
|
# dict in list
|
|
address = [{"home": "home address"}, {"work": "work address"}]
|
|
|
|
def remove_home_address(self):
|
|
"""Remove the home address from dict in the list."""
|
|
self.address[0].pop("home")
|
|
|
|
def add_street_to_home_address(self):
|
|
"""Set street key in the dict in the list."""
|
|
self.address[0]["street"] = "street address"
|
|
|
|
# nested dict
|
|
friend_in_nested_dict = {"name": "Nikhil", "friend": {"name": "Alek"}}
|
|
|
|
def change_friend_name(self):
|
|
"""Change the friend's name in the nested dict."""
|
|
self.friend_in_nested_dict["friend"]["name"] = "Tommy"
|
|
|
|
def remove_friend(self):
|
|
"""Remove the friend from the nested dict."""
|
|
self.friend_in_nested_dict.pop("friend")
|
|
|
|
def add_friend_age(self):
|
|
"""Add an age to the friend in the nested dict."""
|
|
self.friend_in_nested_dict["friend"]["age"] = 30
|
|
|
|
|
|
class ListMutationTestState(rx.State):
|
|
"""A state for testing ReflexList mutation."""
|
|
|
|
# plain list
|
|
plain_friends = ["Tommy"]
|
|
|
|
def make_friend(self):
|
|
"""Add a friend to the list."""
|
|
self.plain_friends.append("another-fd")
|
|
|
|
def change_first_friend(self):
|
|
"""Change the first friend in the list."""
|
|
self.plain_friends[0] = "Jenny"
|
|
|
|
def unfriend_all_friends(self):
|
|
"""Unfriend all friends in the list."""
|
|
self.plain_friends.clear()
|
|
|
|
def unfriend_first_friend(self):
|
|
"""Unfriend the first friend in the list."""
|
|
del self.plain_friends[0]
|
|
|
|
def remove_last_friend(self):
|
|
"""Remove the last friend in the list."""
|
|
self.plain_friends.pop()
|
|
|
|
def make_friends_with_colleagues(self):
|
|
"""Add list of friends to the list."""
|
|
colleagues = ["Peter", "Jimmy"]
|
|
self.plain_friends.extend(colleagues)
|
|
|
|
def remove_tommy(self):
|
|
"""Remove Tommy from the list."""
|
|
self.plain_friends.remove("Tommy")
|
|
|
|
# list in dict
|
|
friends_in_dict = {"Tommy": ["Jenny"]}
|
|
|
|
def remove_jenny_from_tommy(self):
|
|
"""Remove Jenny from Tommy's friends list."""
|
|
self.friends_in_dict["Tommy"].remove("Jenny")
|
|
|
|
def add_jimmy_to_tommy_friends(self):
|
|
"""Add Jimmy to Tommy's friends list."""
|
|
self.friends_in_dict["Tommy"].append("Jimmy")
|
|
|
|
def tommy_has_no_fds(self):
|
|
"""Clear Tommy's friends list."""
|
|
self.friends_in_dict["Tommy"].clear()
|
|
|
|
# nested list
|
|
friends_in_nested_list = [["Tommy"], ["Jenny"]]
|
|
|
|
def remove_first_group(self):
|
|
"""Remove the first group of friends from the nested list."""
|
|
self.friends_in_nested_list.pop(0)
|
|
|
|
def remove_first_person_from_first_group(self):
|
|
"""Remove the first person from the first group of friends in the nested list."""
|
|
self.friends_in_nested_list[0].pop(0)
|
|
|
|
def add_jimmy_to_second_group(self):
|
|
"""Add Jimmy to the second group of friends in the nested list."""
|
|
self.friends_in_nested_list[1].append("Jimmy")
|
|
|
|
|
|
class OtherBase(rx.Base):
|
|
"""A Base model with a str field."""
|
|
|
|
bar: str = ""
|
|
|
|
|
|
class CustomVar(rx.Base):
|
|
"""A Base model with multiple fields."""
|
|
|
|
foo: str = ""
|
|
array: List[str] = []
|
|
hashmap: Dict[str, str] = {}
|
|
test_set: Set[str] = set()
|
|
custom: OtherBase = OtherBase()
|
|
|
|
|
|
class MutableTestState(rx.State):
|
|
"""A test state."""
|
|
|
|
array: List[Union[str, List, Dict[str, str]]] = [
|
|
"value",
|
|
[1, 2, 3],
|
|
{"key": "value"},
|
|
]
|
|
hashmap: Dict[str, Union[List, str, Dict[str, str]]] = {
|
|
"key": ["list", "of", "values"],
|
|
"another_key": "another_value",
|
|
"third_key": {"key": "value"},
|
|
}
|
|
test_set: Set[Union[str, int]] = {1, 2, 3, 4, "five"}
|
|
custom: CustomVar = CustomVar()
|
|
_be_custom: CustomVar = CustomVar()
|
|
|
|
def reassign_mutables(self):
|
|
"""Assign mutable fields to different values."""
|
|
self.array = ["modified_value", [1, 2, 3], {"mod_key": "mod_value"}]
|
|
self.hashmap = {
|
|
"mod_key": ["list", "of", "values"],
|
|
"mod_another_key": "another_value",
|
|
"mod_third_key": {"key": "value"},
|
|
}
|
|
self.test_set = {1, 2, 3, 4, "five"}
|