Test resetting db after each test.

Author: rohe

src/oidcendpoint/in_memory_db.py

@@ -18,3 +18,9 @@ def delete(self, key):
 
     def keys(self):
         return self.db.keys()
+
+    def close(self):
+        pass
+
+    def clear(self):
+        self.db = {}

src/oidcendpoint/sso_db.py

@@ -173,3 +173,9 @@ def remove_sub(self, sub):
         for _sid in self.get("sub2sid", sub):
             self.remove("sid2sub", _sid, sub)
         self.delete("sub2sid", sub)
+
+    def close(self):
+        self._db.close()
+
+    def clear(self):
+        self._db.clear()

tests/test_05_sso_db.py

@@ -126,12 +126,18 @@ def create_sdb(self):
         _db = ShelveDataBase(filename='shelf', flag='n', writeback=True)
         self.sso_db = SSODb(_db)
 
+    def _reset(self):
+        self.sso_db.clear()
+        self.sso_db.close()
+
     def test_map_sid2uid(self):
         self.sso_db.map_sid2uid("session id 1", "Lizz")
         assert self.sso_db.get_sids_by_uid("Lizz") == ["session id 1"]
+        self._reset()
 
     def test_missing_map(self):
         assert self.sso_db.get_sids_by_uid("Lizz") is None
+        self._reset()
 
     def test_multiple_map_sid2uid(self):
         self.sso_db.map_sid2uid("session id 1", "Lizz")
@@ -140,6 +146,7 @@ def test_multiple_map_sid2uid(self):
             "session id 1",
             "session id 2",
         }
+        self._reset()
 
     def test_map_unmap_sid2uid(self):
         self.sso_db.map_sid2uid("session id 1", "Lizz")
@@ -151,13 +158,15 @@ def test_map_unmap_sid2uid(self):
 
         self.sso_db.remove_sid2uid("session id 1", "Lizz")
         assert self.sso_db.get_sids_by_uid("Lizz") == ["session id 2"]
+        self._reset()
 
     def test_get_uid_by_sid(self):
         self.sso_db.map_sid2uid("session id 1", "Lizz")
         self.sso_db.map_sid2uid("session id 2", "Lizz")
 
         assert self.sso_db.get_uid_by_sid("session id 1") == "Lizz"
         assert self.sso_db.get_uid_by_sid("session id 2") == "Lizz"
+        self._reset()
 
     def test_remove_uid(self):
         self.sso_db.map_sid2uid("session id 1", "Lizz")
@@ -166,13 +175,16 @@ def test_remove_uid(self):
         self.sso_db.remove_uid("Lizz")
         assert self.sso_db.get_uid_by_sid("session id 1") is None
         assert self.sso_db.get_sids_by_uid("Lizz") is None
+        self._reset()
 
     def test_map_sid2sub(self):
         self.sso_db.map_sid2sub("session id 1", "abcdefgh")
         assert self.sso_db.get_sids_by_sub("abcdefgh") == ["session id 1"]
+        self._reset()
 
     def test_missing_sid2sub_map(self):
         assert self.sso_db.get_sids_by_sub("abcdefgh") is None
+        self._reset()
 
     def test_multiple_map_sid2sub(self):
         self.sso_db.map_sid2sub("session id 1", "abcdefgh")
@@ -181,6 +193,7 @@ def test_multiple_map_sid2sub(self):
             "session id 1",
             "session id 2",
         }
+        self._reset()
 
     def test_map_unmap_sid2sub(self):
         self.sso_db.map_sid2sub("session id 1", "abcdefgh")
@@ -192,6 +205,7 @@ def test_map_unmap_sid2sub(self):
 
         self.sso_db.remove_sid2sub("session id 1", "abcdefgh")
         assert self.sso_db.get_sids_by_sub("abcdefgh") == ["session id 2"]
+        self._reset()
 
     def test_get_sub_by_sid(self):
         self.sso_db.map_sid2sub("session id 1", "abcdefgh")
@@ -201,6 +215,7 @@ def test_get_sub_by_sid(self):
             "session id 1",
             "session id 2",
         }
+        self._reset()
 
     def test_remove_sub(self):
         self.sso_db.map_sid2sub("session id 1", "abcdefgh")
@@ -212,6 +227,7 @@ def test_remove_sub(self):
         # have not touched the others
         assert self.sso_db.get_sub_by_sid("session id 2") == "012346789"
         assert self.sso_db.get_sids_by_sub("012346789") == ["session id 2"]
+        self._reset()
 
     def test_get_sub_by_uid_same_sub(self):
         self.sso_db.map_sid2sub("session id 1", "abcdefgh")
@@ -223,6 +239,7 @@ def test_get_sub_by_uid_same_sub(self):
         res = self.sso_db.get_subs_by_uid("Lizz")
 
         assert set(res) == {"abcdefgh"}
+        self._reset()
 
     def test_get_sub_by_uid_different_sub(self):
         self.sso_db.map_sid2sub("session id 1", "abcdefgh")
@@ -234,3 +251,4 @@ def test_get_sub_by_uid_different_sub(self):
         res = self.sso_db.get_subs_by_uid("Lizz")
 
         assert set(res) == {"abcdefgh", "012346789"}
+        self._reset()