import logging
import pytest
import uuid
from fpcapture.context import context


def test_set_correlation_id():
    """Assert that setting/retrieving correlation_id returns expected value
    """

    test_correlation_id = str(uuid.uuid1())
    context.reset()  # always reset
    context.set('correlation_id', test_correlation_id)
    assert(context.get_correlation_id() == test_correlation_id)


def test_missing_correlation_id_exception():
    """Assert that context raises AttributeError exception when
    user attempts to retrieve unset correlation_id
    """

    context.reset()  # always reset
    with pytest.raises(AttributeError):
        context.get_correlation_id()


def test_get_logger():
    """Assert that context returns expected logger
    """

    test_correlation_id = str(uuid.uuid1())
    context.reset()  # always reset
    context.set('correlation_id', test_correlation_id)
    assert(isinstance(context.logger, logging.LoggerAdapter))


def test_reset_context():
    """Assert that context properties are properly reset
    """

    context.reset()  # always reset
    context.set('test_property', 'test_value')
    assert(context.test_property == 'test_value')
    context.reset()
    assert(getattr(context, 'test_property', None) is None)