Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 7 additions & 4 deletions picpaste/webdriver/modules/pages/upload.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from saunter.po.webdriver.page import Page
from selenium.webdriver.support.wait import WebDriverWait
from saunter.po.webdriver.select import Select2
from saunter.po.webdriver.checkbox import CheckBox
import os.path
from pages.preview import Preview

Expand All @@ -9,7 +10,7 @@
'button': 'css=input[type="submit"]',
'storetime': 'css=select[name="storetime"]',
'obscure_filename': 'css=select[name="addprivacy"]',
'accept_rules': 'css=select[name="rules"]',
'accept_rules': 'css=input[name="rules"]',
}

class StoreTime(Select2):
Expand All @@ -22,7 +23,7 @@ def __init__(self, driver):
self.locator = locators["obscure_filename"]
self.driver = driver

class AcceptRules(Select2):
class AcceptRules(CheckBox):
def __init__(self, driver):
self.locator = locators["accept_rules"]
self.driver = driver
Expand Down Expand Up @@ -51,10 +52,12 @@ def upload(self, image, storetime="30 Minutes", obscure_filename="basic", accept

self.storetime.selected = "text=%s" % storetime
self.obscure_filename.selected = "text=%s" % obscure_filename
self.accept_rules.selected = "text=%s" % accept_rules
if accept_rules == "Yes":
checkbox = self.driver.find_element_by_locator(locators['accept_rules'])
checkbox.click()

button = self.driver.find_element_by_locator(locators['button'])
button.click()

p = Preview(self.driver).wait_until_loaded()
return p
return p