Files
packager/python_flask/selenium/test.py
2023-05-06 19:22:50 +02:00

141 lines
4.7 KiB
Python
Executable File

#!/usr/bin/env python3
import time
import argparse
import code
import readline
from helium import *
import selenium
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
parser = argparse.ArgumentParser()
parser.add_argument("--repl", action="store_true")
args = parser.parse_args()
opts = selenium.webdriver.firefox.options.Options()
for has_javascript in [True, False]:
profile = selenium.webdriver.FirefoxProfile()
profile.DEFAULT_PREFERENCES["frozen"]["javascript.enabled"] = has_javascript
opts.profile = profile
driver = selenium.webdriver.Remote(
command_executor="http://localhost:4444/wd/hub", options=opts
)
driver.implicitly_wait(0)
Config.implicit_wait_secs = 1
try:
helium.set_driver(driver)
helium.go_to("http://localhost:5000")
assert driver.title == "Packager"
new_entry = Text("Add new package list")
lists_before = find_all(S("table > tbody > tr", below=Text("Package Lists")))
write("newlist", into=TextField(to_right_of="Name"))
write("newlistdesc", into=TextField(to_right_of="Description"))
click(Button("Add"))
lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
assert len(lists_before) == len(lists_after) - 1
nameidx = next(
i
for i, v in enumerate(find_all(S("table > thead > tr > th")))
if v.web_element.text == "Name"
)
descidx = next(
i
for i, v in enumerate(find_all(S("table > thead > tr > th")))
if v.web_element.text == "Description"
)
write("newlist2", into=TextField(to_right_of="Name"))
write("newlistdesc2", into=TextField(to_right_of="Description"))
click(Button("Add"))
lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
assert len(lists_before) == len(lists_after) - 2
new_entry1 = lists_after[-2]
cells = new_entry1.web_element.find_elements_by_tag_name("td")
assert cells[nameidx].text == "newlist"
assert cells[descidx].text == "newlistdesc"
new_entry2 = lists_after[-1]
cells = new_entry2.web_element.find_elements_by_tag_name("td")
assert cells[nameidx].text == "newlist2"
assert cells[descidx].text == "newlistdesc2"
editbtn = new_entry1.web_element.find_element_by_class_name("mdi-pencil")
click(editbtn)
lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
editrow = find_all(S("table > tbody > tr"))[-2]
inputs = editrow.web_element.find_elements_by_xpath("//input")
write("editedname", into=inputs[0])
write("editeddesc", into=inputs[1])
savebtn = editrow.web_element.find_element_by_class_name("mdi-content-save")
click(savebtn)
lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
new_entry1 = lists_after[-2]
cells = new_entry1.web_element.find_elements_by_tag_name("td")
assert cells[nameidx].text == "editedname"
assert cells[descidx].text == "editeddesc"
editbtn = new_entry1.web_element.find_element_by_class_name("mdi-pencil")
click(editbtn)
lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
editrow = find_all(S("table > tbody > tr"))[-2]
inputs = editrow.web_element.find_elements_by_xpath("//input")
write("editedname_again", into=inputs[0])
write("editeddesc_again", into=inputs[1])
cancelbtn = editrow.web_element.find_element_by_class_name("mdi-cancel")
click(cancelbtn)
lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
new_entry1 = lists_after[-2]
cells = new_entry1.web_element.find_elements_by_tag_name("td")
assert cells[nameidx].text == "editedname"
assert cells[descidx].text == "editeddesc"
lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
new_entry1 = lists_after[-2]
deletebtn = new_entry1.web_element.find_element_by_class_name("mdi-delete")
click(deletebtn)
lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
assert len(lists_before) + 1 == len(lists_after)
new_entry2 = lists_after[-1]
deletebtn = new_entry2.web_element.find_element_by_class_name("mdi-delete")
click(deletebtn)
lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
assert len(lists_before) == len(lists_after)
if args.repl:
code.InteractiveConsole(locals=globals()).interact()
finally:
driver.quit()
time.sleep(5)