84 lines
1.9 KiB
Python
84 lines
1.9 KiB
Python
import pytest
|
|
from lettings.models import Address, Letting
|
|
from profiles.models import Profile
|
|
from django.contrib.auth.models import User
|
|
|
|
|
|
@pytest.fixture
|
|
@pytest.mark.django_db
|
|
def sample_address():
|
|
"""
|
|
creates temporary Addresses objects in test DB
|
|
:return: tuple of Address
|
|
"""
|
|
address1 = Address.objects.create(
|
|
number=22,
|
|
street="Quality Street",
|
|
city="New-York",
|
|
state="New-York",
|
|
zip_code=10010,
|
|
country_iso_code="US"
|
|
)
|
|
address2 = Address.objects.create(
|
|
number=18,
|
|
street="Rue Cocotte",
|
|
city="Paris",
|
|
state="Ile-de-France",
|
|
zip_code=75000,
|
|
country_iso_code="FR"
|
|
)
|
|
return address1, address2
|
|
|
|
|
|
@pytest.fixture
|
|
@pytest.mark.django_db
|
|
def sample_letting(sample_address):
|
|
"""
|
|
creates temporary Letting objects in test DB
|
|
:return: tuple of Lettings
|
|
"""
|
|
letting = Letting.objects.create(
|
|
title="Pretty thing",
|
|
address=sample_address[0],
|
|
)
|
|
letting2 = Letting.objects.create(
|
|
title="Ugly thing",
|
|
address=sample_address[1],
|
|
)
|
|
|
|
return letting, letting2
|
|
|
|
|
|
@pytest.fixture
|
|
@pytest.mark.django_db
|
|
def sample_profile():
|
|
"""
|
|
creates temporary Profile objects in test DB
|
|
:return: tuple of Profiles
|
|
"""
|
|
user = User.objects.create(
|
|
username='TestUser',
|
|
password='password',
|
|
)
|
|
user2 = User.objects.create(
|
|
username='TestUser2',
|
|
password='password2',
|
|
)
|
|
user3 = User.objects.create(
|
|
username='TestUser3',
|
|
password='password2',
|
|
)
|
|
profile = Profile.objects.create(
|
|
user=user,
|
|
favorite_city="Paris",
|
|
)
|
|
profile2 = Profile.objects.create(
|
|
user=user2,
|
|
favorite_city="Marseilles",
|
|
)
|
|
profile3 = Profile.objects.create(
|
|
user=user3,
|
|
favorite_city="Rennes",
|
|
)
|
|
return profile, profile2, profile3
|