issues/300/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 email
import imaplib
import time

from playwright._impl._page import Page
from playwright.sync_api import TimeoutError, expect, sync_playwright
from utils.trace_cleaner import trace_cleaner


EMAIL_USERNAME = "expensifytesting1@gmail.com"
EMAIL_PASSWORD = ""

EMAIL_USERNAME_POSTFIX = "40464.7"


FIRST_NAME = "Alice"
LAST_NAME = "Aliceoglu"


def generate_user_email(user_id=None):
    """
    Generate an email address for a user login.
    """
    temp = EMAIL_USERNAME.rsplit("@", 1)
    username = temp[0].strip()
    domain = temp[1].strip()
    return f"{username}+{EMAIL_USERNAME_POSTFIX}{user_id or ''}@{domain}"


def clear_inbox(username, password):
    """
    Delete all the messages from the Inbox.
    """

    return
    with imaplib.IMAP4_SSL(host="imap.gmail.com") as imap:
        imap.login(username, password)
        imap.select("inbox")
        imap.store("1:*", "+FLAGS", "\\Deleted")
        imap.expunge()
        imap.close()


def get_otp_from_email(username, password, retries=12, delay=5):
    """
    Read the OTP email and return the OTP code.
    """

    return "123456"
    with imaplib.IMAP4_SSL(host="imap.gmail.com") as imap:
        imap.login(username, password)
        for i in range(1, retries + 1):
            imap.select("inbox")
            status, messages = imap.search(None, "ALL")
            if status == "OK":
                for message_id in reversed(messages[0].split()):
                    status, data = imap.fetch(message_id, "(RFC822)")
                    if status == "OK":
                        email_message = email.message_from_bytes(data[0][1])
                        subject, encoding = email.header.decode_header(email_message["Subject"])[0]
                        if isinstance(subject, bytes):
                            subject = subject.decode(encoding)
                        if subject.startswith("Expensify magic sign-in code:"):
                            otp_code = subject.split(":")[-1].strip()
                            return otp_code
            time.sleep(delay)
        imap.close()
    raise AssertionError("Failed to read the OTP from the email")


def login_user(page: Page) -> None:

    clear_inbox(EMAIL_USERNAME, EMAIL_PASSWORD)

    user_email = generate_user_email()
    page.get_by_test_id("username").fill(user_email)
    page.get_by_role("button", name="Continue").click()

    try:
        expect(page.get_by_test_id("SignInPage").get_by_test_id("validateCode")).to_be_visible(timeout=5000)
    except (AssertionError, TimeoutError):

        page.get_by_test_id("SignInPage").get_by_role("button", name="Join").click()
    else:

        otp_code = get_otp_from_email(EMAIL_USERNAME, EMAIL_PASSWORD)
        page.get_by_test_id("SignInPage").get_by_test_id("validateCode").fill(otp_code)
        try:
            page.get_by_test_id("SignInPage").get_by_role("button", name="Sign in").click(timeout=2000)
        except (AssertionError, TimeoutError):
            pass


def enter_user_information(page: Page) -> None:

    try:
        expect(page.get_by_text("What do you want to do today?")).to_be_visible(timeout=5000)
    except (AssertionError, TimeoutError):
        pass
    else:

        page.get_by_label("Track and budget expenses").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").click()

    try:
        page.get_by_role("button", name="Close").click(timeout=3000)
    except (AssertionError, TimeoutError):
        pass

    expect(page.get_by_test_id("BaseSidebarScreen")).to_be_visible(timeout=7000)


def reproduce_scenario(page: Page) -> None:

    page.get_by_test_id("CustomBottomTabNavigator").get_by_role("button", name="My settings").click()
    page.get_by_test_id("InitialSettingsPage").get_by_role("menuitem", name="Workspaces").click()
    existing_workspaces = [
        item.split("\n")[0].strip()
        for item in page.get_by_test_id("WorkspacesListPage").get_by_label("row", exact=True).all_inner_texts()
    ]
    workspace_name = "WS Test"
    if workspace_name not in existing_workspaces:
        page.get_by_test_id("WorkspacesListPage").get_by_role("button", name="New workspace").first.click()
        page.wait_for_timeout(1000)
        page.get_by_test_id("WorkspacePageWithSections").get_by_text("Workspace", exact=False).first.click()
        page.get_by_test_id("WorkspaceNamePage").get_by_role("textbox", name="Name").fill(workspace_name)
        page.get_by_test_id("WorkspaceNamePage").get_by_role("button", name="Save").click()
        page.wait_for_timeout(1000)
        page.get_by_test_id("WorkspaceInitialPage").get_by_role("button", name="Back").click()

    page.get_by_label("Inbox").click()
    page.wait_for_timeout(1000)

    page.get_by_test_id("BaseSidebarScreen").get_by_text(workspace_name, exact=True).last.click()

    page.get_by_role("button", name="Create").click()

    page.get_by_role("menuitem", name="Track expense").click()

    page.get_by_label("Manual").click()

    page.get_by_placeholder("0").fill("1000")
    page.get_by_role("button", name="Next").first.click()

    page.get_by_text("Merchant").click()
    page.get_by_role("textbox", name="Merchant").fill("hello")
    page.get_by_role("button", name="Save").click()

    page.get_by_role("button", name="Track expense").click()
    page.get_by_label("View details").last.click()

    page.get_by_label("Expense Report #").last.click()


    expect(page.get_by_test_id("ReportDetailsPage").get_by_text("Expense Report #")).to_be_visible()
    expect(page.locator("text=Hmm... it's not here")).not_to_be_visible()


def test_check_expense_title() -> None:
    with sync_playwright() as playwright:
        browser = playwright.chromium.launch(
            headless=True,
            args=["--disable-web-security", "--disable-features=IsolateOrigins,site-per-process"],
        )
        context = browser.new_context(ignore_https_errors=True)
        context.tracing.start(screenshots=True, snapshots=True, sources=True)
        try:
            page = context.new_page()
            page.goto("https://127.0.0.1:8082/")


            login_user(page)


            enter_user_information(page)


            reproduce_scenario(page)

            page.wait_for_timeout(1000)
        except Exception as e:
            print(f"Test encountered an error: {e}")
            raise
        finally:
            context.tracing.stop(path="/app/expensify/user_tool/output_browser1.zip")
            trace_cleaner("/app/expensify/user_tool/output_browser1.zip")
            context.close()
            browser.close()