from django.core.exceptions import ValidationError from django.db.utils import IntegrityError from django.test import TestCase from apps.dashboard.models import Applet, Item, List, UserApplet from apps.lyric.models import User class ItemModelTest(TestCase): def test_item_is_related_to_list(self): mylist = List.objects.create() item = Item() item.list = mylist item.save() self.assertIn(item, mylist.item_set.all()) def test_cannot_save_null_list_items(self): mylist = List.objects.create() item = Item(list=mylist, text=None) with self.assertRaises(IntegrityError): item.save() def test_cannot_save_empty_list_items(self): mylist = List.objects.create() item = Item(list=mylist, text="") with self.assertRaises(ValidationError): item.full_clean() def test_duplicate_items_are_invalid(self): mylist = List.objects.create() Item.objects.create(list=mylist, text="jklol") with self.assertRaises(ValidationError): item = Item(list=mylist, text="jklol") item.full_clean() def test_still_can_save_same_item_to_different_lists(self): list1 = List.objects.create() list2 = List.objects.create() Item.objects.create(list=list1, text="nojk") item = Item(list=list2, text="nojk") item.full_clean() # should not raise class ListModelTest(TestCase): def test_get_absolute_url(self): mylist = List.objects.create() self.assertEqual(mylist.get_absolute_url(), f"/dashboard/list/{mylist.id}/") def test_list_items_order(self): list1 = List.objects.create() item1 = Item.objects.create(list=list1, text="i1") item2 = Item.objects.create(list=list1, text="item 2") item3 = Item.objects.create(list=list1, text="3") self.assertEqual( list(list1.item_set.all()), [item1, item2, item3], ) def test_lists_can_have_owners(self): user = User.objects.create(email="a@b.cde") mylist = List.objects.create(owner=user) self.assertIn(mylist, user.lists.all()) def test_list_owner_is_optional(self): List.objects.create() def test_list_name_is_first_item_text(self): list_ = List.objects.create() Item.objects.create(list=list_, text="first item") Item.objects.create(list=list_, text="second item") self.assertEqual(list_.name, "first item") class AppletModelTest(TestCase): def setUp(self): self.applet = Applet.objects.create( slug="my-applet", name="My Applet", default_visible=True ) def test_applet_can_be_created(self): self.assertEqual(Applet.objects.get(slug="my-applet"), self.applet) def test_applet_slug_is_unique(self): with self.assertRaises(IntegrityError): Applet.objects.create(slug="my-applet", name="Second") def test_applet_str(self): self.assertEqual(str(self.applet), "My Applet") def test_applet_grid_defaults(self): self.assertEqual(self.applet.grid_cols, 12) self.assertEqual(self.applet.grid_rows, 3) class UserAppletModelTest(TestCase): def setUp(self): self.user = User.objects.create(email="a@b.cde") self.applet, _ = Applet.objects.get_or_create(slug="username", defaults={"name": "Username"}) def test_user_applet_links_user_to_applet(self): ua = UserApplet.objects.create(user=self.user, applet=self.applet, visible=True) self.assertIn(ua, self.user.user_applets.all()) def test_user_applet_unique_per_user_and_applet(self): UserApplet.objects.create(user=self.user, applet=self.applet, visible=True) with self.assertRaises(IntegrityError): UserApplet.objects.create(user=self.user, applet=self.applet, visible=False)