from unittest.mock import patch

from processing_accounting.flows import base


@patch('processing_accounting.flows.base.os')
def test_get_domain(mock_os):
    """Test generating domain for dev.
    """

    mock_os.getenv.return_value = 'dev'
    # verify default domain is dev
    expected_result = 'dev_swf_processing_accounting'
    actual_result = base.get_domain()
    assert expected_result == actual_result


@patch('processing_accounting.flows.base.os')
def test_get_domain_prod(mock_os):
    """Test generating domain for prod.
    """

    mock_os.getenv.return_value = 'prod'
    # verify default domain is prod
    expected_result = 'prod_swf_processing_accounting'
    actual_result = base.get_domain()
    assert expected_result == actual_result


@patch('processing_accounting.flows.base.os')
def test_get_domain_qa(mock_os):
    """Test generating domain for qa.
    """

    mock_os.getenv.return_value = 'qa'
    # verify default domain is qa
    expected_result = 'qa_swf_processing_accounting'
    actual_result = base.get_domain()
    assert expected_result == actual_result