如何在单元测试中修复错误? | Asse田

时间:2017-09-12 04:02:55

标签: python django python-2.7 unit-testing django-1.11

我正在尝试将unit-test写入Django项目中的create function。这是我创建单元测试的第一次经历。为什么我无法创建新数据?从错误中我了解到测试数据库中没有文章。我错在哪里?

tests.py:

class ArticleViewTestCase(TestCase):
def setUp(self):
    self.user = User.objects.create(
        username='user',
        email='user@gmail.com',
        password='password'
    )
    self.client = Client()

def test_article_create(self):
    self.assertTrue(self.user)
    self.client.login(username='user', password='password')
    response = self.client.get('/article/create/')
    self.assertEquals(response.status_code, 200)

    with open('/home/nurzhan/Downloads/planet.jpg', 'rb') as image:
        imageStringIO = StringIO(image.read()) <-- ERROR HERE

    response = self.client.post(
        '/article/create/',
        content_type='multipart/form-data',
        data={
            'head': 'TEST',
            'opt_head': 'TEST',
            'body': 'TEST',
            'location': 1,
            'public': True,
            'idx': 0,
            'image': (imageStringIO, 'image.jpg')
        },
        follow=True
    )
    self.assertEqual(response.status_code, 200)
    self.assertEqual(Article.objects.all().count(), 1)

错误:

Traceback (most recent call last):
  File "/home/nurzhan/CA/article/tests.py", line 26, in test_article_create
    imageStringIO = StringIO(image.read())
TypeError: 'module' object is not callable

3 个答案:

答案 0 :(得分:1)

当单元测试时,Django会在每次测试结束时清空数据库 ,因此每个测试都以空数据库开始。测试开始时很可能没有用户存在,这意味着登录失败。您应该将测试更改为以下内容:

from django.contrib.auth.hashers import make_password
from django.contrib.auth.models import User

class ArticleViewTestCase(TestCase):

    def test_article_create(self):
        User.objects.create(username='alice', password=make_password('topsecret'))
        logged_in = self.client.login(username='alice', password="topsecret")
        self.assertTrue(logged_in)
        # Continue your test here

掌握了这一点之后,您最终会希望将用户创建移至setUp()方法。

答案 1 :(得分:1)

您可以覆盖form_invalid并在测试中检查数据

class ArticleCreateView(CreateView):
    # YOUR code here

   def form_invalid(self, form):
        data = {'status': False, 'errors': form.errors}
        return JsonResponse(data, , status=500)

在测试方法中:

with open('/home/nurzhan/Downloads/planet.jpg', 'rb') as image:
    response = self.client.post('/article/create/',
        data={
            'head': 'TEST',
            'opt_head': 'TEST',
            'body': 'TEST',
            'location': 1,
            'public': True,
            'idx': 0,
            'image': image
        },
        follow=True, format='multipart'
     )

答案 2 :(得分:0)

插入测试代码PDB测试工具:

# Below response = self.client.get('/article/create/')
import pdb;pdb.set_trace()

然后检查response.data会有答案给你答案:)

测试代码的一些提示:

如果您不需要检查用户登录数据,请使用 self.client.force_login(self.user)您需要仅提供用户对象及其清晰。 型动物 当你有很多https方法可以测试时,U可以将get和post分成不同的测试方法,但是它是可选的 - 仅建议

请尝试不要使用来自db的对象使用len() Article.objects.all().count()获取多个对象 然后你不需要使用len()

articles_amount = Article.objects.all().count()
self.assertEqual(articles_amount, 1)