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/')