#68 test_database.py sletter database-fil mellom hver test, nå fungerer testene

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()