LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

make sure DBEncryptedFileMetadataManager closes db connection

Browse source
This commit is contained in:
Kay Kurokawa 2017-06-30 16:18:54 -04:00
parent 99c2130be7
commit 55c90e0657
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lbrynet/lbry_file/EncryptedFileMetadataManager.py
View file

@ -24,7 +24,7 @@ class DBEncryptedFileMetadataManager(object):
return self._open_db() return self._open_db()
def stop(self): def stop(self):
self.db_conn = None self.db_conn.close()
return defer.succeed(True) return defer.succeed(True)
def get_all_streams(self): def get_all_streams(self):

1
tests/unit/lbryfile/test_EncryptedFileMetadataManager.py
View file

@ -14,6 +14,7 @@ class DBEncryptedFileMetadataManagerTest(unittest.TestCase):
self.manager = DBEncryptedFileMetadataManager(self.db_dir) self.manager = DBEncryptedFileMetadataManager(self.db_dir)
def tearDown(self): def tearDown(self):
self.manager.stop()
shutil.rmtree(self.db_dir) shutil.rmtree(self.db_dir)
@defer.inlineCallbacks @defer.inlineCallbacks