"""Test shared utilities helpers."""

from unittest.mock import patch

from prs import config
from prs.constants import tools
from prs.utils import shared_utils


def test_get_file_code_success():
    """Test get_file_code success."""
    result = shared_utils.get_file_code()
    assert result


def test_get_file_code_failure():
    """Test get_file_code failure."""
    file_code_failure = 'ABc01'
    result = shared_utils.get_file_code()
    assert result != file_code_failure


def test_generate_file_base_name_success(mocker):
    """Test generate_file_base_name success."""
    tool_name = tools.LABEL_DETAILS
    country_name = 'USA'
    file_code = 'ABC01'
    current_date_time = '2017-08-17-04.06.27pm'
    expected_result = '{}/{}/{}-{}-{}'.format(
        config.SERVICE_NAME, tool_name, country_name.replace(' ', '_'),
        file_code, current_date_time)

    mocker.patch.object(
        shared_utils, 'get_current_datetime',
        return_value=current_date_time)

    result = shared_utils.generate_file_base_name(
        tool_name, country_name, file_code)
    assert result == expected_result


@patch('prs.utils.shared_utils.g')
def test_log_info_to_loggly(mock_g):
    """Test log_info_to_loggly success."""
    data = 'test message'
    shared_utils.log_info_to_loggly(data)

    mock_g.ows.log.info.assert_called_with(data)