from django.contrib.auth.models import User from django.test import TestCase from django.urls import reverse from bookmarks.models import Toast from bookmarks.tests.helpers import BookmarkFactoryMixin, random_sentence, disable_logging class ToastsViewTestCase(TestCase, BookmarkFactoryMixin): def setUp(self) -> None: user = self.get_or_create_test_user() self.client.force_login(user) def create_toast(self, user: User = None, message: str = None, acknowledged: bool = False): if not user: user = self.user if not message: message = random_sentence() toast = Toast(owner=user, key='test', message=message, acknowledged=acknowledged) toast.save() return toast def test_should_render_unacknowledged_toasts(self): self.create_toast() self.create_toast() self.create_toast(acknowledged=True) response = self.client.get(reverse('bookmarks:index')) # Should render toasts container self.assertContains(response, '