python-amazon-sp-api 1.7.5__py3-none-any.whl → 2.0.10__py3-none-any.whl

tests/client/test_base.py

@@ -1,163 +0,0 @@
-import os
-import pytest
-
-from sp_api.api import FulfillmentInbound
-from sp_api.base import AccessTokenClient
-from sp_api.base import Marketplaces, MissingCredentials, Client, SellingApiForbiddenException
-from sp_api.base.credential_provider import FromCodeCredentialProvider, FromEnvironmentVariablesCredentialProvider, \
-    FromSecretsCredentialProvider, FromConfigFileCredentialProvider, required_credentials
-from sp_api.base.exceptions import MissingScopeException
-
-refresh_token = '<refresh_token>'
-lwa_app_id = '<lwa_app_id>'
-lwa_client_secret = '<lwa_client_secret>'
-
-
-class Res:
-    status_code = 200
-    method = 'GET'
-    headers = {}
-    def json(self):
-        return {'foo': 'bar'}
-
-    def __getattr__(self, item):
-        return item
-
-
-def test_client_request():
-    try:
-        Client()._request('', data=dict())
-    except SellingApiForbiddenException as e:
-        assert isinstance(e, SellingApiForbiddenException)
-
-
-def test_client_timeout():
-    client = Client(timeout=1)
-    assert client.timeout == 1
-    client = Client()
-    assert client.timeout is None
-
-
-def test_api_response_has_next_token():
-    res = FulfillmentInbound().get_shipments(QueryType='SHIPMENT')
-    assert res.next_token is not None
-
-
-def test_marketplaces():
-    assert Marketplaces.DE.region == 'eu-west-1'
-    assert Marketplaces.US.marketplace_id == 'ATVPDKIKX0DER'
-
-
-def test_from_code_credential_provider():
-    p = FromCodeCredentialProvider(credentials=dict(
-        refresh_token=refresh_token,
-        lwa_app_id=lwa_app_id,
-        lwa_client_secret=lwa_client_secret,
-    ))
-    assert p.credentials is not None
-    assert isinstance(p.credentials, dict)
-
-
-def test_from_code_credential_provider_no_refresh_token():
-    p = FromCodeCredentialProvider(credentials=dict(
-        lwa_app_id=lwa_app_id,
-        lwa_client_secret=lwa_client_secret,
-    ))
-    assert p.credentials is not None
-    assert isinstance(p.credentials, dict)
-    assert p.credentials.get('refresh_token') is None
-
-
-@pytest.mark.order(-2)
-def test_env_vars_provider():
-    os.environ['SP_API_REFRESH_TOKEN'] = 'foo'
-    os.environ['LWA_APP_ID'] = 'foo'
-    os.environ['LWA_CLIENT_SECRET'] = 'foo'
-
-    p = FromEnvironmentVariablesCredentialProvider()()
-    assert 'refresh_token' in p
-
-    os.environ.pop('SP_API_REFRESH_TOKEN')
-    os.environ.pop('LWA_APP_ID')
-    os.environ.pop('LWA_CLIENT_SECRET')
-
-
-@pytest.mark.order(-1)
-def test_from_secrets():
-    os.environ['SP_API_AWS_SECRET_ID'] = 'testing/sp-api-foo'
-    try:
-        p = FromSecretsCredentialProvider()()
-        assert 'refresh_token' in p
-        assert p.get('refresh_token') == 'foo'
-        os.environ.pop('SP_API_AWS_SECRET_ID')
-    except MissingCredentials as e:
-        assert isinstance(e, MissingCredentials)
-
-
-def test_from_config_file_provider():
-    try:
-        p = FromConfigFileCredentialProvider()()
-        assert p.get('refresh_token') is not None
-    except MissingCredentials as e:
-        assert isinstance(e, MissingCredentials)
-
-
-def test_req():
-    assert len(required_credentials) == 2
-
-
-def test_client():
-    client = Client(marketplace=Marketplaces.UK)
-    assert client.marketplace_id == Marketplaces.UK.marketplace_id
-    assert client.credentials is not None
-    assert client.endpoint == Marketplaces.UK.endpoint
-    assert client.region == Marketplaces.UK.region
-    assert client.restricted_data_token is None
-    assert isinstance(client._auth, AccessTokenClient)
-
-    assert isinstance(client._get_cache_key(), str)
-    assert isinstance(client._get_cache_key('test'), str)
-
-    assert client.headers['host'] == client.endpoint[8:]
-    assert len(client.headers.keys()) == 5
-
-    assert client.auth is not None
-    try:
-        x = client.grantless_auth
-    except MissingScopeException as e:
-        assert isinstance(e, MissingScopeException)
-
-    try:
-        client._request('', data={})
-    except SellingApiForbiddenException as e:
-        assert isinstance(e, SellingApiForbiddenException)
-    try:
-        client._request('', params={})
-    except SellingApiForbiddenException as e:
-        assert isinstance(e, SellingApiForbiddenException)
-
-    check = client._check_response(Res())
-    assert check.payload['foo'] == 'bar'
-
-    r = Res()
-    r.method = 'POST'
-    check = client._check_response(r)
-    assert check.payload['foo'] == 'bar'
-    assert check('foo') == 'bar'
-    assert check.foo == 'bar'
-    assert check()['foo'] == 'bar'
-
-    r.method = 'DELETE'
-    check = client._check_response(r)
-    assert check.payload['foo'] == 'bar'
-    assert check('foo') == 'bar'
-    assert check.foo == 'bar'
-    assert check()['foo'] == 'bar'
-
-    client.grantless_scope = 'sellingpartnerapi::notifications'
-    assert client.grantless_auth is not None
-
-    try:
-        client._request_grantless_operation('')
-    except SellingApiForbiddenException as e:
-        assert isinstance(e, SellingApiForbiddenException)

