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.""" """Consts for testing."""
TEST_DB = "test.db" 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.http import create_app
from frigate.models import Event, Recordings 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): class TestHttp(unittest.TestCase):
@ -106,9 +106,8 @@ class TestHttp(unittest.TestCase):
self.db.close() self.db.close()
try: try:
os.remove("test.db") for file in TEST_DB_CLEANUPS:
os.remove("test.db-shm") os.remove(file)
os.remove("test.db-wal")
except OSError: except OSError:
pass pass