Skip to content

Commit

Permalink
Added test for HashModel all_pks()
Browse files Browse the repository at this point in the history
  • Loading branch information
Simon Prickett committed Apr 12, 2022
1 parent 274ff78 commit eefa5a9
Showing 1 changed file with 29 additions and 0 deletions.
29 changes: 29 additions & 0 deletions tests/test_hash_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -374,6 +374,35 @@ async def test_saves_model_and_creates_pk(m):
member2 = await m.Member.get(member.pk)
assert member2 == member

@pytest.mark.asyncio
async def test_all_pks(m):
member = m.Member(
first_name="Simon",
last_name="Prickett",
email="s@example.com",
join_date=today,
age=97,
bio="This is a test user to be deleted.",
)

await member.save()

member1 = m.Member(
first_name="Andrew",
last_name="Brookins",
email="a@example.com",
join_date=today,
age=38,
bio="This is a test user to be deleted.",
)

await member1.save()

pk_list = []
async for pk in await m.Member.all_pks():
pk_list.append(pk)

assert len(pk_list) == 2

@pytest.mark.asyncio
async def test_delete(m):
Expand Down

0 comments on commit eefa5a9

Please sign in to comment.