Add cleanup files in const

This commit is contained in:
Nick Mowen 2022-06-16 10:16:13 -06:00
parent f7f29437d3
commit 28b875521a
2 changed files with 4 additions and 4 deletions

View File

@ -1,3 +1,4 @@
"""Consts for testing."""
TEST_DB = "test.db"
TEST_DB_CLEANUPS = ["test.db", "test.db-shm", "test.db-wal"]

View File

@ -14,7 +14,7 @@ from frigate.config import FrigateConfig
from frigate.http import create_app
from frigate.models import Event, Recordings
from frigate.test.const import TEST_DB
from frigate.test.const import TEST_DB, TEST_DB_CLEANUPS
class TestHttp(unittest.TestCase):
@ -106,9 +106,8 @@ class TestHttp(unittest.TestCase):
self.db.close()
try:
os.remove("test.db")
os.remove("test.db-shm")
os.remove("test.db-wal")
for file in TEST_DB_CLEANUPS:
os.remove(file)
except OSError:
pass