16 lines
594 B
Python
16 lines
594 B
Python
import pytest
|
|
from django.core.files.uploadedfile import SimpleUploadedFile
|
|
from plants.models import Location
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_location_cover_photo_field_exists():
|
|
"""Location can be saved with and without a cover_photo."""
|
|
loc = Location.objects.create(name='Test Garden')
|
|
assert loc.cover_photo.name == '' or loc.cover_photo.name is None
|
|
|
|
fake_img = SimpleUploadedFile('cover.jpg', b'fake-image-data', content_type='image/jpeg')
|
|
loc.cover_photo = fake_img
|
|
loc.save()
|
|
loc.refresh_from_db()
|
|
assert loc.cover_photo.name.startswith('locations/')
|