Small improvements to coverage

This commit is contained in:
Trenton H
2023-06-06 10:35:51 -07:00
parent ee19307ea2
commit 45d8c945e2
2 changed files with 47 additions and 3 deletions

View File

@@ -69,7 +69,7 @@ class Consumer(LoggingMixin):
status,
message=None,
document_id=None,
):
): # pragma: no cover
payload = {
"filename": os.path.basename(self.filename) if self.filename else None,
"task_id": self.task_id,
@@ -352,7 +352,7 @@ class Consumer(LoggingMixin):
self.run_pre_consume_script()
def progress_callback(current_progress, max_progress):
def progress_callback(current_progress, max_progress): # pragma: no cover
# recalculate progress to be within 20 and 80
p = int((current_progress / max_progress) * 50 + 20)
self._send_progress(p, 100, "WORKING")