def company_business_type_post_migrate(sender, **kwargs):
"""
Ensures all business types are loaded to the database.
Any new business types are created, and any existing ones are updated if the label has changed.
"""
load_constants_to_database(BusinessTypeConstant, BusinessType)
def test_load_constants_to_database():
"""
Test loading constants to the database.
Makes sure that new values are created, existing ones are updated and none are deleted.
"""
initial_objects = [
{
'id': 'e2b77f5f-a3d9-4c48-9d40-5a5427ddcfc2',
'name': 'Object 1',
},
{
'id': 'c2ed6ff6-4a09-41ba-bda2-f4cdb2f96833',
'name': 'Object 2',
},
{
'id': '09afd6ef-deff-4b0f-9c5b-4816d3ddac09',
'name': 'Object 3',
},
{
'id': 'c8ecf162-f14a-4ab2-a570-fb70a2435e6b',
'name': 'Object 4',
},
]
MetadataModel.objects.bulk_create(
[MetadataModel(**data) for data in initial_objects])
load_constants_to_database(_MetadataModelConstant, MetadataModel)
expected_items = {
(UUID('e2b77f5f-a3d9-4c48-9d40-5a5427ddcfc2'),
'Object 1'), # not deleted
(UUID('c2ed6ff6-4a09-41ba-bda2-f4cdb2f96833'),
'Object 2a'), # name updated
(UUID('09afd6ef-deff-4b0f-9c5b-4816d3ddac09'),
'Object 3b'), # name updated
(UUID('c8ecf162-f14a-4ab2-a570-fb70a2435e6b'),
'Object 4'), # unchanged
(UUID('6ea0e2a2-0b2b-408c-a621-aff49f58496e'), 'Object 5'), # created
}
actual_items = {(obj.id, obj.name) for obj in MetadataModel.objects.all()}
assert actual_items == expected_items