Skip to content

Commit eefa5a9

Browse files
author
Simon Prickett
committed
Added test for HashModel all_pks()
1 parent 274ff78 commit eefa5a9

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

tests/test_hash_model.py

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -374,6 +374,35 @@ async def test_saves_model_and_creates_pk(m):
374374
member2 = await m.Member.get(member.pk)
375375
assert member2 == member
376376

377+
@pytest.mark.asyncio
378+
async def test_all_pks(m):
379+
member = m.Member(
380+
first_name="Simon",
381+
last_name="Prickett",
382+
383+
join_date=today,
384+
age=97,
385+
bio="This is a test user to be deleted.",
386+
)
387+
388+
await member.save()
389+
390+
member1 = m.Member(
391+
first_name="Andrew",
392+
last_name="Brookins",
393+
394+
join_date=today,
395+
age=38,
396+
bio="This is a test user to be deleted.",
397+
)
398+
399+
await member1.save()
400+
401+
pk_list = []
402+
async for pk in await m.Member.all_pks():
403+
pk_list.append(pk)
404+
405+
assert len(pk_list) == 2
377406

378407
@pytest.mark.asyncio
379408
async def test_delete(m):

0 commit comments

Comments
 (0)