import pytest
from snowflake.snowpark.session import Session

from configuration.local_connection import get_connection_parameters


def pytest_addoption(parser):
    parser.addoption("--snowflake-session", action="store", default="live")


@pytest.fixture(scope="module")
def session(request) -> Session:
    """
    Pytest fixture to provide either live or dummy Snowflake connection for testing
    :param request:
    :return: Snowflake session builder config
    """
    if request.config.getoption("--snowflake-session") == "local":
        return Session.builder.config("local_testing", True).create()
    else:
        connection_params = get_connection_parameters("streamlit_backoffice_connection")
        return Session.builder.configs(connection_params).create()