diff --git a/tricircle/tests/unit/utils.py b/tricircle/tests/unit/utils.py index f9b8c9c1..a4886710 100644 --- a/tricircle/tests/unit/utils.py +++ b/tricircle/tests/unit/utils.py @@ -107,6 +107,9 @@ class DotDict(dict): return dummy_value_map[item] return self.get(item) + def __setattr__(self, name, value): + self[name] = value + def to_dict(self): return self @@ -119,6 +122,11 @@ class DotDict(dict): def save(self, session=None): pass + def update_fields(self, obj_data): + for k, v in obj_data.items(): + if k in self: + setattr(self, k, v) + class DotList(list): def all(self):