python django单元测试文件现场测试模型示例

Posted 2021-05-09

from django.core.files import File
from django.test import TestCase

import mock


class FooModelTestCase(TestCase):
    def test_bar_field(self):
        file_mock = mock.MagicMock(spec=File)
        file_mock.name = 'test.pdf'
        foo = FooModel(bar=file_mock)
        self.assertEqual(foo.bar.name, file_mock.name)