def test_get_all_list_schema_classes(self, schema, _get_plugin_schemas):
list_schema_value = {"something": "something"}
schema.list_schemas = list_schema_value.copy()
_get_plugin_schemas.return_value = {"something_else": "something_else"}
result = schemas.get_all_list_schema_classes()
self.assertEqual(result, {
"something": "something",
"something_else": "something_else"
})
self.assertEqual(_get_plugin_schemas.call_count, 1)
self.assertEqual(schema.list_schemas, list_schema_value)
def get_column_context(self, **kwargs):
"""
Return the context for our columns
"""
active_schema = self.column_schema
all_list_schemas = get_all_list_schema_classes()
if 'tag' in kwargs and kwargs['tag'] in all_list_schemas:
if 'subtag' in kwargs and kwargs['subtag'] in all_list_schemas[kwargs['tag']]:
active_schema = all_list_schemas[kwargs['tag']][kwargs['subtag']]
elif 'default' in all_list_schemas[kwargs['tag']]:
active_schema = all_list_schemas[kwargs['tag']]['default']
else:
active_schema = all_list_schemas['default']
return _get_column_context(active_schema, **kwargs)
def get_column_context(self, **kwargs):
"""
Return the context for our columns
"""
active_schema = self.column_schema
all_list_schemas = get_all_list_schema_classes()
if 'tag' in kwargs and kwargs['tag'] in all_list_schemas:
if 'subtag' in kwargs and kwargs['subtag'] in all_list_schemas[
kwargs['tag']]:
active_schema = all_list_schemas[kwargs['tag']][
kwargs['subtag']]
elif 'default' in all_list_schemas[kwargs['tag']]:
active_schema = all_list_schemas[kwargs['tag']]['default']
else:
active_schema = all_list_schemas['default']
return _get_column_context(active_schema, **kwargs)
