reflex/tests/test_config.py

import os
from typing import Dict

import pytest

import reflex as rx
from reflex import constants
from reflex.config import DBConfig, get_config
from reflex.constants import get_value


@pytest.fixture
def config_no_db_url_values(base_config_values) -> Dict:
    """Create config values with no db_url.

    Args:
        base_config_values: Base config fixture.

    Returns:
        Config values.
    """
    base_config_values.pop("db_url")
    return base_config_values


@pytest.fixture(autouse=True)
def config_empty_db_url_values(base_config_values):
    """Create config values with empty db_url.

    Args:
        base_config_values: Base config values fixture.

    Yields:
        Config values
    """
    base_config_values["db_url"] = None
    yield base_config_values
    os.environ.pop("DB_URL", None)


@pytest.fixture
def config_none_db_url_values(base_config_values):
    """Create config values with None (string) db_url.

    Args:
        base_config_values: Base config values fixture.

    Yields:
        Config values
    """
    base_config_values["db_url"] = "None"
    yield base_config_values
    os.environ.pop("DB_URL")


def test_config_db_url(base_config_values):
    """Test defined db_url is not changed.

    Args:
        base_config_values: base_config_values fixture.
    """
    os.environ.pop("DB_URL")
    config = rx.Config(**base_config_values)
    assert config.db_url == base_config_values["db_url"]


def test_default_db_url(config_no_db_url_values):
    """Test that db_url is assigned the default value if not passed.

    Args:
        config_no_db_url_values: Config values with no db_url defined.
    """
    config = rx.Config(**config_no_db_url_values)
    assert config.db_url == constants.DB_URL


def test_empty_db_url(config_empty_db_url_values):
    """Test that db_url is not automatically assigned if an empty value is defined.

    Args:
        config_empty_db_url_values: Config values with empty db_url.
    """
    config = rx.Config(**config_empty_db_url_values)
    assert config.db_url is None


def test_none_db_url(config_none_db_url_values):
    """Test that db_url is set 'None' (string) assigned if an 'None' (string) value is defined.

    Args:
        config_none_db_url_values: Config values with None (string) db_url.
    """
    config = rx.Config(**config_none_db_url_values)
    assert config.db_url == "None"


def test_db_url_precedence(base_config_values, sqlite_db_config_values):
    """Test that db_url is not overwritten when db_url is defined.

    Args:
        base_config_values: config values that include db_ur.
        sqlite_db_config_values: DB config values.
    """
    db_config = DBConfig(**sqlite_db_config_values)
    base_config_values["db_config"] = db_config
    config = rx.Config(**base_config_values)
    assert config.db_url == base_config_values["db_url"]


def test_db_url_from_db_config(config_no_db_url_values, sqlite_db_config_values):
    """Test db_url generation from db_config.

    Args:
        config_no_db_url_values: Config values with no db_url.
        sqlite_db_config_values: DB config values.
    """
    db_config = DBConfig(**sqlite_db_config_values)
    config_no_db_url_values["db_config"] = db_config
    config = rx.Config(**config_no_db_url_values)
    assert config.db_url == db_config.get_url()


@pytest.mark.parametrize(
    "key, value, expected_value_type_in_config",
    (
        ("TIMEOUT", "1", int),
        ("CORS_ALLOWED_ORIGINS", "[1, 2, 3]", list),
        ("DB_NAME", "dbname", str),
    ),
)
def test_get_value(monkeypatch, key, value, expected_value_type_in_config):
    monkeypatch.setenv(key, value)
    casted_value = get_value(key, type_=expected_value_type_in_config)

    assert isinstance(casted_value, expected_value_type_in_config)


@pytest.mark.parametrize(
    "kwargs, expected",
    [
        ({"app_name": "test_app", "api_url": "http://example.com"}, "/event"),
        ({"app_name": "test_app", "api_url": "http://example.com/api"}, "/api/event"),
        ({"app_name": "test_app", "event_namespace": "/event"}, "/event"),
        ({"app_name": "test_app", "event_namespace": "event"}, "/event"),
        ({"app_name": "test_app", "event_namespace": "event/"}, "/event"),
    ],
)
def test_event_namespace(mocker, kwargs, expected):
    """Test the event namespace.

    Args:
        mocker: The pytest mock object.
        kwargs: The Config kwargs.
        expected: Expected namespace
    """
    conf = rx.Config(**kwargs)
    mocker.patch("reflex.config.get_config", return_value=conf)

    config = get_config()
    assert config.get_event_namespace() == expected