Add init in child classes

 Add inits in child classes to overwrite base page class. Add locators in self.locator to make code cleaner and easier to expand.

Author: mkopr

pages.py

@@ -10,33 +10,41 @@
 # Depends on the page functionality we can have more functions for new classes
 
 class MainPage(Page):
+    def __init__(self, driver):
+        self.locator = MainPageLocatars
+        super(MainPage, self).__init__(driver)
+
     def check_page_loaded(self):
-        return True if self.find_element(*MainPageLocatars.LOGO) else False
+        return True if self.find_element(*self.locator.LOGO) else False
 
     def search_item(self, item):
-        self.find_element(*MainPageLocatars.SEARCH).send_keys(item)
-        self.find_element(*MainPageLocatars.SEARCH).send_keys(Keys.ENTER)
-        return self.find_element(*MainPageLocatars.SEARCH_LIST).text
+        self.find_element(*self.locator.SEARCH).send_keys(item)
+        self.find_element(*self.locator.SEARCH).send_keys(Keys.ENTER)
+        return self.find_element(*self.locator.SEARCH_LIST).text
 
     def click_sign_up_button(self):
-        self.hover(*MainPageLocatars.ACCOUNT)
-        self.find_element(*MainPageLocatars.SIGNUP).click()
+        self.hover(*self.locator.ACCOUNT)
+        self.find_element(*self.locator.SIGNUP).click()
         return SignUpPage(self.driver)
 
     def click_sign_in_button(self):
-        self.hover(*MainPageLocatars.ACCOUNT)
-        self.find_element(*MainPageLocatars.LOGIN).click()
+        self.hover(*self.locator.ACCOUNT)
+        self.find_element(*self.locator.LOGIN).click()
         return LoginPage(self.driver)
 
 class LoginPage(Page):
+    def __init__(self, driver):
+        self.locator = LoginPageLocatars
+        super(LoginPage, self).__init__(driver)
+
     def enter_email(self, user):
-        self.find_element(*LoginPageLocatars.EMAIL).send_keys(users.get_user(user)["email"])
+        self.find_element(*self.locator.EMAIL).send_keys(users.get_user(user)["email"])
 
     def enter_password(self, user):
-        self.find_element(*LoginPageLocatars.PASSWORD).send_keys(users.get_user(user)["password"])
+        self.find_element(*self.locator.PASSWORD).send_keys(users.get_user(user)["password"])
 
     def click_login_button(self):
-        self.find_element(*LoginPageLocatars.SUBMIT).click()
+        self.find_element(*self.locator.SUBMIT).click()
 
     def login(self, user):
         self.enter_email(user)
@@ -49,7 +57,7 @@ def login_with_valid_user(self, user):
 
     def login_with_in_valid_user(self, user):
         self.login(user)
-        return self.find_element(*LoginPageLocatars.ERROR_MESSAGE).text    
+        return self.find_element(*self.locator.ERROR_MESSAGE).text
 
 class HomePage(Page):
     pass