scanner: Add a CommitTask
For committing the database during scanning Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>
This commit is contained in:
parent
065c192714
commit
91d0922518
|
@ -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()
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue