testcases added for dropdown and table

Author: ravishankar

tests/test_elements.py

@@ -9,7 +9,7 @@ def test_select_box():
 
         select_value = "1"
         obj.fill({"select_dropdown": select_value})
-        element = obj.element("name", "select_dropdown")
+        element = obj.element("select_dropdown", "name")
         for ele in element.find_elements_by_tag_name("option"):
             if ele.text == "One":
                 assert ele.is_selected() is True

tests/test_parser.py

@@ -1,5 +1,5 @@
 from s_tool.driver import SeleniumDriver
-from s_tool.parser import select_options
+from s_tool.parser import get_table, select_options
 
 from . import TEST_URL
 
@@ -10,16 +10,16 @@ def test_select():
         obj.get(TEST_URL)
 
         # Test dropdown options with element id
-        options_dicts = select_options(obj.element("id", "sel1"))
+        options_dicts = select_options(obj.element("sel1"))
         actual_result = {"1": "1", "2": "2", "3": "3", "4": "4"}
         assert options_dicts == actual_result
 
         # Test for invalid element
-        options_dicts = select_options(obj.element("id", "select_id"))
+        options_dicts = select_options(obj.element("select_id", "id"))
         assert options_dicts == {}
 
         # Test for disabled dropdown
-        options_dicts = select_options(obj.element("xpath", '//select[@disabled=""]'))
+        options_dicts = select_options(obj.element('//select[@disabled=""]', "xpath"))
         assert options_dicts == {
             "default": "default",
             "1": "One",
@@ -28,11 +28,19 @@ def test_select():
         }
 
         # Test for swaping keys and values\
-        elem = obj.element("xpath", '//select[@disabled=""]')
+        elem = obj.element('//select[@disabled=""]', "xpath")
         options_dicts = select_options(elem, swap=True)
         assert options_dicts == {
             "default": "default",
             "One": "1",
             "Two": "2",
             "Three": "3",
         }
+
+
+def test_table():
+    with SeleniumDriver("chrome", headless=True) as obj:
+        obj.get(TEST_URL)
+        table = obj.element('//table[@class="table"]', "xpath")
+        table_data = get_table(table)
+        assert len(table_data) == 4