def test_store_user_address_create_new_address_if_not_associated(address):
"""Ensure storing an address that is not associated to the given user
triggers the creation of a new address, but uses the existing one instead.
"""
user = User.objects.create_user("*****@*****.**", "password")
expected_user_addresses_count = 1
store_user_address(user, address, AddressType.BILLING)
assert user.addresses.count() == expected_user_addresses_count
assert user.default_billing_address_id != address.pk
def test_store_user_address_uses_existing_one(address):
"""Ensure storing an address that is already associated to the given user doesn't
create a new address, but uses the existing one instead.
"""
user = User.objects.create_user("*****@*****.**", "password")
user.addresses.add(address)
expected_user_addresses_count = 1
store_user_address(user, address, AddressType.BILLING)
assert user.addresses.count() == expected_user_addresses_count
assert user.default_billing_address_id == address.pk
def test_store_user_address_uses_existing_one_despite_duplicated(address):
"""Ensure storing an address handles the possibility of an user
having the same address associated to them multiple time is handled properly.
It should use the first identical address associated to the user.
"""
same_address = Address.objects.create(**address.as_data())
user = User.objects.create_user("*****@*****.**", "password")
user.addresses.set([address, same_address])
expected_user_addresses_count = 2
store_user_address(user, address, AddressType.BILLING)
assert user.addresses.count() == expected_user_addresses_count
assert user.default_billing_address_id == address.pk