excel.tests.test_tasks

Complexity

Total Complexity

4

Size/Duplication

Total Lines	94
Duplicated Lines	75.53 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	ImportTemplateTaskTest.test_import_from_template()	13	13	1
A	ImportTemplateTaskTest.test_import_from_template_errors()	13	13	1
A	ImportTemplateTaskTest.setUp()	6	6	1
A	ImportTemplateTaskTest.test_on_failure()	35	35	1

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Jul  9 11:51:09 2019

@author: Paolo Cozzi <cozzi@ibba.cnr.it>
"""

from unittest.mock import patch
from billiard.einfo import ExceptionInfo

from django.core import mail
from django.test import TestCase

from common.constants import ERROR
from common.tests import WebSocketMixin
from image_app.models import Submission

from .common import BaseExcelMixin
from ..tasks import ImportTemplateTask


class ImportTemplateTaskTest(WebSocketMixin, BaseExcelMixin, TestCase):

This code seems to be duplicated in your project.

    def setUp(self):
        # calling my base class setup
        super().setUp()

        # setting task
        self.my_task = ImportTemplateTask()

    def test_on_failure(self):
        """Testing on failure methods"""

        exc = Exception("Test")
        task_id = "test_task_id"
        args = [self.submission_id]
        kwargs = {}
        einfo = ExceptionInfo

        # call on_failure method
        self.my_task.on_failure(exc, task_id, args, kwargs, einfo)

        # check submission status and message
        submission = Submission.objects.get(pk=self.submission_id)

        # check submission.state changed
        self.assertEqual(submission.status, ERROR)
        self.assertEqual(
            submission.message,
            "Error in Template loading: Test")

        # test email sent
        self.assertEqual(len(mail.outbox), 1)

        # read email
        email = mail.outbox[0]

        self.assertEqual(
            "Error in Template loading: %s" % self.submission_id,
            email.subject)

        message = 'Error'
        notification_message = 'Error in Template loading: Test'

        self.check_message(message, notification_message)

    @patch("excel.tasks.upload_template", return_value=True)
    def test_import_from_template(self, my_upload):
        """Testing template import"""

        # NOTE that I'm calling the function directly, without delay
        # (AsyncResult). I've patched the time consuming task
        res = self.my_task.run(submission_id=1)

        # assert a success with data uploading
        self.assertEqual(res, "success")

        # assert that method were called
        self.assertTrue(my_upload.called)

    @patch("excel.tasks.upload_template", return_value=False)
    def test_import_from_template_errors(self, my_upload):
        """Testing template import with errors"""

        # NOTE that I'm calling the function directly, without delay
        # (AsyncResult). I've patched the time consuming task
        res = self.my_task.run(submission_id=1)

        # assert a success with data uploading
        self.assertEqual(res, "Error in uploading Template data")

        # assert that method were called
        self.assertTrue(my_upload.called)