I can persist tabmanager state

tests/controls/test_tabsmanager.py

@@ -3,6 +3,7 @@ from fasthtml.components import *
 
 from myfasthtml.controls.TabsManager import TabsManager
 from myfasthtml.core.instances import InstancesManager
+from myfasthtml.test.matcher import matches, NoChildren
 from .conftest import session
 
 
@@ -13,18 +14,75 @@ def tabs_manager(session):
   InstancesManager.reset()
 
 
-def test_tabs_manager_is_registered(session, tabs_manager):
-  from_instance_manager = InstancesManager.get(session, tabs_manager.get_id())
-  assert from_instance_manager == tabs_manager
-
-
-def test_i_can_add_tab(tabs_manager):
-  tab_id = tabs_manager.add_tab("Users", Div("Content 1"))
+class TestTabsManagerBehaviour:
+  def test_tabs_manager_is_registered(self, session, tabs_manager):
+    from_instance_manager = InstancesManager.get(session, tabs_manager.get_id())
+    assert from_instance_manager == tabs_manager
   
-  assert tab_id is not None
-  assert tab_id in tabs_manager.get_state().tabs
-  assert tabs_manager.get_state().tabs[tab_id]["label"] == "Users"
-  assert tabs_manager.get_state().tabs[tab_id]["component_type"] is None  # Div is not BaseInstance
-  assert tabs_manager.get_state().tabs[tab_id]["component_id"] is None  # Div is not BaseInstance
-  assert tabs_manager.get_state().tabs_order == [tab_id]
-  assert tabs_manager.get_state().active_tab == tab_id
+  def test_i_can_add_tab(self, tabs_manager):
+    tab_id = tabs_manager.add_tab("Users", Div("Content 1"))
+    
+    assert tab_id is not None
+    assert tab_id in tabs_manager.get_state().tabs
+    assert tabs_manager.get_state().tabs[tab_id]["label"] == "Users"
+    assert tabs_manager.get_state().tabs[tab_id]["component_type"] is None  # Div is not BaseInstance
+    assert tabs_manager.get_state().tabs[tab_id]["component_id"] is None  # Div is not BaseInstance
+    assert tabs_manager.get_state().tabs_order == [tab_id]
+    assert tabs_manager.get_state().active_tab == tab_id
+  
+  def test_i_can_add_multiple_tabs(self, tabs_manager):
+    tab_id1 = tabs_manager.add_tab("Users", Div("Content 1"))
+    tab_id2 = tabs_manager.add_tab("User2", Div("Content 2"))
+    
+    assert len(tabs_manager.get_state().tabs) == 2
+    assert tabs_manager.get_state().tabs_order == [tab_id1, tab_id2]
+    assert tabs_manager.get_state().active_tab == tab_id2
+
+
+class TestTabsManagerRender:
+  def test_i_can_render_when_no_tabs(self, tabs_manager):
+    res = tabs_manager.render()
+    
+    expected = Div(
+      Div(NoChildren(), id=f"{tabs_manager.get_id()}-header"),
+      Div(id=f"{tabs_manager.get_id()}-content"),
+      id=tabs_manager.get_id(),
+    )
+    
+    assert matches(res, expected)
+  
+  def test_i_can_render_when_one_tab(self, tabs_manager):
+    tabs_manager.add_tab("Users", Div("Content 1"))
+    res = tabs_manager.render()
+    
+    expected = Div(
+      Div(
+        Button(),
+        id=f"{tabs_manager.get_id()}-header"
+      ),
+      Div(
+        Div("Content 1")
+      ),
+      id=tabs_manager.get_id(),
+    )
+    assert matches(res, expected)
+  
+  def test_i_can_render_when_multiple_tabs(self, tabs_manager):
+    tabs_manager.add_tab("Users1", Div("Content 1"))
+    tabs_manager.add_tab("Users2", Div("Content 2"))
+    tabs_manager.add_tab("Users3", Div("Content 3"))
+    res = tabs_manager.render()
+    
+    expected = Div(
+      Div(
+        Button(),
+        Button(),
+        Button(),
+        id=f"{tabs_manager.get_id()}-header"
+      ),
+      Div(
+        Div("Content 3")
+      ),
+      id=tabs_manager.get_id(),
+    )
+    assert matches(res, expected)