diff --git a/test/test_database.py b/test/test_database.py index 5f020b5e963d622c02d68b6088a9cf1673035257..890b3375b7c9ea329e391957732bda32a0ce692a 100644 --- a/test/test_database.py +++ b/test/test_database.py @@ -5,7 +5,7 @@ import unittest import json from time import sleep from datetime import datetime -from soitool.database import Database +from soitool.database import Database, DBPATH from soitool.coder import get_code_length_needed TESTDATA_PATH = Path(__file__).parent.parent / "soitool/testdata" @@ -17,6 +17,7 @@ class DatabaseTest(unittest.TestCase): def setUp(self): """Connect to/create database.""" self.database = Database() + self.addCleanup(self.delete_db) def test_connection(self): """Assert connection is not None.""" @@ -288,6 +289,12 @@ class DatabaseTest(unittest.TestCase): ] self.assertRegex(code, "[A-Z0-9]") + def delete_db(self): + """Delete generated db-file.""" + del self.database + if os.path.exists(DBPATH): + os.remove(DBPATH) + if __name__ == "__main__": unittest.main()