23 lines
904 B
Python
23 lines
904 B
Python
#
|
|
# Copyright (c) 2021 Airbyte, Inc., all rights reserved.
|
|
#
|
|
|
|
from airbyte_cdk.sources.streams.http.auth import MultipleTokenAuthenticator
|
|
from source_github import SourceGithub
|
|
|
|
|
|
def test_single_token():
|
|
authenticator = SourceGithub._get_authenticator({"access_token": "123"})
|
|
assert isinstance(authenticator, MultipleTokenAuthenticator)
|
|
assert ["123"] == authenticator._tokens
|
|
authenticator = SourceGithub._get_authenticator({"credentials": {"access_token": "123"}})
|
|
assert ["123"] == authenticator._tokens
|
|
authenticator = SourceGithub._get_authenticator({"credentials": {"personal_access_token": "123"}})
|
|
assert ["123"] == authenticator._tokens
|
|
|
|
|
|
def test_multiple_tokens():
|
|
authenticator = SourceGithub._get_authenticator({"access_token": "123, 456"})
|
|
assert isinstance(authenticator, MultipleTokenAuthenticator)
|
|
assert ["123", "456"] == authenticator._tokens
|