tests/client/test_credential_provider.py

@@ -1,45 +0,0 @@
-import os
-from unittest import mock
-
-from sp_api.base.credential_provider import FromCachedSecretsCredentialProvider
-
-
-REFRESH_TOKEN = '<refresh_token>'
-LWA_APP_ID = '<lwa_app_id>'
-LWA_CLIENT_SECRET = '<lwa_client_secret>'
-
-
-def test_from_cached_secrets_cp_without_secret_id_set():
-    with mock.patch.dict(os.environ, {"SP_API_AWS_SECRET_ID": ""}):
-        cp = FromCachedSecretsCredentialProvider()
-        cp.load_credentials()
-
-    assert cp.credentials is None
-
-
-def test_from_cached_secrets_cp_without_cache_available():
-    with mock.patch.dict(os.environ, {"SP_API_AWS_SECRET_ID": "test"}), \
-            mock.patch.object(FromCachedSecretsCredentialProvider, "_get_secret_cache", return_value=None):
-        cp = FromCachedSecretsCredentialProvider()
-        cp.load_credentials()
-
-    assert cp.credentials is None
-
-
-def test_from_cached_secrets_cp_with_cache_available():
-    secret_content = {
-        "SP_API_REFRESH_TOKEN": REFRESH_TOKEN,
-        "LWA_APP_ID": LWA_APP_ID,
-        "LWA_CLIENT_SECRET": LWA_CLIENT_SECRET,
-    }
-
-    with mock.patch.dict(os.environ, {"SP_API_AWS_SECRET_ID": "test"}), \
-            mock.patch.object(FromCachedSecretsCredentialProvider, "get_secret_content", return_value=secret_content):
-        cp = FromCachedSecretsCredentialProvider()
-        cp.load_credentials()
-
-    assert cp.credentials == {
-        "refresh_token": REFRESH_TOKEN,
-        "lwa_app_id": LWA_APP_ID,
-        "lwa_client_secret": LWA_CLIENT_SECRET,
-    }

tests/client/test_helpers.py

