diff --git a/test/ruby_ui/accordion_test.rb b/test/ruby_ui/accordion_test.rb
index a6a0c27b..1d5cf5fc 100644
--- a/test/ruby_ui/accordion_test.rb
+++ b/test/ruby_ui/accordion_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::AccordionTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::AccordionTest < ComponentTest
   def test_render_with_default_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Accordion do
         RubyUI.AccordionItem do
           RubyUI.AccordionDefaultTrigger { "Title" }
@@ -19,7 +17,7 @@ def test_render_with_default_items
   end
 
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Accordion do
         RubyUI.AccordionItem do
           RubyUI.AccordionTrigger do |trigger|
diff --git a/test/ruby_ui/alert_dialog_test.rb b/test/ruby_ui/alert_dialog_test.rb
index acb61f33..9c27fb71 100644
--- a/test/ruby_ui/alert_dialog_test.rb
+++ b/test/ruby_ui/alert_dialog_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::AlertDialogTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::AlertDialogTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.AlertDialog do
         RubyUI.AlertDialogTrigger do
           RubyUI.Button { "Show dialog" }
diff --git a/test/ruby_ui/alert_test.rb b/test/ruby_ui/alert_test.rb
index 7e407909..00b44365 100644
--- a/test/ruby_ui/alert_test.rb
+++ b/test/ruby_ui/alert_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::AlertTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::AlertTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Alert do
         RubyUI.AlertTitle { "Pro tip" }
         RubyUI.AlertDescription { "Simply, don't include an icon and your alert will look like this." }
