doczyai-pipelines/Doczy.AI_Automation/tests/test_interface_0.py

import random
import time

from selenium.common import InvalidArgumentException

from pages.interface_0_page_objects import Interface_0
from utils.element_related_methods import get_title
from utils.common_methods import sign_in
from utils.logger import get_logger


class Test_interface_0:

    logger = get_logger()

    def test_validate_navigation(self, driver, config):
        self.logger.info("Verifying designated Interface 0 Page opens successfully")
        driver.get(config['url_interface_0'])
        get_title(driver, "interface_0 · Streamlit")
        if driver.title == 'interface_0 · Streamlit':
            self.logger.info("Designated Interface 0 Page opened successfully")
            assert True
        else:
            self.logger.info("Designated Interface 0 Page not opened successfully")
            assert False

    def test_verify_sign_in(self, driver, config):
        self.logger.info("Verifying Single Sign-On (SSO) Login for Interface 0")
        driver.get(config['url_interface_0'])
        sign_in(driver, config)
        self.logger.info("Signed in successful")
        get_title(driver, "interface_0 · Streamlit")
        if driver.title == 'interface_0 · Streamlit':
            self.logger.info("Interface 0 page opened successfully")
            assert True
        else:
            self.logger.info("Interface 0 page not opened successfully")
            assert False

    # def test_verify_client_name_dropdown(self, driver, config):
    #     self.logger.info(f"Opening URL {config['url_interface_0']}")
    #     driver.get(config['url_interface_0'])
    #     self.logger.info(f"URL {config['url_interface_0']} opened")
    #     sign_in(driver, config)
    #     interface_0_page = Interface_0(driver)
    #     interface_0_page.click_client_name_dropdown()
    #     # get_title(driver, "interface_0 · Streamlit")
    #     if driver.title == 'interface_0 · Streamlit':
    #         assert True
    #     else:
    #         assert False

    def test_verify_batch_creation(self, driver, config):
        self.logger.info("Verifying S3 bucket is created with batch ID")
        driver.get(config['url_interface_0'])
        sign_in(driver, config)
        self.logger.info("Signed in successful")
        interface_0_page = Interface_0(driver)
        interface_0_page.select_client()
        interface_0_page.upload_files(config)
        interface_0_page.create_batch()
        self.logger.info(f"Created batch with Batch ID: {interface_0_page.p_element.text}")
        assert interface_0_page.p_element.text.startswith("batch_")

    def test_verify_no_files_uploaded(self, driver, config):
        self.logger.info("Verifying S3 bucket is created with batch ID")
        driver.get(config['url_interface_0'])
        sign_in(driver, config)
        self.logger.info("Signed in successful")
        interface_0_page = Interface_0(driver)
        interface_0_page.select_client()
        interface_0_page.create_batch()
        # self.logger.info(f"Created batch with Batch ID: {p_element.text}")
        # try:
        assert interface_0_page.p_element.text is "Select files to create batch", "Batch ID is created instead of message"
        # except AssertionError:
        #     self.logger.info(f"Batch ID is created instead of message")
        #     assert False


    # def test_multiple_files(self, driver, config):
    #     contract_files = config["contract_files"]
    #     # random.seed(5)
    #     random_count = random.randint(1, len(contract_files))  # Choose a random count between 1 and the length of the array
    #
    #     # Get a random selection of elements from the array
    #     random_elements = random.sample(contract_files, random_count)
    #
    #     # Print the random count and the selected elements
    #     print("Random count:", random_count)
    #     print("Random elements:", random_elements)
    #     contracts = '\nC:\\Ankit\\Marked Up PDFs\\test\\'.join(random_elements)
    #     print(contracts)
    #     # for file in contract_files:
    #     #     print(file)
    #     # print(contract_files)