import pytest, json
from model_bakery import baker
from products.models import NewsAndPressReleases, NewsGallery


@pytest.mark.django_db()
class TestNewsGalleryModel:
    def test_return_str_model(self):
        modified_description = json.dumps({"delta": "", "html": "Hello World"})

        news = baker.make(
            NewsAndPressReleases,
            title="Sample News 1",
            short_description=modified_description,
            long_description=modified_description,
            type="news",
        )

        news_gallery = baker.make(NewsGallery, news=news)

        assert str(news_gallery) == f"Image of '{news_gallery.news.title}'"