import pytest, json
from model_bakery import baker
from products.models import Products, ProductGallery


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

        product = baker.make(
            Products, name="Sample Product 1", description=modified_body
        )

        pg = baker.make(ProductGallery, product=product)

        assert str(pg) == f"Image of {pg.product.name}"