Экземпляр объекта модели Django test.py ограничен одним тестом - PullRequest
Новая
       23

Экземпляр объекта модели Django test.py ограничен одним тестом

0 голосов
/ 01 ноября 2019

Хорошо, это довольно странно. Это мой код, который работает:

test.py 

from django.test import TestCase
from django.urls import reverse, resolve
from django.utils import timezone
import datetime

from ..models import Realtor, Listing, Contact
from ..views import listing

from django.test import override_settings
from PIL import Image
from io import BytesIO
from django.core.files import File
import tempfile

def get_image_file(name='test.png', ext='png', size=(50, 50), color=(256, 0, 0)):
    file_obj = BytesIO()
    image = Image.new("RGB", size=size, color=color)
    image.save(file_obj, ext)
    file_obj.seek(0)
    return File(file_obj, name=name)

class RealestateListingViewTest(TestCase):
    @override_settings(MEDIA_ROOT=tempfile.TemporaryDirectory(prefix='mediatest').name)
    def setUp(self):
        image_file = get_image_file
        Realtor.objects.create(name='sample_realtor', photo=get_image_file())
        Listing.objects.create(title='listing_sample', address='sample', realtor=Realtor.objects.get(pk=1), city='sample', state='sample', zipcode='1234', price='555555', bedrooms='1', bathrooms='1', garage='1', sqft='123', lot_size='123', photo_main=get_image_file(), photo_1=get_image_file(), photo_2=get_image_file(), photo_3=get_image_file(), photo_4=get_image_file(), photo_5=get_image_file(), photo_6=get_image_file())
        url = reverse('btre:listing', kwargs={'listing_id': 1})
        self.response = self.client.get(url)

    def test_listing_status_code(self):
        self.assertEquals(self.response.status_code, 200)

Приведенный выше код работает, но всякий раз, когда я добавляю в тест еще один тест, например: 

    def test_listing_url_resolves_listing_view(self):
        view = resolve('/realestate/listing/1')
        self.assertEqual(view.func, listing)

Я получаю эту ошибку: 

======================================================================
ERROR: test_listing_url_resolves_listing_view (realestate.tests.test_view_listing.RealestateListingViewTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "C:\Users\Storm\Dev\btre_project\realestate\tests\test_view_listing.py", line 35, in setUp
    Listing.objects.create(title='listing_sample', address='sample', realtor=Realtor.objects.get(pk=1), city='sample', state='sample', zipcode='1234', price='555555', bedrooms='1', bathrooms='1', garage='1', sqft='123', lot_size='123', photo_main=get_image_file(), photo_1=get_image_file(), photo_2=get_image_file(), photo_3=get_image_file(), photo_4=get_image_file(), photo_5=get_image_file(), photo_6=get_image_file())
  File "C:\Users\Storm\Envs\btre\lib\site-packages\django\db\models\manager.py", line 82, in manager_method
    return getattr(self.get_queryset(), name)(*args, **kwargs)
  File "C:\Users\Storm\Envs\btre\lib\site-packages\django\db\models\query.py", line 408, in get
    self.model._meta.object_name
realestate.models.Realtor.DoesNotExist: Realtor matching query does not exist.

У кого-нибудь есть идеи, как и почему это происходит?

Также обратите внимание, что это работает, пока существует только один тестсуществует в классе TestCase, поэтому мне это очень странно.

Также я попробовал этот метод для создания файла изображения: 

class RealestateListingViewTest(TestCase):
    @staticmethod
    def get_image_file(name='test.png', ext='png', size=(50, 50), color=(256, 0, 0)):
        file_obj = BytesIO()
        image = Image.new("RGB", size=size, color=color)
        image.save(file_obj, ext)
        file_obj.seek(0)
        return File(file_obj, name=name)

1 Ответ

0 голосов
/ 01 ноября 2019

Вы не показали тестируемое представление, но, вероятно, 1 в этом URL является PK объекта.

Проблема в том, что хотя база данных очищается после каждого теста, последовательности не сбрасываются. Поэтому во второй раз, когда вы создаете элемент в методе setUp, его значение PK не будет равно 1.

Вы не должны полагаться на конкретные идентификаторы. Присвойте созданный элемент переменной и используйте его: 

    listing = Listing.objects.create(title='listing_sample', address='sample', realtor=Realtor.objects.get(pk=1), city='sample', state='sample', zipcode='1234', price='555555', bedrooms='1', bathrooms='1', garage='1', sqft='123', lot_size='123', photo_main=get_image_file(), photo_1=get_image_file(), photo_2=get_image_file(), photo_3=get_image_file(), photo_4=get_image_file(), photo_5=get_image_file(), photo_6=get_image_file())
    url = reverse('btre:listing', kwargs={'listing_id': listing.id})
...