Тесты django для проверки подлинности - почему эта исправленная функция все еще вызывается? - PullRequest
Новая
       27

Тесты django для проверки подлинности - почему эта исправленная функция все еще вызывается?

0 голосов
/ 22 апреля 2020

Я пытаюсь протестировать функцию в одной из моих моделей и пытаюсь смоделировать файловую систему, используя mock.patch. Неважно, что я пытаюсь, похоже, он не перехватывает метод.

Модель для тестирования:

app / models.py 

from django.db import models
from .utils.storageutils import get_file
from .utils.datautils import derive_data

Class DataThing(models.Model):
    #defined here

    def set_data_from_file(self):
        data = derive_data(get_file('filepath'))
        setattr(self, 'derived_data', data)
        self.save()

app / utils / datautils.py 

import pandas as pd

def derive_data(data_from_file):
    df = pd.DataFrame('...')
    #do stuff with dataframe
    return df

app / tests / unit_tests / tests_models.py 

from django.test import TestCase
import mock
from app.models import DataThing

class DataThingModelTest(TestCase):
    @classmethod
    def setUpTestData(cls):
      cls.datathing = DataThing

    @mock.patch('app.models.derive_data')
    def test_set_data_from_file(self, mocked_derive_data):
        mocked_derive_data.return_value=('pretend_dataframe')
        self.datathing.set_data_from_file()
        self.assertEquals(self.datathing.derived_data, 'pretend_dataframe')

Я бы ожидайте, что это пройдет. Тем не менее, я получаю сообщение об ошибке, потому что datathing.set_data_from_file() все равно в конечном итоге вызывает utils.storageutils.get_file. Я попытался внести исправления в app.utils.datautils.derive_data, но у меня возникла та же проблема.

1 Ответ

0 голосов
/ 22 апреля 2020

Мне также нужно было исправить функцию get_file 

@mock.patch('app.models.get_file')
@mock.patch('app.models.derive_data')
def test_set_data_from_file(self, mocked_derive_data, mocked_ger_file):
    mocked_get_file.return_value=('placeholder')
    mocked_derive_data.return_value=('pretend_dataframe')
    self.datathing.set_data_from_file()
    self.assertEquals(self.datathing.derived_data, 'pretend_dataframe')
...