more selenium tests

python_flask/selenium/test.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python3
 
 import time
+import argparse
+import code
+import readline
 
 from helium import *
 import selenium
@@ -8,59 +11,130 @@ 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()
 
-profile = selenium.webdriver.FirefoxProfile()
-profile.set_preference("javascript.enabled", "false")
-profile.DEFAULT_PREFERENCES['frozen']['javascript.enabled'] = False
-opts.profile = profile
+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
+    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")
+    try:
+        helium.set_driver(driver)
+        helium.go_to("http://localhost:5000")
 
-    assert driver.title == "Packager"
+        assert driver.title == "Packager"
 
-    new_entry = Text("Add new package list")
+        new_entry = Text("Add new package list")
 
-    lists_before = find_all(S("table > tbody > tr", below=Text("Package Lists")))
+        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"))
+        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")))
+        lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
 
-    assert len(lists_before) == len(lists_after) - 1
+        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")
+        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"
+        )
 
-    new_entry = lists_after[-1]
+        write("newlist2", into=TextField(to_right_of="Name"))
+        write("newlistdesc2", into=TextField(to_right_of="Description"))
+        click(Button("Add"))
 
-    cells = new_entry.web_element.find_elements_by_tag_name("td")
+        lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
+        assert len(lists_before) == len(lists_after) - 2
 
-    assert cells[nameidx].text == "newlist"
-    assert cells[descidx].text == "newlistdesc"
+        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"
 
-    lists_before = lists_after
+        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"
 
-    deletebtn = new_entry.web_element.find_element_by_class_name("mdi-delete")
+        editbtn = new_entry1.web_element.find_element_by_class_name("mdi-pencil")
+        click(editbtn)
 
-    click(deletebtn)
+        lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
+        editrow = find_all(S("table > tbody > tr"))[-2]
 
-    lists_after = find_all(S("table > tbody > tr", below=Text("Package Lists")))
+        inputs = editrow.web_element.find_elements_by_xpath("//input")
+        write("editedname", into=inputs[0])
+        write("editeddesc", into=inputs[1])
 
-    assert len(lists_before) - 1 == len(lists_after)
+        savebtn = editrow.web_element.find_element_by_class_name("mdi-content-save")
+        click(savebtn)
 
-    import code; code.interact(local=locals())
-    time.sleep(5)
-finally:
-    driver.quit()
+        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)