From a78e9c750a7cfd5787b35e9f40236a1a32962d03 Mon Sep 17 00:00:00 2001 From: Daira Hopwood Date: Mon, 22 Feb 2016 16:04:19 +0000 Subject: [PATCH] WIP: add close method for MagicFolderDB and make MagicFolderDbTests use it. Signed-off-by: Daira Hopwood --- src/allmydata/magicfolderdb.py | 3 +++ src/allmydata/test/test_magic_folder.py | 2 ++ 2 files changed, 5 insertions(+) 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()