From eeceb52c0614ee08928c1424385ee6753183d494 Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Wed, 26 Dec 2018 06:09:48 -0400 Subject: [PATCH] Fix failing source tests Signed-off-by: Roberto Rosario --- mayan/apps/sources/tests/test_models.py | 5 ++--- mayan/apps/sources/tests/test_views.py | 13 +++---------- 2 files changed, 5 insertions(+), 13 deletions(-) diff --git a/mayan/apps/sources/tests/test_models.py b/mayan/apps/sources/tests/test_models.py index d7bc05fb15..97794ecb19 100644 --- a/mayan/apps/sources/tests/test_models.py +++ b/mayan/apps/sources/tests/test_models.py @@ -19,9 +19,8 @@ from mayan.apps.documents.tests import ( from mayan.apps.metadata.models import MetadataType from ..literals import SOURCE_UNCOMPRESS_CHOICE_Y -from ..models import ( - EmailBaseModel, POP3Email, WatchFolderSource, WebFormSource -) +from ..models import POP3Email, WatchFolderSource, WebFormSource +from ..models.email_sources import EmailBaseModel from .literals import ( TEST_EMAIL_ATTACHMENT_AND_INLINE, TEST_EMAIL_BASE64_FILENAME, diff --git a/mayan/apps/sources/tests/test_views.py b/mayan/apps/sources/tests/test_views.py index 6ea19509f3..45989d89da 100644 --- a/mayan/apps/sources/tests/test_views.py +++ b/mayan/apps/sources/tests/test_views.py @@ -80,14 +80,7 @@ class DocumentUploadTestCase(GenericDocumentViewTestCase): obj=self.document_type, permission=permission_document_create ) - with open(TEST_SMALL_DOCUMENT_PATH, mode='rb') as file_object: - response = self.post( - viewname='sources:upload_interactive', args=(self.source.pk,), - data={ - 'source-file': file_object, - 'document_type_id': self.document_type.pk, - }, follow=True - ) + response = self._request_upload_wizard_view() self.assertTrue(b'queued' in response.content) self.assertEqual(Document.objects.count(), 1) @@ -254,9 +247,9 @@ class StagingFolderViewTestCase(GenericViewTestCase): self.assertEqual(len(list(self.staging_folder.get_files())), 0) -class SourcesTestCase(GenericDocumentViewTestCase): +class SourcesViewsTestCase(GenericViewTestCase): def setUp(self): - super(SourcesTestCase, self).setUp() + super(SourcesViewsTestCase, self).setUp() self.login_user() def _create_web_source(self):