"""Environment tests."""

import importlib
from unittest.mock import patch

import pytest

from dim_refresh_etl.util import environment


@patch('dim_refresh_etl.util.environment.os.environ')
def test_dev(mock_os):
    """Test DEV environment."""
    mock_os.get.return_value = 'dev'
    importlib.reload(environment)
    assert environment.name == environment.DEV == 'dev'


@patch('dim_refresh_etl.util.environment.os.environ')
def test_test(mock_os):
    """Test TEST environment."""
    mock_os.get.return_value = 'test'
    importlib.reload(environment)
    assert environment.name == environment.TEST == 'test'


@patch('dim_refresh_etl.util.environment.os.environ')
def test_prod(mock_os):
    """Test PROD environment."""
    mock_os.get.return_value = 'prod'
    importlib.reload(environment)
    assert environment.name == environment.PROD == 'prod'


@patch('dim_refresh_etl.util.environment.os.environ')
def test_invalid(mock_os):
    """Test invalid OS."""
    mock_os.get.return_value = 'blah'
    with pytest.raises(Exception):
        importlib.reload(environment)