Skip to content

Commit 40b5295

Browse files
committed
test_model_json_updates: rename helper function to sync_entities #25
Also an explicit assert seems clearer to read in this case
1 parent 118f868 commit 40b5295

File tree

2 files changed

+17
-16
lines changed

2 files changed

+17
-16
lines changed

objectbox/model/idsync.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,7 @@ def sync(self) -> bool:
305305

306306
def sync_model(model: Model, model_filepath: str = "objectbox-model.json") -> bool:
307307
""" Syncs the provided model with the model JSON file.
308-
Returns True if the model JSON was written. """
308+
Returns True if changes were made and the model JSON was written. """
309309

310310
id_sync = IdSync(model, model_filepath)
311311
return id_sync.sync()

tests/test_idsync.py

Lines changed: 16 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -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:
421422
def 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

516517
def test_model_uid_already_assigned(env):

0 commit comments

Comments
 (0)