diff --git a/src/allmydata/magicfolderdb.py b/src/allmydata/magicfolderdb.py index b1c81fe57..5ad6e7444 100644 --- a/src/allmydata/magicfolderdb.py +++ b/src/allmydata/magicfolderdb.py @@ -53,6 +53,9 @@ class MagicFolderDB(object): self.connection = connection self.cursor = connection.cursor() + def close(self): + self.connection.close() + def get_db_entry(self, relpath_u): """ Retrieve the entry in the database for a given path, or return None diff --git a/src/allmydata/test/test_magic_folder.py b/src/allmydata/test/test_magic_folder.py index c4591ed7e..7317a5dec 100644 --- a/src/allmydata/test/test_magic_folder.py +++ b/src/allmydata/test/test_magic_folder.py @@ -42,6 +42,8 @@ class MagicFolderDbTests(unittest.TestCase): self.failUnlessEqual(self.db.VERSION, 1) def tearDown(self): + if hasattr(self, 'db'): + self.db.close() shutil.rmtree(self.temp) return super(MagicFolderDbTests, self).tearDown()