test_source.py

#
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
#

from unittest.mock import MagicMock

import logging
from source_sentry.source import SourceSentry


def test_source_wrong_credentials(requests_mock):
    source = SourceSentry()
    status, error = source.check_connection(logger=logging.getLogger("airbyte"), config={"auth_token": "test_auth_token"})
    assert not status


def test_check_connection(requests_mock):
    source = SourceSentry()
    logger_mock = MagicMock()
    requests_mock.get(url="https://sentry.io/api/0/projects/test-org/test-project/", json={"id": "id", "name": "test-project"})
    config = {"auth_token": "token", "organization": "test-org", "project": "test-project", "hostname": "sentry.io"}
    assert source.check_connection(logger_mock, config) == (True, None)


def test_streams(mocker):
    source = SourceSentry()
    config_mock = MagicMock()
    config_mock["auth_token"] = "test-token"
    config_mock["organization"] = "test-organization"
    config_mock["project"] = "test-project"
    streams = source.streams(config_mock)
    expected_streams_number = 5
    assert len(streams) == expected_streams_number