File tree Expand file tree Collapse file tree 1 file changed +29
-0
lines changed Expand file tree Collapse file tree 1 file changed +29
-0
lines changed Original file line number Diff line number Diff 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
379408async def test_delete (m ):
You can’t perform that action at this time.
0 commit comments