def test_retrieve_company_profile_mixin_not_logged_in(rf):
request = rf.get('/')
request.sso_user = None
mixin = mixins.PrepopulateFormMixin()
mixin.request = request
assert mixin.company_profile is None
def test_retrieve_company_profile_mixin_not_ok(rf):
request = rf.get('/')
request.sso_user = mock.Mock(session_id=123)
mixin = mixins.PrepopulateFormMixin()
mixin.request = request
url = 'http://api.trade.great:8000/supplier/company/'
with requests_mock.mock() as mocked:
mocked.get(url, status_code=503)
company_profile = mixin.company_profile
assert company_profile is None
def test_retrieve_company_profile_mixin_name_guessing_user(
rf, full_name, first_name, last_name, company_profile):
company_profile.return_value = create_response(status_code=404)
request = rf.get('/')
request.user = SSOUser(session_id=123,
first_name=first_name,
last_name=last_name)
mixin = mixins.PrepopulateFormMixin()
mixin.request = request
assert mixin.guess_given_name == first_name
assert mixin.guess_family_name == last_name
def test_retrieve_company_profile_mixin_success(rf):
request = rf.get('/')
request.sso_user = mock.Mock(session_id=123)
mixin = mixins.PrepopulateFormMixin()
mixin.request = request
url = 'http://api.trade.great:8000/supplier/company/'
expected = {'key': 'value'}
with requests_mock.mock() as mocked:
mocked.get(url, json=expected)
company_profile = mixin.company_profile
assert company_profile == expected
def test_retrieve_company_profile_mixin_name_guessing(
rf, full_name, first_name, last_name
):
request = rf.get('/')
request.sso_user = mock.Mock(session_id=123)
mixin = mixins.PrepopulateFormMixin()
mixin.request = request
url = 'http://api.trade.great:8000/supplier/company/'
expected = {'postal_full_name': full_name}
with requests_mock.mock() as mocked:
mocked.get(url, json=expected)
assert mixin.guess_given_name == first_name
assert mixin.guess_family_name == last_name