From a81fa1af543a539a4a44afa514505907cdf00e29 Mon Sep 17 00:00:00 2001 From: Luke Murphy Date: Tue, 2 Feb 2021 22:05:56 +0100 Subject: [PATCH] Use super for SimpleDatabase operations Closes https://github.com/decentral1se/xbotlib/issues/5. --- test_xbotlib.py | 7 +++++++ xbotlib.py | 4 ++-- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/test_xbotlib.py b/test_xbotlib.py index 6c3601e..0466518 100644 --- a/test_xbotlib.py +++ b/test_xbotlib.py @@ -101,3 +101,10 @@ def test_config(config): assert config.serve assert config.storage == "file" assert config.output == "." + + +def test_simple_message_delete(tmp_db_path): + db = SimpleDatabase(tmp_db_path, log) + db["foo"] = "bar" + del db["foo"] + assert "foo" not in db diff --git a/xbotlib.py b/xbotlib.py index a38b509..57154de 100644 --- a/xbotlib.py +++ b/xbotlib.py @@ -59,12 +59,12 @@ class SimpleDatabase(dict): def __setitem__(self, key, val): """Write data to the database.""" - dict.__setitem__(self, key, val) + super().__setitem__(key, val) self._dumps() def __delitem__(self, key): """Remove data from the database.""" - dict.__delitem__(key) + super().__delitem__(key) self._dumps() def update(self, *args, **kwargs):