From 99990a27f3a30198201e750205664d285c0f1374 Mon Sep 17 00:00:00 2001
From: "Anders H. Rebner" <anderhre@stud.ntnu.no>
Date: Mon, 23 Mar 2020 19:21:07 +0100
Subject: [PATCH] =?UTF-8?q?#68=20test=5Fdatabase.py=20sletter=20database-f?=
=?UTF-8?q?il=20mellom=20hver=20test,=20n=C3=A5=20fungerer=20testene?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
test/test_database.py | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/test/test_database.py b/test/test_database.py
index 5f020b5..890b337 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()
--
GitLab