import pytest
from selenium.webdriver.support.ui import WebDriverWait

from tests.config_reader import get_setup_configs
from tests.helpers import login, logout, save_session_backup
from tests.setup.sessions.test_session import create_session
from tests.setup.tokens.test_trading_token import run_token_setup, get_product_codes
from tests.step_actions import StepActions


@pytest.mark.session
@pytest.mark.trading_token
@pytest.mark.parametrize("config", get_setup_configs("combined"), ids=lambda c: c["product_name"])
def test_session_and_token_setup(driver, test_context, config):

    token = config["token"]

    wait = WebDriverWait(driver, 20)
    step = StepActions(driver, wait, test_context)

    driver.get(config["url"])
    step.capture("open_application")

    login(
        driver,
        config["login"]["username"],
        config["login"]["password"]
    )
    step.capture("login")
    save_session_backup(driver, test_context, "login_session_backup")

    create_session(driver, step, config)

    for product_code in get_product_codes(token):
        product_token = dict(token)
        product_token["product_code"] = product_code
        run_token_setup(driver, step, product_token)

    logout(driver)