Merge pull request #242 from stumpylog/resolve-test-warnings

Reduce warnings in test suite
This commit is contained in:
Quinn Casey 2022-03-11 05:54:19 -08:00 committed by GitHub
commit c19dd81ecf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 24 additions and 14 deletions

View File

@ -59,8 +59,10 @@ class Command(BaseCommand):
old_paths = [document.source_path, document.thumbnail_path] old_paths = [document.source_path, document.thumbnail_path]
raw_document = GnuPG.decrypted(document.source_file, passphrase) with document.source_file as file_handle:
raw_thumb = GnuPG.decrypted(document.thumbnail_file, passphrase) raw_document = GnuPG.decrypted(file_handle, passphrase)
with document.thumbnail_file as file_handle:
raw_thumb = GnuPG.decrypted(file_handle, passphrase)
document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED

View File

@ -199,19 +199,22 @@ class Command(BaseCommand):
os.makedirs(os.path.dirname(original_target), exist_ok=True) os.makedirs(os.path.dirname(original_target), exist_ok=True)
with open(original_target, "wb") as f: with open(original_target, "wb") as f:
f.write(GnuPG.decrypted(document.source_file)) with document.source_file as out_file:
os.utime(original_target, times=(t, t)) f.write(GnuPG.decrypted(out_file))
os.utime(original_target, times=(t, t))
os.makedirs(os.path.dirname(thumbnail_target), exist_ok=True) os.makedirs(os.path.dirname(thumbnail_target), exist_ok=True)
with open(thumbnail_target, "wb") as f: with open(thumbnail_target, "wb") as f:
f.write(GnuPG.decrypted(document.thumbnail_file)) with document.thumbnail_file as out_file:
os.utime(thumbnail_target, times=(t, t)) f.write(GnuPG.decrypted(out_file))
os.utime(thumbnail_target, times=(t, t))
if archive_target: if archive_target:
os.makedirs(os.path.dirname(archive_target), exist_ok=True) os.makedirs(os.path.dirname(archive_target), exist_ok=True)
with open(archive_target, "wb") as f: with open(archive_target, "wb") as f:
f.write(GnuPG.decrypted(document.archive_path)) with document.archive_path as out_file:
os.utime(archive_target, times=(t, t)) f.write(GnuPG.decrypted(out_file))
os.utime(archive_target, times=(t, t))
else: else:
self.check_and_copy( self.check_and_copy(
document.source_path, document.checksum, original_target document.source_path, document.checksum, original_target

View File

@ -61,6 +61,6 @@ class TestDocumentAdmin(DirectoriesMixin, TestCase):
def test_created(self): def test_created(self):
doc = Document.objects.create( doc = Document.objects.create(
title="test", created=timezone.datetime(2020, 4, 12) title="test", created=timezone.make_aware(timezone.datetime(2020, 4, 12))
) )
self.assertEqual(self.doc_admin.created_(doc), "2020-04-12") self.assertEqual(self.doc_admin.created_(doc), "2020-04-12")

View File

@ -10,6 +10,7 @@ from unittest import mock
import pytest import pytest
from django.conf import settings from django.conf import settings
from django.contrib.auth.models import User from django.contrib.auth.models import User
from django.utils import timezone
from django.test import override_settings from django.test import override_settings
from rest_framework.test import APITestCase from rest_framework.test import APITestCase
from whoosh.writing import AsyncWriter from whoosh.writing import AsyncWriter
@ -567,11 +568,15 @@ class TestDocumentApi(DirectoriesMixin, APITestCase):
d3.tags.add(t) d3.tags.add(t)
d3.tags.add(t2) d3.tags.add(t2)
d4 = Document.objects.create( d4 = Document.objects.create(
checksum="4", created=datetime.datetime(2020, 7, 13), content="test" checksum="4",
created=timezone.datetime(2020, 7, 13),
content="test",
) )
d4.tags.add(t2) d4.tags.add(t2)
d5 = Document.objects.create( d5 = Document.objects.create(
checksum="5", added=datetime.datetime(2020, 7, 13), content="test" checksum="5",
added=timezone.datetime(2020, 7, 13),
content="test",
) )
d6 = Document.objects.create(checksum="6", content="test2") d6 = Document.objects.create(checksum="6", content="test2")

View File

@ -162,7 +162,7 @@ class TestMatching(TestCase):
def test_match_regex(self): def test_match_regex(self):
self._test_matching( self._test_matching(
"alpha\w+gamma", r"alpha\w+gamma",
"MATCH_REGEX", "MATCH_REGEX",
( (
"I have alpha_and_gamma in me", "I have alpha_and_gamma in me",
@ -181,7 +181,7 @@ class TestMatching(TestCase):
) )
def test_tach_invalid_regex(self): def test_tach_invalid_regex(self):
self._test_matching("[[", "MATCH_REGEX", [], ["Don't match this"]) self._test_matching("[", "MATCH_REGEX", [], ["Don't match this"])
def test_match_fuzzy(self): def test_match_fuzzy(self):

View File

@ -121,7 +121,7 @@ class RasterisedDocumentParser(DocumentParser):
# TODO catch all for various issues with PDFminer.six. # TODO catch all for various issues with PDFminer.six.
# If PDFminer fails, fall back to OCR. # If PDFminer fails, fall back to OCR.
self.log( self.log(
"warn", "warning",
"Error while getting text from PDF document with " "pdfminer.six", "Error while getting text from PDF document with " "pdfminer.six",
exc_info=True, exc_info=True,
) )