python-amazon-sp-api 1.9.34__py3-none-any.whl → 2.0.7__py3-none-any.whl

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)