"""Base Accounting Work Flow Class."""

from unittest.mock import patch

from accounting.flows import base


@patch('accounting.flows.base.os')
def test_get_domain(mock_os):
    """Test generating domain for dev."""
    mock_os.getenv.return_value = 'dev'
    mock_os.getenv.return_value = 'dev_test'
    # verify default domain is dev
    expected_result = 'dev_test_swf_accounting'
    actual_result = base.get_domain()
    assert expected_result == actual_result


@patch('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_accounting'
    actual_result = base.get_domain()
    assert expected_result == actual_result


@patch('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_accounting'
    actual_result = base.get_domain()
    assert expected_result == actual_result