def _(self, obj_class, data=None, add_fields=True, with_background_tasks=False): """Handle generation of `Control` objects.""" data = data if data is not None else {} # pylint: disable=protected-access obj_name = models.Control._inflector.table_singular # pylint: enable=protected-access obj_dict = self.obj_to_dict(models.Control(), obj_name) defaults = { obj_name: { "title": factories.random_str(), "context": None, "recipients": "Admin,Control Operators,Control Owners", "send_by_default": 0, "assertions": '["test assertion"]', "review_status": all_models.Review.STATES.UNREVIEWED, "review_status_display_name": "some status", } } obj_dict[obj_name].update(defaults[obj_name]) if issubclass(obj_class, Synchronizable): obj_dict[obj_name].update(self._get_synchronizable_obj_dict()) obj_dict[obj_name].update(data[obj_name] if obj_name in data else data) return self.generate(models.Control, obj_name=obj_name, data=obj_dict, with_background_tasks=with_background_tasks)
def _(self, obj_class, data=None, add_fields=True, with_background_tasks=False): """Handle generation of `Control` objects.""" data = data if data is not None else {} # pylint: disable=protected-access obj_name = models.Control._inflector.table_singular # pylint: enable=protected-access obj_dict = self.obj_to_dict(models.Control(), obj_name) defaults = { obj_name: { "title": factories.random_str(), "context": None, "recipients": "Admin,Control Operators,Control Owners", "send_by_default": 0, "assertions": [{ "id": factories.ControlAssertionFactory().id }] } } obj_dict[obj_name].update(defaults[obj_name]) obj_dict[obj_name].update(data[obj_name] if obj_name in data else data) return self.generate(models.Control, obj_name=obj_name, data=obj_dict, with_background_tasks=with_background_tasks)