@@ -10,7 +10,11 @@ def __init__(self, page, live_server):
1010 super ().__init__ (page , live_server )
1111
1212 self .bucket_items = self .page .get_by_test_id ("bucket-row" )
13- self .show_triaged_toggle = self .page .get_by_test_id ("show-triaged-toggle" )
13+ self .show_all = self .page .get_by_test_id ("all" )
14+ self .show_needs_triage = self .page .get_by_test_id ("needs_triage" )
15+ self .show_triaged = self .page .get_by_test_id ("triaged" )
16+ self .domain_input = self .page .get_by_test_id ("domain-input" )
17+ self .domain_search_btn = self .page .get_by_test_id ("domain-search-btn" )
1418
1519 def bucket_count (self ):
1620 """Return the number of buckets displayed on the page."""
@@ -20,10 +24,30 @@ def wait_for_buckets_to_load(self):
2024 """Wait for bucket rows to appear on the page."""
2125 self .bucket_items .first .wait_for (state = "visible" )
2226
23- def click_show_triaged_toggle (self ):
24- """Toggle between Show Triaged / Hide Triaged."""
27+ def click_on_state (self , locator ):
2528 with self .page .expect_response (
2629 lambda r : "/reportmanager/rest/buckets/" in r .url
2730 and r .request .method == "GET"
2831 ):
29- self .show_triaged_toggle .click ()
32+ locator .click ()
33+
34+ def click_show_all (self ):
35+ """Click All button to show all buckets including triaged."""
36+ self .click_on_state (self .show_all )
37+
38+ def click_needs_triage (self ):
39+ """Click Needs Triage to hide triaged buckets."""
40+ self .click_on_state (self .show_needs_triage )
41+
42+ def click_triaged (self ):
43+ """Click Triaged to show only triaged buckets."""
44+ self .click_on_state (self .show_triaged )
45+
46+ def search_domain (self , domain ):
47+ """Type a domain into the filter input and click Search."""
48+ self .domain_input .fill (domain )
49+ with self .page .expect_response (
50+ lambda r : "/reportmanager/rest/buckets/" in r .url
51+ and r .request .method == "GET"
52+ ):
53+ self .domain_search_btn .click ()
0 commit comments