X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=test%2Fmemorydb.py;h=cd830cbd8acee32bbbe307c25cc7c959da535ddd;hb=HEAD;hp=2bbddabf511c403f76805234e87ce2bf80c863e0;hpb=d3f8ed31357f1be1805f806e94b478116a8012e6;p=roundup.git diff --git a/test/memorydb.py b/test/memorydb.py index 2bbddab..cd830cb 100644 --- a/test/memorydb.py +++ b/test/memorydb.py @@ -118,6 +118,8 @@ class BasicDatabase(dict): def get(self, infoid, value, default=None): return self[infoid].get(value, default) def getall(self, infoid): + if infoid not in self: + raise KeyError(infoid) return self[infoid] def set(self, infoid, **newvalues): self[infoid].update(newvalues)