harden test test_google_albums_sync

gilesknap · Feb 17, 2020 · 9cd6242 · 9cd6242
1 parent 9595049
commit 9cd6242
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/test-units/test_errors.py b/test/test-units/test_errors.py
@@ -183,11 +183,11 @@ def download_faves(self, expected=4, no_response=False, trash=True):
         )
         s.gp.start(s.parsed_args)
 
-        db = LocalData(s.root)
-        # Total of 1 out of media items
-        db.cur.execute("SELECT COUNT() FROM SyncFiles")
-        count = db.cur.fetchone()
-        self.assertEqual(expected, count[0])
+        with LocalData(s.root) as db:
+            # Total of 1 out of media items
+            db.cur.execute("SELECT COUNT() FROM SyncFiles")
+            count = db.cur.fetchone()
+            self.assertEqual(expected, count[0])
 
     class DummyResponse:
         @staticmethod