import sys, os base_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, base_path) from playwright.sync_api import sync_playwright, expect from playwright.sync_api._context_manager import PlaywrightContextManager from playwright._impl._browser import Browser from playwright._impl._page import Page import re, time import imaplib import email from utils.trace_cleaner import trace_cleaner EMAIL = "mancity9271+1005@gmail.com" APP_PASSWORD = "" NEWDOT_URL = "https://dev.new.expensify.com:8082/" class IMAPOTPExtractor: def __init__(self, email_address, password): if not self._validate_email(email_address): raise ValueError("Invalid email address format.") self.original_email = email_address self.password = password def _validate_email(self, email_address): return bool(re.match(r"[^@]+@[^@]+\.[^@]+", email_address)) def set_new_email(self, email): self.new_email = email def check_for_otp(self): otp = "123456" return otp try: mail = imaplib.IMAP4_SSL("imap.gmail.com") mail.login(self.original_email, self.password) mail.select("inbox") status, messages = mail.search(None, "UNSEEN") if status != "OK": return None email_ids = messages[0].split()[::-1] for email_id in email_ids: status, msg_data = mail.fetch(email_id, '(RFC822)') if status != "OK": print("Failed to fetch email ID:", email_id) continue for response_part in msg_data: if isinstance(response_part, tuple): msg = email.message_from_bytes(response_part[1]) if self.new_email in msg["To"]: otp_code = self._extract_otp_from_email(msg) if otp_code: mail.store(email_id, '+FLAGS', '\\Deleted') mail.expunge() mail.close() mail.logout() return otp_code mail.close() mail.logout() return None except imaplib.IMAP4.error as e: print(f"IMAP error: {e}") return None def _extract_otp_from_email(self, msg): email_body = None if msg.is_multipart(): for part in msg.walk(): if part.get_content_type() == "text/plain": email_body = part.get_payload(decode=True).decode() break else: email_body = msg.get_payload(decode=True).decode() if email_body: otp_code = self._extract_otp(email_body) return otp_code return None def _extract_otp(self, text): match = re.search(r"\b\d{6}\b", text) return match.group(0) if match else None class UserLogin(): gmail_account_helper = IMAPOTPExtractor(EMAIL, APP_PASSWORD) def new_dot_login(self,page, email): page.goto(NEWDOT_URL) page.locator('input[type="email"]').fill(email) page.wait_for_timeout(2000) page.get_by_role("button", name="Continue").nth(0).click() page.wait_for_timeout(10000) self.gmail_account_helper.set_new_email(email) otp = self.gmail_account_helper.check_for_otp() if otp: page.locator('input[data-testid="validateCode"]').fill(otp) else: print("OTP not received.") def test_invoice_report_amount(): with sync_playwright() as p: browser = p.chromium.launch(headless=True, args=[ "--ignore-certificate-errors", "--disable-web-security", "--disable-features=IsolateOrigins,site-per-process" ]) context = browser.new_context() context.tracing.start(screenshots=True, snapshots=True, sources=True) page = context.new_page() try: user = UserLogin() user.new_dot_login(page, EMAIL) page.wait_for_timeout(2000) page.get_by_label("My settings").click() page.locator("div:nth-child(2) > div:nth-child(2) > div > div > div > div > div > div > div > div:nth-child(2) > .css-view-175oi2r").click() page.wait_for_timeout(1000) page.get_by_label("New workspace").first.click() page.locator("div:nth-child(2) > div:nth-child(5) > div > div > div > div > div > div > div > div > div:nth-child(2) > div").click() page.get_by_test_id("selection-list-text-input").click() page.get_by_test_id("selection-list-text-input").press("CapsLock") page.wait_for_timeout(1000) page.get_by_test_id("selection-list-text-input").fill("K") page.get_by_test_id("selection-list-text-input").press("CapsLock") page.wait_for_timeout(1000) page.get_by_test_id("selection-list-text-input").fill("Kes") page.get_by_test_id("selection-list").get_by_label("KES - KSh").click() page.get_by_text("Members").click() page.wait_for_timeout(1000) page.get_by_role("button", name="Invite member").click() page.get_by_test_id("selection-list-text-input").fill("test@gmail.com") page.wait_for_timeout(2000) page.locator("[id=\"\\31 78516\"]").get_by_label("Joanna Cheng").click() page.get_by_role("button", name="Next").click() page.wait_for_timeout(1000) page.get_by_test_id("WorkspaceInviteMessagePage").get_by_role("button", name="Invite").click() page.get_by_text("More features").click() page.get_by_label("Send and receive invoices.").click() page.wait_for_timeout(1000) page.get_by_label("Back").click() page.get_by_label("Inbox").click() page.get_by_label("Start chat (Floating action)").click() page.wait_for_timeout(1000) page.get_by_text("Send invoice").click() page.get_by_label("Select a currency").click() page.get_by_test_id("selection-list-text-input").press("CapsLock") page.wait_for_timeout(1000) page.get_by_test_id("selection-list-text-input").fill("K") page.get_by_test_id("selection-list-text-input").press("CapsLock") page.wait_for_timeout(1000) page.get_by_test_id("selection-list-text-input").fill("Kes ") page.get_by_label("KES - KSh").click() page.get_by_placeholder("0").fill("100") page.get_by_role("button", name="Next").click() page.wait_for_timeout(1000) page.get_by_test_id("selection-list-text-input").fill("test@gmail.com") page.wait_for_timeout(2000) page.get_by_test_id("selection-list").get_by_label("Joanna Cheng").click() page.get_by_role("button", name="Next").click() page.get_by_role("textbox", name="Your company name").press("CapsLock") page.wait_for_timeout(1000) page.get_by_role("textbox", name="Your company name").fill("JTL") page.get_by_role("textbox", name="Your company website").click() page.get_by_role("textbox", name="Your company website").fill("https://qa-trainer-jschool.web.app/") page.wait_for_timeout(2000) page.get_by_role("button", name="Send KES 100.00 invoice").click() page.wait_for_timeout(2000) page.get_by_label("View details").click() page.locator("div:nth-child(2) > div > div:nth-child(3) > div > div > div > div > div > div > div > div > div > div > .css-view-175oi2r").click() page.get_by_label("Select a currency").click() page.get_by_test_id("selection-list-text-input").fill("USD") page.wait_for_timeout(2000) page.get_by_label("USD - $").click() page.wait_for_timeout(2000) page.get_by_role("button", name="Save").click() page.wait_for_timeout(3000) expect(page.get_by_text("TotalKES 0.00Out-of-pocket")).not_to_be_visible() except Exception as e: print(f"An error occurred during the test: {e}") raise e finally: context.tracing.stop(path="/app/expensify/user_tool/output_browser1.zip") trace_cleaner("/app/expensify/user_tool/output_browser1.zip") browser.close()