def retrieve_saved_tokens(customer):
"""Shortcut for retrieve_saved_tokens from the Gateway module.
Added as a convenience to allow access to core functions via
the bridge module exclusively.
"""
return gateway.retrieve_saved_tokens(customer)
def retrieve_saved_tokens(django_user=None, customer_code=None):
"""Shortcut for retrieve_saved_tokens from the Gateway module.
Added as a convenience to allow access to core functions via
the bridge module exclusively.
"""
return gateway.retrieve_saved_tokens(django_user=django_user,
customer_code=customer_code)
def test_retrieve_saved_tokens_by_customer_code():
"""Tests that tokens can be retrieved for a customer code."""
tokens = gateway.retrieve_saved_tokens('1')
assert tokens.django_user is None
assert tokens.customer_code == '1'
def test_retrieve_saved_tokens_by_django_user():
"""Tests that tokens can be retrieved for a Django user."""
tokens = gateway.retrieve_saved_tokens('1')
assert tokens.django_user == '1'
assert tokens.customer_code is None