eff127ee20
Signed-off-by: Sergey Chvalyuk <grubberr@gmail.com> --------- Signed-off-by: Sergey Chvalyuk <grubberr@gmail.com> Co-authored-by: Octavia Squidington III <octavia-squidington-iii@users.noreply.github.com>
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
#
|
|
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
|
|
#
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
import responses
|
|
from source_iterable.source import SourceIterable
|
|
from source_iterable.streams import Lists
|
|
|
|
|
|
@responses.activate
|
|
@pytest.mark.parametrize("body, status, expected_streams", ((b"", 401, 7), (b"alpha@gmail.com\nbeta@gmail.com", 200, 44)))
|
|
def test_source_streams(mock_lists_resp, config, body, status, expected_streams):
|
|
responses.add(responses.GET, "https://api.iterable.com/api/lists/getUsers?listId=1", body=body, status=status)
|
|
streams = SourceIterable().streams(config=config)
|
|
assert len(streams) == expected_streams
|
|
|
|
|
|
def test_source_check_connection_ok(config):
|
|
with patch.object(Lists, "read_records", return_value=iter([{"id": 1}])):
|
|
assert SourceIterable().check_connection(MagicMock(), config=config) == (True, None)
|
|
|
|
|
|
def test_source_check_connection_failed(config):
|
|
with patch.object(Lists, "read_records", return_value=iter([])):
|
|
assert SourceIterable().check_connection(MagicMock(), config=config)[0] is False
|