"""Tests for project-specific SQS queue utility functions."""

from flexmock import flexmock
import pytest

from availability import config
from availability.connectors import sqs
from availability.logic.queue import utils
from tests.utils import mock_availability_queue


def test_get_availability_queue_error():
    """Test not receiving a queue rises an exception."""
    (flexmock(sqs).should_receive('get_queue').and_return(None))
    with pytest.raises(ValueError):
        utils.get_availability_queue()


def test_get_availability_queue_success():
    """Test not receiving a queue rises an exception."""
    with mock_availability_queue() as queue:
        assert queue.attributes.get('VisibilityTimeout') == (
            str(config.SQS_MESSAGE_VISIBILITY_TIMEOUT_SECONDS))