def test_add_label(self): # Creating object for test driver = self.driver login = LoginPage(driver) home = HomePage(driver) settings = SettingsPage(driver) foldersandabels = FoldersAndLabelsPage(driver) # Open a Firefox window driver.get('https://beta.protonmail.com/') # Connect to ProtonMail login.enter_username('*****@*****.**') login.enter_password('qwerty123') login.click_login() time.sleep(1) # Close PopUp Window and Click on Settings Tab home.click_not_now() home.click_settings_tab() time.sleep(1) # Click on the Folders and Labels link settings.click_foldersandlabels() time.sleep(1) # Add a Label and Submit foldersandabels.create_label('MyLabelSubmit') foldersandabels.click_submit() time.sleep(1) # Add a Label and Cancel foldersandabels.create_label('MyLabelCancel') foldersandabels.click_cancel() time.sleep(1)