scanner: Add a CommitTask

For committing the database during scanning

Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>
This commit is contained in:
Anna Schumaker 2021-08-25 10:46:57 -04:00
parent 065c192714
commit 91d0922518
2 changed files with 12 additions and 0 deletions

View File

@ -1,5 +1,10 @@
# Copyright 2021 (c) Anna Schumaker.
import db
from gi.repository import GObject
class Task(GObject.GObject):
def run_task(self): raise NotImplementedError
class CommitTask(Task):
def run_task(self): db.commit()

View File

@ -9,3 +9,10 @@ class TestScannerTask(unittest.TestCase):
self.assertIsInstance(t, GObject.GObject)
with self.assertRaises(NotImplementedError):
t.run_task()
class TestScannerCommitTask(unittest.TestCase):
def test_scanner_commit_task(self):
ct = task.CommitTask()
self.assertIsInstance(ct, task.Task)
self.assertEqual(ct.run_task(), None)