Skip to content

Commit

Permalink
testing (Table): test equality
Browse files Browse the repository at this point in the history
  • Loading branch information
aarcex3 committed Oct 17, 2024
1 parent b847b37 commit ecf8683
Showing 1 changed file with 31 additions and 0 deletions.
31 changes: 31 additions & 0 deletions tests/table/instance/test_instance_equality.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
from piccolo.testing.test_case import AsyncTableTest
from tests.example_apps.music.tables import Band, Manager


class TestInstanceEquality(AsyncTableTest):
tables = [Manager, Band]

async def asyncSetUp(self):
await super().asyncSetUp()

self.manager = Manager(name="Guido")
await self.manager.save()

self.band = Band(
name="Pythonistas", manager=self.manager.id, popularity=100
)
await self.band.save()

async def test_instance_equality(self) -> None:
"""
Make sure for instance equailty.
"""
band_pk = await self.band.objects().first()
band = await self.band.objects(Band.manager).get(
(Band._meta.primary_key == band_pk.id)
)
manager_pk = await self.manager.objects().first()
manager = await self.manager.objects().get(
Manager._meta.primary_key == manager_pk.id
)
self.assertTrue(band.manager == manager)

0 comments on commit ecf8683

Please sign in to comment.