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):
374
374
member2 = await m .Member .get (member .pk )
375
375
assert member2 == member
376
376
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
377
406
378
407
@pytest .mark .asyncio
379
408
async def test_delete (m ):
You can’t perform that action at this time.
0 commit comments