def test_build_draft_registry_1(self):
info = {
'myapp': [
{
'app_label': 'myapp',
'fields': ['img1', 'img2', 'img3'],
'specificity': 1,
'mapping': {
'SIZE': '100',
'DIMENSIONS': '100x100',
}
},
{
'app_label': 'myapp',
'fields': ['img1', 'img3'],
'specificity': 2,
'mapping': {
'FORMAT': 'png',
}
},
{
'app_label': 'myapp',
'fields': ['img3'],
'specificity': 3,
'mapping': {
'SIZE': '500',
'DIMENSIONS': '500x640',
'FORMAT': 'jpeg',
}
},
],
}
expected_draft_registry = {
'img1': {
'SIZE': '100',
'DIMENSIONS': '100x100',
'FORMAT': 'png'
},
'img2': {
'SIZE': '100',
'DIMENSIONS': '100x100'
},
'img3': {
'SIZE': '500',
'DIMENSIONS': '500x640',
'FORMAT': 'jpeg'
},
}
draft_registry = VimageConfig.build_draft_registry(info)
self.assertDictEqual(draft_registry, expected_draft_registry)
def test_build_draft_registry_4(self):
info = {
'myapp1': [
{
'app_label': 'myapp',
'fields': ['img1', 'img2', 'img3'],
'specificity': 1,
'mapping': {
'SIZE': '100',
'DIMENSIONS': '100x100',
}
},
],
'myapp2': [
{
'app_label': 'myapp',
'fields': ['img4', 'img5', 'img6'],
'specificity': 1,
'mapping': {
'DIMENSIONS': '666x666', # :)
}
},
],
}
expected_draft_registry = {
'img1': {
'SIZE': '100',
'DIMENSIONS': '100x100'
},
'img2': {
'SIZE': '100',
'DIMENSIONS': '100x100'
},
'img3': {
'SIZE': '100',
'DIMENSIONS': '100x100'
},
'img4': {
'DIMENSIONS': '666x666'
},
'img5': {
'DIMENSIONS': '666x666'
},
'img6': {
'DIMENSIONS': '666x666'
},
}
draft_registry = VimageConfig.build_draft_registry(info)
self.assertDictEqual(draft_registry, expected_draft_registry)