bdc41c59dc
* add invoices to Airbyte connector * Add unit tests * Remove logging statements * Update documentation * Update documentation * Fix integration tests * Address PR comments * Bump docker image tag * Update changelog * Update * auto-bump connector version --------- Co-authored-by: Kshitij Grover <k.grover123@gmail.com> Co-authored-by: Octavia Squidington III <octavia-squidington-iii@users.noreply.github.com>
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
#
|
|
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
|
|
#
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
import responses
|
|
from requests.exceptions import HTTPError
|
|
from source_orb.source import SourceOrb
|
|
|
|
|
|
@responses.activate
|
|
def test_check_connection_success(mocker):
|
|
responses.add(
|
|
responses.GET,
|
|
"https://api.billwithorb.com/v1/ping",
|
|
)
|
|
source = SourceOrb()
|
|
logger_mock = MagicMock()
|
|
assert source.check_connection(logger_mock, MagicMock()) == (True, None)
|
|
|
|
|
|
@responses.activate
|
|
def test_check_connection_fail(mocker):
|
|
responses.add(responses.GET, "https://api.billwithorb.com/v1/ping", json={"error": "Unauthorized"}, status=401)
|
|
source = SourceOrb()
|
|
logger_mock = MagicMock()
|
|
(ok, err) = source.check_connection(logger_mock, MagicMock())
|
|
assert (ok, type(err)) == (False, HTTPError)
|
|
|
|
|
|
def test_streams(mocker):
|
|
source = SourceOrb()
|
|
config_mock = MagicMock()
|
|
sample_config = {"api_key": "test-token", "start_date": "2023-01-25T00:00:00Z"}
|
|
config_mock.get.side_effect = sample_config.get
|
|
streams = source.streams(config_mock)
|
|
expected_streams_number = 6
|
|
assert len(streams) == expected_streams_number
|