diff --git a/test/ruby_ui/aspect_ratio_test.rb b/test/ruby_ui/aspect_ratio_test.rb
index 5e63fe60..cf2b1ddf 100644
--- a/test/ruby_ui/aspect_ratio_test.rb
+++ b/test/ruby_ui/aspect_ratio_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::AspectRatioTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::AspectRatioTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.AspectRatio(aspect_ratio: "16/9") do |aspect|
         aspect.img(
           alt: "Placeholder",
diff --git a/test/ruby_ui/avatar_test.rb b/test/ruby_ui/avatar_test.rb
index 4be51538..e4b8e550 100644
--- a/test/ruby_ui/avatar_test.rb
+++ b/test/ruby_ui/avatar_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::AvatarTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::AvatarTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Avatar do
         RubyUI.AvatarImage(src: "https://avatars.githubusercontent.com/u/246692?v=4", alt: "joeldrapper")
         RubyUI.AvatarFallback { "JD" }
diff --git a/test/ruby_ui/badge_test.rb b/test/ruby_ui/badge_test.rb
index 18de47b4..44ddacf4 100644
--- a/test/ruby_ui/badge_test.rb
+++ b/test/ruby_ui/badge_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::BadgeTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::BadgeTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Badge { "Badge" }
     end
 
diff --git a/test/ruby_ui/button_test.rb b/test/ruby_ui/button_test.rb
index 3e6a5f59..ffa10a62 100644
--- a/test/ruby_ui/button_test.rb
+++ b/test/ruby_ui/button_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::ButtonTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::ButtonTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Button(variant: :primary) { "Primary" }
     end
 
diff --git a/test/ruby_ui/calendar_test.rb b/test/ruby_ui/calendar_test.rb
index 1f1fe378..5c3d4745 100644
--- a/test/ruby_ui/calendar_test.rb
+++ b/test/ruby_ui/calendar_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::CalendarTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::CalendarTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Input(type: "string", placeholder: "Select a date", class: "rounded-md border shadow", id: "date", data_controller: "ruby-ui--input")
       RubyUI.Calendar(input_id: "#date", class: "rounded-md border shadow")
     end
diff --git a/test/ruby_ui/card_test.rb b/test/ruby_ui/card_test.rb
index 1054d600..d6aa6f94 100644
--- a/test/ruby_ui/card_test.rb
+++ b/test/ruby_ui/card_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::CardTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::CardTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Card(class: "w-96") do
         RubyUI.CardHeader do
           RubyUI.CardTitle { 'You might like "RubyUI"' }
diff --git a/test/ruby_ui/chart_test.rb b/test/ruby_ui/chart_test.rb
index 2b0d2894..b565a11c 100644
--- a/test/ruby_ui/chart_test.rb
+++ b/test/ruby_ui/chart_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::ChartTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::ChartTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       options = {
         type: "bar",
         data: {
diff --git a/test/ruby_ui/checkbox_test.rb b/test/ruby_ui/checkbox_test.rb
index 03daca3f..2c9fe5c3 100644
--- a/test/ruby_ui/checkbox_test.rb
+++ b/test/ruby_ui/checkbox_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::CheckboxTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::CheckboxTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Checkbox(id: "terms")
     end
 
diff --git a/test/ruby_ui/clipboard_test.rb b/test/ruby_ui/clipboard_test.rb
index 02bd5017..dd0d6c84 100644
--- a/test/ruby_ui/clipboard_test.rb
+++ b/test/ruby_ui/clipboard_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::ClipboardTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::ClipboardTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Clipboard(success: "Copied!", error: "Copy Failed!")
     end
 
diff --git a/test/ruby_ui/codeblock_test.rb b/test/ruby_ui/codeblock_test.rb
index 7a2fddf7..49c7d9b1 100644
--- a/test/ruby_ui/codeblock_test.rb
+++ b/test/ruby_ui/codeblock_test.rb
@@ -2,9 +2,7 @@
 
 require "test_helper"
 
-class RubyUI::CodeblockTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::CodeblockTest < ComponentTest
   def test_render_with_all_items
     code = <<~CODE
       def hello_world
@@ -12,7 +10,7 @@ def hello_world
       end
     CODE
 
-    output = phlex_context do
+    output = phlex do
       RubyUI.Codeblock(code, syntax: :ruby)
     end
 
diff --git a/test/ruby_ui/collapsible_test.rb b/test/ruby_ui/collapsible_test.rb
index 7221b3af..dfc311e2 100644
--- a/test/ruby_ui/collapsible_test.rb
+++ b/test/ruby_ui/collapsible_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::CollapsibleTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::CollapsibleTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Collapsible(open: true) do
         RubyUI.CollapsibleTrigger do
           RubyUI.Button(variant: :ghost, icon: true) do |button|
diff --git a/test/ruby_ui/combobox_test.rb b/test/ruby_ui/combobox_test.rb
index fc108a85..1354096c 100644
--- a/test/ruby_ui/combobox_test.rb
+++ b/test/ruby_ui/combobox_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::ComboboxTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::ComboboxTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Combobox do
         RubyUI.ComboboxInput()
         RubyUI.ComboboxTrigger do
diff --git a/test/ruby_ui/command_test.rb b/test/ruby_ui/command_test.rb
index 95524407..1a4011a9 100644
--- a/test/ruby_ui/command_test.rb
+++ b/test/ruby_ui/command_test.rb
@@ -2,9 +2,7 @@
 
 require "test_helper"
 
-class RubyUI::CommandTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::CommandTest < ComponentTest
   def test_render_with_all_items
     components_list = [
       {name: "Accordion", path: "#"},
@@ -21,7 +19,7 @@ def test_render_with_all_items
       {name: "Settings", path: "#"}
     ]
 
-    output = phlex_context do
+    output = phlex do
       RubyUI.CommandDialog do
         RubyUI.CommandDialogTrigger do
           RubyUI.Button(variant: "outline", class: "w-56 pr-2 pl-3 justify-between") do |button|
diff --git a/test/ruby_ui/context_menu_test.rb b/test/ruby_ui/context_menu_test.rb
index 64bd4bf8..1dc50dab 100644
--- a/test/ruby_ui/context_menu_test.rb
+++ b/test/ruby_ui/context_menu_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::ContextMenuTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::ContextMenuTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.ContextMenu do
         RubyUI.ContextMenuTrigger(class: "flex h-[150px] w-[300px] items-center justify-center rounded-md border border-dashed text-sm") { "Right click here" }
         RubyUI.ContextMenuContent(class: "w-64") do
diff --git a/test/ruby_ui/dialog_test.rb b/test/ruby_ui/dialog_test.rb
index 73f930fd..b6814feb 100644
--- a/test/ruby_ui/dialog_test.rb
+++ b/test/ruby_ui/dialog_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::DialogTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::DialogTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Dialog do
         RubyUI.DialogTrigger do
           RubyUI.Button { "Open Dialog" }
diff --git a/test/ruby_ui/dropdown_menu_test.rb b/test/ruby_ui/dropdown_menu_test.rb
index 4e0df149..fce350c1 100644
--- a/test/ruby_ui/dropdown_menu_test.rb
+++ b/test/ruby_ui/dropdown_menu_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::DropdownMenuTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::DropdownMenuTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.DropdownMenu do
         RubyUI.DropdownMenuTrigger(class: "w-full") do
           RubyUI.Button(variant: :outline) { "Open" }
diff --git a/test/ruby_ui/form_test.rb b/test/ruby_ui/form_test.rb
index 33871e71..067dcca7 100644
--- a/test/ruby_ui/form_test.rb
+++ b/test/ruby_ui/form_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::FormTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::FormTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Form do
         RubyUI.FormField do
           RubyUI.FormFieldLabel { "Label" }
diff --git a/test/ruby_ui/hover_card_test.rb b/test/ruby_ui/hover_card_test.rb
index 16bc1f14..4a578820 100644
--- a/test/ruby_ui/hover_card_test.rb
+++ b/test/ruby_ui/hover_card_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::HoverCardTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::HoverCardTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.HoverCard do
         RubyUI.HoverCardTrigger do
           RubyUI.Button(variant: :link) { "@joeldrapper" }
diff --git a/test/ruby_ui/inline_code_test.rb b/test/ruby_ui/inline_code_test.rb
index 501c3330..eb17bcd5 100644
--- a/test/ruby_ui/inline_code_test.rb
+++ b/test/ruby_ui/inline_code_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::InlineCodeTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::InlineCodeTest < ComponentTest
   def test_render_inline_code
-    output = phlex_context do
+    output = phlex do
       RubyUI::InlineCode() { "This is an inline code block" }
     end
 
diff --git a/test/ruby_ui/inline_link_test.rb b/test/ruby_ui/inline_link_test.rb
index 6f646fda..6446bdb1 100644
--- a/test/ruby_ui/inline_link_test.rb
+++ b/test/ruby_ui/inline_link_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::InlineLinkTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::InlineLinkTest < ComponentTest
   def test_render_inline_link
-    output = phlex_context do
+    output = phlex do
       RubyUI::InlineLink(href: "#") { "Link" }
     end
 
diff --git a/test/ruby_ui/input_test.rb b/test/ruby_ui/input_test.rb
index 7f6d48e8..9194ee4a 100644
--- a/test/ruby_ui/input_test.rb
+++ b/test/ruby_ui/input_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::InputTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::InputTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Input(type: "email", placeholder: "Email")
     end
 
@@ -14,7 +12,7 @@ def test_render_with_all_items
   end
 
   def test_render_with_value
-    output = phlex_context do
+    output = phlex do
       RubyUI.Input(type: "email", value: "user@email.com")
     end
 
diff --git a/test/ruby_ui/link_test.rb b/test/ruby_ui/link_test.rb
index 85b2e6aa..12a714fd 100644
--- a/test/ruby_ui/link_test.rb
+++ b/test/ruby_ui/link_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::LinkTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::LinkTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Link(href: "#") { "Link" }
     end
 
diff --git a/test/ruby_ui/masked_input_test.rb b/test/ruby_ui/masked_input_test.rb
index d9d4580e..5e291ef1 100644
--- a/test/ruby_ui/masked_input_test.rb
+++ b/test/ruby_ui/masked_input_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::MaskedInputTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::MaskedInputTest < ComponentTest
   def test_render
-    output = phlex_context do
+    output = phlex do
       RubyUI.MaskedInput(data: {maska: "#####-###"})
     end
 
diff --git a/test/ruby_ui/pagination_test.rb b/test/ruby_ui/pagination_test.rb
index 036c2c95..8759b4b1 100644
--- a/test/ruby_ui/pagination_test.rb
+++ b/test/ruby_ui/pagination_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::PaginationTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::PaginationTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Pagination do
         RubyUI.PaginationContent do
           RubyUI.PaginationItem(href: "#") do |item|
diff --git a/test/ruby_ui/popover_test.rb b/test/ruby_ui/popover_test.rb
index 4baccc9e..cc5b8656 100644
--- a/test/ruby_ui/popover_test.rb
+++ b/test/ruby_ui/popover_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::PopoverTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::PopoverTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Popover do
         RubyUI.PopoverTrigger(class: "w-full") do
           RubyUI.Button(variant: :outline) { "Open Popover" }
diff --git a/test/ruby_ui/select_test.rb b/test/ruby_ui/select_test.rb
index 14e223ab..b1f5b1be 100644
--- a/test/ruby_ui/select_test.rb
+++ b/test/ruby_ui/select_test.rb
@@ -2,9 +2,7 @@
 
 require "test_helper"
 
-class RubyUI::SelectTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::SelectTest < ComponentTest
   def test_render_with_all_items
     people = [
       ["John Doe", 1],
@@ -12,7 +10,7 @@ def test_render_with_all_items
       ["Sam Smith", 3]
     ]
 
-    output = phlex_context do
+    output = phlex do
       RubyUI.Select do
         RubyUI.SelectInput(name: "NAME")
         RubyUI.SelectTrigger do
diff --git a/test/ruby_ui/sheet_test.rb b/test/ruby_ui/sheet_test.rb
index cde80005..3b5fe46a 100644
--- a/test/ruby_ui/sheet_test.rb
+++ b/test/ruby_ui/sheet_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::SheetTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::SheetTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Sheet do
         RubyUI.SheetTrigger do
           RubyUI.Button(variant: :outline) { "Open Sheet" }
diff --git a/test/ruby_ui/shortcut_key_test.rb b/test/ruby_ui/shortcut_key_test.rb
index 55b12262..03c03a20 100644
--- a/test/ruby_ui/shortcut_key_test.rb
+++ b/test/ruby_ui/shortcut_key_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::ShortcutKeyTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::ShortcutKeyTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.ShortcutKey do |shortcut|
         shortcut.span(class: "text-xs") { "⌘" }
         shortcut.plain "K"
diff --git a/test/ruby_ui/table_test.rb b/test/ruby_ui/table_test.rb
index 3983f28d..2ae1e4ba 100644
--- a/test/ruby_ui/table_test.rb
+++ b/test/ruby_ui/table_test.rb
@@ -2,9 +2,7 @@
 
 require "test_helper"
 
-class RubyUI::TableTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::TableTest < ComponentTest
   def test_render_with_all_items
     invoices = [
       {identifier: "INV-0001", status: "Active", method: "Credit Card", amount: 100},
@@ -13,7 +11,7 @@ def test_render_with_all_items
       {identifier: "INV-0004", status: "Inactive", method: "Credit Card", amount: 100}
     ]
 
-    output = phlex_context do
+    output = phlex do
       RubyUI.Table do
         RubyUI.TableCaption { "Employees at Acme inc." }
         RubyUI.TableHeader do
diff --git a/test/ruby_ui/tabs_test.rb b/test/ruby_ui/tabs_test.rb
index 374cccc1..f0e028e6 100644
--- a/test/ruby_ui/tabs_test.rb
+++ b/test/ruby_ui/tabs_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::TabsTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::TabsTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Tabs(default_value: "account", class: "w-96") do
         RubyUI.TabsList do
           RubyUI.TabsTrigger(value: "account") { "Account" }
diff --git a/test/ruby_ui/text_test.rb b/test/ruby_ui/text_test.rb
index dba4f73d..1817688e 100644
--- a/test/ruby_ui/text_test.rb
+++ b/test/ruby_ui/text_test.rb
@@ -2,12 +2,10 @@
 
 require "test_helper"
 
-class RubyUI::TypographyTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::TypographyTest < ComponentTest
   def test_heading_with_levels
     (1..4).each do |level|
-      output = phlex_context do
+      output = phlex do
         RubyUI::Heading(level: level.to_s) { "This is an H#{level} title" }
       end
 
@@ -17,7 +15,7 @@ def test_heading_with_levels
   end
 
   def test_heading_with_custom_size
-    output = phlex_context do
+    output = phlex do
       RubyUI::Heading(as: "h2", size: "7") { "Custom Heading" }
     end
 
@@ -28,7 +26,7 @@ def test_heading_with_custom_size
 
   def test_text_with_sizes
     (1..9).each do |size|
-      output = phlex_context do
+      output = phlex do
         RubyUI::Text(size: size.to_s) { "Size #{size} text" }
       end
 
@@ -47,7 +45,7 @@ def test_text_with_sizes
 
   def test_text_with_weights
     %w[light regular medium bold].each do |weight|
-      output = phlex_context do
+      output = phlex do
         RubyUI::Text(weight: weight) { "#{weight.capitalize} text" }
       end
 
@@ -58,7 +56,7 @@ def test_text_with_weights
 
   def test_text_as_different_elements
     %w[p span div label].each do |element|
-      output = phlex_context do
+      output = phlex do
         RubyUI::Text(as: element) { element.capitalize.to_s }
       end
 
@@ -68,7 +66,7 @@ def test_text_as_different_elements
   end
 
   def test_lead_text
-    output = phlex_context do
+    output = phlex do
       RubyUI::Text(as: "p", size: "5", weight: "medium") { "A modal dialog that interrupts the user." }
     end
 
@@ -78,7 +76,7 @@ def test_lead_text
   end
 
   def test_large_text
-    output = phlex_context do
+    output = phlex do
       RubyUI::Text(size: "5", weight: "semibold") { "Are you sure absolutely sure?" }
     end
 
@@ -88,7 +86,7 @@ def test_large_text
   end
 
   def test_small_text
-    output = phlex_context do
+    output = phlex do
       RubyUI::Text(size: "2", weight: "medium") { "Email address" }
     end
 
@@ -98,7 +96,7 @@ def test_small_text
   end
 
   def test_muted_text
-    output = phlex_context do
+    output = phlex do
       RubyUI::Text(size: "2", class: "text-muted-foreground") { "Enter your email address." }
     end
 
diff --git a/test/ruby_ui/textarea_test.rb b/test/ruby_ui/textarea_test.rb
index 5513ff22..c3f240ef 100644
--- a/test/ruby_ui/textarea_test.rb
+++ b/test/ruby_ui/textarea_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::TextareaTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::TextareaTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Textarea(rows: 4, placeholder: "Comment")
     end
 
@@ -14,7 +12,7 @@ def test_render_with_all_items
   end
 
   def test_render_with_value
-    output = phlex_context do
+    output = phlex do
       RubyUI.Textarea { "Value" }
     end
 
diff --git a/test/ruby_ui/theme_toggle_test.rb b/test/ruby_ui/theme_toggle_test.rb
index d514aa27..42a59c00 100644
--- a/test/ruby_ui/theme_toggle_test.rb
+++ b/test/ruby_ui/theme_toggle_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::ThemeToggleTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::ThemeToggleTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.ThemeToggle do |toggle|
         toggle.light_mode do
           RubyUI.Button(variant: :primary) { "Light" }
diff --git a/test/ruby_ui/tooltip_test.rb b/test/ruby_ui/tooltip_test.rb
index eca36ebf..a3ccb430 100644
--- a/test/ruby_ui/tooltip_test.rb
+++ b/test/ruby_ui/tooltip_test.rb
@@ -2,11 +2,9 @@
 
 require "test_helper"
 
-class RubyUI::TooltipTest < Minitest::Test
-  include Phlex::Testing::ViewHelper
-
+class RubyUI::TooltipTest < ComponentTest
   def test_render_with_all_items
-    output = phlex_context do
+    output = phlex do
       RubyUI.Tooltip do
         RubyUI.TooltipTrigger do
           RubyUI.Button(variant: :outline, icon: true) { "?" }
diff --git a/test/test_helper.rb b/test/test_helper.rb
index 06aa4c68..fce11baf 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -17,20 +17,16 @@ module RubyUI
   end
 end
 
-def phlex_context(&)
-  render Phlex::HTML.new, &
-end
-
-module Phlex
-  module Testing
-    module ViewHelper
-      def render(component, &)
-        component.call(view_context:, &)
-      end
+class ComponentTest < Minitest::Test
+  def render(component, &)
+    component.call(view_context:, &)
+  end
 
-      def view_context = nil
-    end
+  def phlex(&)
+    render Phlex::HTML.new, &
   end
+
+  def view_context = nil
 end
 
 # this is a tracepoint that will output the path of all files loaded that contain the string "phlex"