@@ -25,6 +25,7 @@ def __init__(self):
2525 Store .remove_db_files (self .db_path )
2626
2727 def sync (self , model : Model ) -> bool :
28+ """ Returns True if changes were made and the model JSON was written. """
2829 self .model = model
2930 return sync_model (self .model , self .model_path )
3031
@@ -421,25 +422,25 @@ class EntityA:
421422def test_model_json_updates (env ):
422423 """ Tests situations where the model JSON should be written/should not be written. """
423424
424- def assert_model_json_written ( value : bool , * entities : _Entity ):
425+ def sync_entities ( * entities : _Entity ):
425426 model = Model ()
426427 for entity in entities :
427428 model .entity (entity )
428- assert env .sync (model ) == value
429+ return env .sync (model )
429430
430431 # Init
431432 @Entity ()
432433 class EntityA :
433434 id = Id ()
434435 name = String ()
435- assert_model_json_written ( True , EntityA )
436+ assert sync_entities ( EntityA )
436437
437438 # Add entity
438439 @Entity ()
439440 class EntityB :
440441 id = Id ()
441442 name = String ()
442- assert_model_json_written ( True , EntityB )
443+ assert sync_entities ( EntityB )
443444
444445 entityb_uid = EntityB .uid
445446
@@ -448,7 +449,7 @@ class EntityB:
448449 class EntityC :
449450 id = Id ()
450451 name = String ()
451- assert_model_json_written ( True , EntityC )
452+ assert sync_entities ( EntityC )
452453
453454 # Noop
454455 model = Model ()
@@ -461,25 +462,25 @@ class EntityD:
461462 id = Id ()
462463 name = String ()
463464 age = Int8 ()
464- assert_model_json_written ( True , EntityC , EntityD )
465+ assert sync_entities ( EntityC , EntityD )
465466
466467 # Noop
467- assert_model_json_written ( False , EntityC , EntityD )
468+ assert sync_entities ( EntityC , EntityD ) is False
468469
469470 # Replace entity
470471 @Entity ()
471472 class EntityE :
472473 id = Id ()
473- assert_model_json_written ( True , EntityD , EntityE )
474+ assert sync_entities ( EntityD , EntityE )
474475
475476 # Noop
476- assert_model_json_written ( False , EntityD , EntityE )
477+ assert sync_entities ( EntityD , EntityE ) is False
477478
478479 # Remove entity
479- assert_model_json_written ( True , EntityD )
480+ assert sync_entities ( EntityD )
480481
481482 # Noop
482- assert_model_json_written ( False , EntityD )
483+ assert sync_entities ( EntityD ) is False
483484
484485 # Add property
485486 @Entity ()
@@ -488,7 +489,7 @@ class EntityD:
488489 name = String ()
489490 age = Int8 ()
490491 my_prop = String ()
491- assert_model_json_written ( True , EntityD )
492+ assert sync_entities ( EntityD )
492493
493494 my_prop_uid = EntityD .get_property ("my_prop" ).uid
494495
@@ -499,18 +500,18 @@ class EntityD:
499500 name = String ()
500501 age = Int8 ()
501502 my_prop_renamed = String (uid = my_prop_uid )
502- assert_model_json_written ( True , EntityD )
503+ assert sync_entities ( EntityD )
503504
504505 # Noop
505- assert_model_json_written ( False , EntityD )
506+ assert sync_entities ( EntityD ) is False
506507
507508 # Remove property
508509 @Entity ()
509510 class EntityD :
510511 id = Id ()
511512 name = String ()
512513 age = Int8 ()
513- assert_model_json_written ( True , EntityD )
514+ assert sync_entities ( EntityD )
514515
515516
516517def test_model_uid_already_assigned (env ):
0 commit comments