Function process_test_file

seleniumbase/console_scripts/sb_objectify.py:222–3030

(
    code_lines, selector_dict=None, object_dict=None, add_comments=False
)

def process_test_file(
    code_lines, selector_dict=None, object_dict=None, add_comments=False
):

    seleniumbase_lines = []
    page_selectors = []
    changed = []  # The classes of page_objects.py to add to the test import

    for line in code_lines:
        line = line.rstrip()

        # Keep lines with "%s" in them as they were
        if r"%s" in line:
            seleniumbase_lines.append(line)
            continue

        # Handle self.drag_and_drop(SELECTOR, SELECTOR)
        if not object_dict:
            data = re.match(
                r"""^(\s*)self\.drag_and_drop"""
                r"""\((r?['"][\S\s]+['"]),\s?([\S\s]+)\)([\S\s]*)$""",
                line,
            )
        else:
            data = re.match(
                r"""^(\s*)self\.drag_and_drop"""
                r"""\(([\S]+),\s?([\S]+)\)([\S\s]*)$""",
                line,
            )
        if data:
            whitespace = data.group(1)
            selector1 = "%s" % data.group(2)
            selector1 = remove_extra_slashes(selector1)
            page_selectors.append(selector1)
            selector2 = "%s" % data.group(3)
            selector2 = remove_extra_slashes(selector2)
            page_selectors.append(selector2)
            comments = data.group(4)
            command = """%sself.drag_and_drop(%s, %s)%s""" % (
                whitespace,
                selector1,
                selector2,
                comments,
            )
            if selector_dict:
                if add_comments:
                    comments = "  # %s, %s" % (selector1, selector2)
                selector1 = optimize_selector(selector1)
                selector2 = optimize_selector(selector2)
                if selector1 in selector_dict.keys() and (
                    selector2 in selector_dict.keys()
                ):
                    selector_object1 = selector_dict[selector1]
                    selector_object2 = selector_dict[selector2]
                    changed.append(selector_object1.split(".")[0])
                    changed.append(selector_object2.split(".")[0])
                    command = """%sself.drag_and_drop(%s, %s)%s""" % (
                        whitespace,