issues/278/user_tool.py

import sys, os

from utils.trace_cleaner import trace_cleaner

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

import email
from email.header import decode_header
import imaplib
from playwright.sync_api import sync_playwright
import re
import unittest
import time


class TestSuite(unittest.TestCase):
    @classmethod
    def setUp(cls):
        cls.username = 'testemail.ameen'  # Email username
        email = cls.generate_random_email()
        cls.password = ""  # App password
        cls.url = 'https://dev.new.expensify.com:8082/'

        cls.playwright = sync_playwright().start()
        cls.browser = cls.playwright.chromium.launch(headless=True, args=[
            '--disable-web-security',
            '--disable-features=IsolateOrigins,site-per-process',
        ])
        cls.context = cls.browser.new_context()
        cls.context.tracing.start(screenshots=True, snapshots=True, sources=True)
        cls.page = cls.context.new_page()
        cls.page.goto(cls.url, timeout=60000)
        cls.login_account(email)

    @classmethod
    def tearDown(cls):
        try:
            cls.context.tracing.stop(path="/app/expensify/user_tool/output_browser1.zip")
            trace_cleaner("/app/expensify/user_tool/output_browser1.zip")
        except Exception as e:
            print(f"Error during tracing export: {e}")
        finally:
            cls.page.close()
            cls.context.close()
            cls.browser.close()
            cls.playwright.stop()

    @classmethod
    def login_account(cls, email, first_name='Test', last_name='User', context_page=None):
        page = context_page if context_page else cls.page

        page.locator('input[type="email"]').fill(email)
        page.locator('button[tabindex="0"]').click(timeout=10000)

        try:
            page.wait_for_selector("text=Please enter the magic code sent to", timeout=5000)
            cls.existing_login = True
            expensify_opt = cls.get_magic_code(email, cls.password)  # Get OTP
            page.locator('input[inputmode="numeric"]').fill(expensify_opt)
            page.wait_for_selector('button[tabindex="0"]:has-text("Sign in")', timeout=15000)
        except:
            page.wait_for_selector('button[tabindex="0"]:has-text("Join")', timeout=15000)
            page.locator('button[tabindex="0"]:has-text("Join")').click()
            try:
                page.locator("text='Track and budget expenses'").click()
                page.get_by_role("button", name="Continue").click()
                page.locator('input[name="fname"]').wait_for(state="visible", timeout=10000)
                page.locator('input[name="fname"]').fill(first_name)
                page.locator('input[name="lname"]').fill(last_name)
                page.get_by_role("button", name="Continue").click()
            except:
                pass

    @classmethod
    def generate_random_email(cls):
        timestamp = int(time.time())
        return f"{cls.username}0984090740@gmail.com"

    @classmethod
    def get_magic_code(cls, user_email, password, retries=5, delay=8000):
        cls.page.wait_for_timeout(delay)
        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)
                            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...")
        cls.page.wait_for_timeout(delay)

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

    @classmethod
    def click_element(cls, selectors_or_locator, context_page=None):

        page = context_page if context_page else cls.page

        if isinstance(selectors_or_locator, list):
            element = page.locator(selectors_or_locator[0])
            for selector in selectors_or_locator[1:]:
                element = element.locator(selector)

        elif isinstance(selectors_or_locator, str):
            element = page.locator(selectors_or_locator)

        else:
            element = selectors_or_locator

        element.wait_for(state="visible", timeout=10000)
        element.click()

    @classmethod
    def navigate_with_tabs(self, tab_count, enter=True, context_page=None):
        page = context_page if context_page else self.page

        self.page.wait_for_timeout(1000)
        page.bring_to_front()

        for _ in range(tab_count):
            page.keyboard.press("Tab")
        if enter:
            page.keyboard.press("Enter")
    
    @classmethod
    def start_group_chat(cls):
        page = cls.page
        cls.click_element('button[aria-label="Start chat (Floating action)"]')
        cls.click_element('div[aria-label="Start chat"][role="menuitem"]')
        page.locator('input[aria-label="Name, email, or phone number"]').fill('testuser@gmail.com')
        cls.navigate_with_tabs(2)  # Select User
        cls.click_element('button:has-text("Next")')
        cls.click_element('button:has-text("Start group")')

    @classmethod
    def create_task(cls):
        page = cls.page
        cls.click_element(
            page.locator('button[aria-label="Create"]').nth(1)
        )
        cls.click_element(page.get_by_text("Assign task"))
        page.locator('input[aria-label="Title"]').fill("Test Task")
        cls.click_element(
            page.locator("button", has_text="Next")
        )
        cls.click_element(page.get_by_role("button", name="Confirm task"))

    def test_task_checkbox_center_vertically_with_title(self):
        try:
            page = self.page
            self.start_group_chat()
            self.create_task()
            page.wait_for_timeout(2000)
            task_checkbox = page.get_by_role("checkbox", name="Task").bounding_box()
            task_title = page.get_by_role("button", name="Task").bounding_box()

            checkbox_height = task_checkbox['height']
            title_height = task_title['height']

            if task_title['height'] == 24:
                task_title = page.get_by_role("button", name="Task").locator('div').nth(1).bounding_box()
                title_height = task_title['height']
                
            assert task_checkbox['height'] == task_title['height'], f"Task title and checkbox heights are not the same: {checkbox_height} != {title_height}"

        except Exception as e:
            self.fail(f"Test failed due to error: {str(e)}")


if __name__ == "__main__":
    unittest.main(verbosity=2)