def test_get_limits_for_no_models(self):
blank_conf = EditRegionConfiguration()
template = Template(
"""{
"x": {
"name": "test"
}
}"""
)
blank_conf.raw_config = blank_conf.decode_template_region_configuration(template_instance=template)
blank_conf.config = blank_conf.get_template_region_configuration(raw_data=blank_conf.raw_config)
result = blank_conf.get_limits_for(region="x", chunk=User)
self.assertEqual(0, result)
def test_get_limits_for(self):
blank_conf = EditRegionConfiguration()
template = Template(
"""{
"x": {
"name": "test",
"models": {
"auth.User": 1,
"auth.Group": 0,
"auth.Permission": null
}
}
}"""
)
blank_conf.raw_config = blank_conf.decode_template_region_configuration(template_instance=template)
blank_conf.config = blank_conf.get_template_region_configuration(raw_data=blank_conf.raw_config)
result = blank_conf.get_limits_for(region="x", chunk=User)
self.assertEqual(1, result)
# 0 means don't show up!
result = blank_conf.get_limits_for(region="x", chunk=Group)
self.assertEqual(0, result)
result = blank_conf.get_limits_for(region="x", chunk=Permission)
self.assertEqual(None, result)
