from unittest.mock import Mock

from getstream_connector.integrations import flask


CORRELATION_ID = "TEST_CORRELATION_ID"


def test_get_correlation_id(mocker):
    mocker.patch("getstream_connector.integrations.flask.g", Mock(correlation_id=CORRELATION_ID))
    assert flask.get_correlation_id() == CORRELATION_ID


def test_get_correlation_id_no_correlation_id(mocker):
    mocker.patch("getstream_connector.integrations.flask.g", Mock(spec_set=()))
    assert flask.g is not None
    assert flask.get_correlation_id() is None


def test_get_correlation_id_no_g(mocker):
    mocker.patch("getstream_connector.integrations.flask.g", None)
    assert flask.g is None
    assert flask.get_correlation_id() is None