Extract superclass

ministryofjustice · Feb 14, 2024 · ca287f9 · ca287f9
1 parent 272f121
commit ca287f9
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/selenium/conftest.py b/tests/selenium/conftest.py
@@ -37,15 +37,17 @@ def screenshot(name="screenshot"):
     yield screenshot
 
 
-class DetailsPage:
+class Page:
     def __init__(self, selenium):
         self.selenium = selenium
 
+
+class DetailsPage(Page):
     def secondary_heading(self):
         return self.selenium.find_element(By.TAG_NAME, "h2")
 
 
-class HomePage:
+class HomePage(Page):
     def __init__(self, selenium):
         self.selenium = selenium
 
@@ -64,7 +66,7 @@ def link(self):
         return self.element.find_element(By.CSS_SELECTOR, "h3 a")
 
 
-class SearchPage:
+class SearchPage(Page):
     def __init__(self, selenium):
         self.selenium = selenium