issues/50577_1094/user_tool.py

import sys, os

base_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, base_path)

import time
import imaplib
import email
import re
import os
from email.header import decode_header
from playwright.sync_api import sync_playwright, expect
from playwright.sync_api._context_manager import PlaywrightContextManager
from utils.trace_cleaner import trace_cleaner


GMAIL_NAME = "nomandevtur+50577"
GMAIL_APP_PASSWORD = ""


def get_test_user_info(seed=None, first_name=None, last_name=None):
    if first_name is None:
        first_name = GMAIL_NAME

    if last_name is None:
        last_name = ""

    email_addr = f"{GMAIL_NAME}+{seed}@gmail.com" if seed is not None else f"{GMAIL_NAME}@gmail.com"
    
    return {"email": email_addr, "password": GMAIL_APP_PASSWORD, "first_name": first_name, "last_name": last_name}


def delete_email_inbox(user_email, password, retries=5, delay=10):
    imap = imaplib.IMAP4_SSL("imap.gmail.com")
    imap.login(user_email, password)

    imap.select("inbox")
    status, all_emails = imap.search(None, "ALL")
    if status == "OK":
        for email_id in all_emails[0].split():
            imap.store(email_id, "+FLAGS", "\Deleted")
        imap.expunge()


def get_magic_code(user_email, password, retries=5, delay=5):

    imap = imaplib.IMAP4_SSL("imap.gmail.com")
    imap.login(user_email, password)

    for _ in range(retries):
        imap.select("inbox")
        status, messages = imap.search(None, '(UNSEEN SUBJECT "Expensify magic sign-in code:")')

        if status == "OK":
            email_ids = messages[0].split()
            print(email_ids)

            if email_ids:
                latest_email_id = email_ids[-1]
                status, msg_data = imap.fetch(latest_email_id, "(RFC822)")

                for response_part in msg_data:
                    if isinstance(response_part, tuple):
                        msg = email.message_from_bytes(response_part[1])
                        subject, encoding = decode_header(msg["Subject"])[0]
                        if isinstance(subject, bytes):
                            subject = subject.decode(encoding or "utf-8")


                        match = re.search(r"Expensify magic sign-in code: (\d+)", subject)
                        if match:
                            code = match.group(1)
                            imap.logout()
                            return code
            else:
                print("No unread emails found with the subject. Retrying...")
        else:
            print("Failed to retrieve emails. Retrying...")


        time.sleep(delay)

    
    imap.logout()
    print("Max retries reached. Email not found.")
    return None


def select_activity(page, first_name, last_name, activity_text):
    expect(page.get_by_text("What do you want to do today?")).to_be_visible()
    

    page.get_by_label(activity_text).click()
    

    page.get_by_role("textbox", name="First name").fill(first_name)
    page.get_by_role("textbox", name="Last name").fill(last_name)
    page.get_by_role("button", name="Continue").last.click()


def login_user(page, user_info, activity_text="Track and budget expenses"):

    page.context.clear_cookies()

    page.goto('https://dev.new.expensify.com:8082/')
    page.wait_for_load_state('load')
    
    try:
        expect(page.get_by_label("Inbox")).to_be_visible(timeout=10000)
        return
    except:
        pass

    page.get_by_test_id("username").fill(user_info["email"])
    page.get_by_role("button", name="Continue").click()

    join_button = page.get_by_role("button", name="Join")
    validate_code_input = page.locator('input[data-testid="validateCode"]')
    expect(join_button.or_(validate_code_input)).to_be_visible()

    if (join_button.is_visible()):
        join_button.click(timeout=3000)
    else:
        magic_code = "123456"
        validate_code_input.fill(magic_code)

    page.wait_for_timeout(3000)

    select_activity_dialog = page.get_by_text("What do you want to do today?")
    if select_activity_dialog.count() > 0:
        select_activity(page, user_info["first_name"], user_info["last_name"], activity_text)


def get_context(p: PlaywrightContextManager, user_info, browser, is_phone_setup=False):
    """
    Creates and returns a browser context and the initial page.
    Supports mobile or desktop context setup.
    """
    permissions = ['clipboard-read', 'clipboard-write']

    data_dir = 'mobile_context' if is_phone_setup else 'desktop_context'
    data_dir += f"_{user_info['email']}"
    os.makedirs(data_dir, exist_ok=True)  

    context_options = {
        "permissions": permissions,
        "storage_state": None  
    }

    if is_phone_setup:
        context_options.update({
            "viewport": {"width": 390, "height": 844},  
            "user_agent": "Mozilla/5.0 (iPhone; CPU iPhone OS 14_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.0 Mobile/15E148 Safari/604.1",
        })

    context = browser.new_context(**context_options)
    page = context.new_page()
    return context, page

    

def test_inconsistent_button_behavior():
    with sync_playwright() as p:
        user_info = get_test_user_info(10)

        delete_email_inbox(user_info["email"], user_info["password"], retries=6, delay=5)

        browser = p.chromium.launch(channel="chrome", headless=True, slow_mo=500, args=[
            "--disable-web-security",
            "--disable-features=IsolateOrigins,site-per-process"
        ])

        context, page = get_context(p, user_info, browser)
        # Start tracing
        context.tracing.start(screenshots=True, snapshots=True, sources=True)

        try:
            login_user(page, user_info)

            page.get_by_label("Start chat (Floating action)").click()
            page.get_by_label("Submit expense").nth(0).click()
            page.get_by_label("Manual").click()
            page.get_by_placeholder("0").fill("100")
            page.locator("#numPadContainerView").get_by_role("button", name="Next").click()

            page.get_by_test_id("selection-list").get_by_label("Nomandevtur+50577's Workspace").click()

            page.get_by_role("menuitem", name="Merchant Required").click()
            page.get_by_role("textbox", name="Merchant").fill("hello")
            page.get_by_role("button", name="Save").click()
            page1 = context.new_page()
            page1.bring_to_front()
            page1.goto(page.url)
            page1.wait_for_load_state('load')
            page.wait_for_timeout(5000)

            page.bring_to_front()
            page.get_by_role("button", name="Submit PKR").click()

            page.wait_for_timeout(1000)
            page1.bring_to_front()
            page1.get_by_role("button", name="Submit PKR").click()
            page.wait_for_timeout(1000)

            text_locator = page1.get_by_text("Please select a participant.")
            expect(text_locator).to_be_visible()

            page.bring_to_front()
            page.get_by_label("Create").click()
            page.get_by_label("Assign task").click()
            page.get_by_role("textbox", name="Title").fill("877")
            page.get_by_role("button", name="Next").click()

            page1.bring_to_front()
            page1.goto(page.url)
            page1.wait_for_load_state('load')

            page.wait_for_timeout(5000)

            page.bring_to_front()
            page.get_by_role("button", name="Confirm task").click()

            page.wait_for_timeout(1000)
            page1.bring_to_front()
            page1.get_by_role("button", name="Confirm task").click()
            page.wait_for_timeout(1000)
            
            text_locator = page1.get_by_text("Please enter a title and select a share destination.")
            expect(text_locator).to_be_visible()

            page.close()
        finally:
            trace_path = "/app/expensify/user_tool/output_browser1.zip"
            context.tracing.stop(path=trace_path)
            trace_cleaner(trace_path)
            browser.close()


if __name__ == '__main__':
    test_inconsistent_button_behavior()