@@ -1,142 +0,0 @@
-import enum
-import os
-from datetime import datetime, timedelta
-from io import BytesIO
-
-from sp_api.api import FulfillmentInbound, Orders
-from sp_api.base import fill_query_params, sp_endpoint, create_md5, nest_dict, deprecated
-from sp_api.util import KeyMaker, load_all_pages, throttle_retry, load_date_bound
-from sp_api.util.load_all_pages import make_sleep_time
-
-key_mapping = {
-    'sku': ['seller_sku', 'sellerSku'],
-    'title': ['product_name']
-}
-test_obj = {
-    'goo': {'x': {}},
-    'seller_sku': 1,
-    'product_name': {
-        'sellerSku': [
-            'seller_sku',
-            3,
-            {
-                'sellerSku': 22,
-                'product_name': {
-                    'title': 'Foo',
-                    'x': 'bar'
-                }
-            }
-        ]
-    }
-}
-
-
-def test_key_maker_from_dict():
-    r = KeyMaker(key_mapping, deep=True).convert_keys(test_obj)
-    assert isinstance(r, dict)
-    assert r.get('sku') == 1
-    assert r.get('seller_sku') is None
-    assert isinstance(r.get('title'), dict)
-    assert isinstance(r.get('title').get('sku'), list)
-
-
-def test_key_maker_from_list():
-    r = KeyMaker(key_mapping, deep=True).convert_keys([test_obj])
-    assert isinstance(r, list)
-    assert len(r) == 1
-
-    assert r[0].get('sku') == 1
-    assert r[0].get('seller_sku') is None
-    assert isinstance(r[0].get('title'), dict)
-    assert isinstance(r[0].get('title').get('sku'), list)
-
-
-def test_key_maker_from_dict_not_deep():
-    r = KeyMaker(key_mapping, deep=False).convert_keys(test_obj)
-    assert r.get('sku') == 1
-    assert r.get('seller_sku') is None
-    assert isinstance(r.get('title'), dict)
-    assert isinstance(r.get('title').get('sellerSku'), list)
-
-
-def test_load_all_pages():
-    @throttle_retry()
-    @load_all_pages(extras=dict(QueryType='NEXT_TOKEN'))
-    def load_shipments(**kwargs):
-        return FulfillmentInbound().get_shipments(**kwargs)
-
-    for x in load_shipments(QueryType='SHIPMENT'):
-        assert x.payload is not None
-
-
-def test_load_all_pages_orders():
-    @throttle_retry()
-    @load_all_pages()
-    def load_all_orders(**kwargs):
-        return Orders().get_orders(**kwargs)
-
-    for x in load_all_orders(CreatedAfter='TEST_CASE_200', MarketplaceIds=["ATVPDKIKX0DER"]):
-        assert x.payload is not None
-
-
-def test_make_sleep_time():
-    x = make_sleep_time(2, False, 2)
-    assert x == 2
-
-    y = make_sleep_time(2, True, 2)
-    assert y == 0.5
-
-
-def test_load_all_pages1():
-    x = load_all_pages()
-    assert x is not None
-
-
-def test_fill_query_params():
-    assert fill_query_params('{}/{}', 'foo', 'bar') == 'foo/bar'
-
-
-def test_sp_endpoint_():
-    assert sp_endpoint('foo') is not None
-
-    @sp_endpoint('/api/call', method='POST')
-    def my_endpoint(**kwargs):
-        assert kwargs['path'] == '/api/call'
-        assert kwargs['method'] == 'POST'
-    my_endpoint()
-    assert my_endpoint.__name__ == 'my_endpoint'
-
-
-def test_create_md5():
-    b = BytesIO()
-    b.write(b'foo')
-    b.seek(0)
-    m = create_md5(b)
-    assert m == 'rL0Y20zC+Fzt72VPzMSk2A=='
-
-
-def test_nest_dict():
-    x = nest_dict({
-        "AmazonOrderId":1,
-        "ShipFromAddress.Name" : "Seller",
-        "ShipFromAddress.AddressLine1": "Street",
-    })
-    assert x['ShipFromAddress']['AddressLine1'] == 'Street'
-
-
-def test_deprecated():
-    assert deprecated(lambda x: x + 1)(1) == 2
-
-
-@load_date_bound()
-def dummy(**kwargs):
-    return lambda: kwargs
-
-
-def test_load_date_bound():
-    start = datetime.now() - timedelta(days=70)
-    end = datetime.now()
-    x = list(dummy(dataStartTime=start, dataEndTime=end))
-    assert len(x) == 3
-    assert x[1]()['dataStartTime'] == start + timedelta(days=30)
-    assert x[1]()['dataEndTime'] == start + timedelta(days=60)