issues/166/user_tool.py (142 lines of code) (raw):
import os
import re, time
from datetime import datetime, timezone
from email.header import decode_header
from PIL import Image
from io import BytesIO
import imaplib
import email
import re
from playwright.sync_api import Page, TimeoutError, expect, sync_playwright
from utils.trace_cleaner import trace_cleaner
USERNAME = "contributer0"
APP_PASSWORD = ""
USER_EMAIL = f"{USERNAME}+51388_1@gmail.com"
NEW_DOT_URL = "https://dev.new.expensify.com:8082/"
def get_magic_code(user_email, password, since=None, retries=5, delay=10):
if since is None:
since = datetime.now(timezone.utc).replace(second=0, microsecond=0)
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()
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)
email_date = email.utils.parsedate_to_datetime(msg["Date"])
if match and email_date >= since:
code = match.group(1)
imap.logout()
return code
print("Email doesn't match conditions. Retrying...")
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 create_user(page: Page, firstname: str = "User", lastname: str = "Test"):
page.get_by_role("button", name="Join").click()
# Update profile
page.get_by_text("Track and budget expenses").click()
page.get_by_role("button", name="Continue").last.click()
page.get_by_role("textbox", name="First name").fill(firstname)
page.get_by_role("textbox", name="Last name").fill(lastname)
try:
page.get_by_role("button", name="Continue").click(timeout=2000)
except TimeoutError:
pass
try:
page.get_by_role("button", name="Get Started").click(timeout=200)
except TimeoutError:
pass
def login(page: Page):
# get current timestamp
magic_code = "123456"
if magic_code is None:
raise ValueError("Failed to retrieve magic code")
page.get_by_role("textbox").fill(magic_code)
def login_or_create_user(
page: Page, user_email: str = USER_EMAIL, lastname: str = "Test"
):
page.get_by_role("textbox", name="Phone or email").fill(user_email)
page.get_by_role("button", name="Continue").click()
try:
page.get_by_role("button", name="Join").wait_for(state="visible", timeout=2000)
create_user(page, lastname)
except TimeoutError:
login(page)
page.get_by_label("Inbox").first.wait_for(state="visible")
def create_sample_jpg():
# Generates an image in memory and returns the bytes
image = Image.new("RGB", (100, 100), color="blue")
img_byte_arr = BytesIO()
image.save(img_byte_arr, format="JPEG")
img_byte_arr.seek(0)
return img_byte_arr.getvalue()
def test_51388():
with sync_playwright() as p:
# Launch chromium and open new page
browser = p.chromium.launch(
channel="chrome",
headless=False,
args=[
"--ignore-certificate-errors",
"--disable-web-security",
"--disable-features=IsolateOrigins,site-per-process"
],
proxy={"server": "http://localhost:8080"},
slow_mo=500
)
# mobile_device = p.devices['Pixel 7']
mobile_device = p.devices['iPhone 12 Pro']
context = browser.new_context(**mobile_device)
# Start tracing
context.tracing.start(screenshots=True, snapshots=True, sources=True)
page = context.new_page()
try:
# Login Expensify
page.goto(NEW_DOT_URL)
login_or_create_user(page)
# go to Settings
page.get_by_label("My settings").click()
# enable debug mode
page.get_by_label("Troubleshoot").click()
page.get_by_label("Debug mode").click()
page.get_by_label("Back").click()
page.wait_for_timeout(1000)
# create a new workspace
page.get_by_test_id("InitialSettingsPage").get_by_label("Workspaces").click()
page.get_by_label("New workspace").first.click()
page.get_by_test_id("WorkspaceInitialPage").get_by_label("Profile").click()
page.get_by_label("Test User's Workspace").last.click()
workspace_name = f"workspace{3456}"
page.get_by_role("textbox", name="Name").fill(workspace_name)
page.get_by_role("button", name="Save").click()
page.wait_for_timeout(1000)
page.get_by_test_id("WorkspacePageWithSections").get_by_label("Back").click()
page.get_by_test_id("WorkspaceInitialPage").get_by_label("Back").click()
page.get_by_test_id("WorkspacesListPage").get_by_label("Back").click()
page.get_by_label("Inbox").click()
page.get_by_label("Navigates to a chat").get_by_text(workspace_name).last.click()
# send a message
page.locator("#composer").fill("hello")
page.get_by_label("Send").click()
# Tap header
try:
page.get_by_label(workspace_name).click(timeout=5000)
except:
page.get_by_label("Test User").click(timeout=5000)
# press debug
page.get_by_label("Debug").click()
# tap delete
page.get_by_role("button", name="Delete").click()
page.get_by_label("Back").last.click()
# verify that not found detail page is visible and no blank page displayed
expect(page.get_by_text("Hmm... it's not here")).to_be_visible()
page.reload()
page.wait_for_timeout(1000)
expect(page.get_by_text("Hmm... it's not here")).to_be_visible()
page.wait_for_timeout(1000)
finally:
# Stop tracing, export the trace, clean it, and close the browser
context.tracing.stop(path="/app/expensify/user_tool/output_browser1.zip")
trace_cleaner("/app/expensify/user_tool/output_browser1.zip")
context.close()
browser.close()