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