如何在django中初始化测试数据库?

时间:2015-09-21 03:56:09

标签: python django django-testing django-tests

我正在为用django编写的项目编写一个测试用例,它给出了一个看起来像{u'message': u'', u'result': {u'username': u'john', u'user_fname': u'', u'user_lname': u'', u'cur_time': 1442808291000.0, u'dofb': None, u'sex': u'M', u'u_email': u'', u'role': u'', u'session_key': u'xxhhxhhhx', u'mobile': None}, u'error': 0} 的意外输出 在这里我们可以看到其他字段是空的,因为我刚刚在测试用例中创建了用户,但没有给出其他信息。数据库是从生产数据库创建的,但未初始化,它仍为空。 这就是为什么它让其他字段为空。它正在查询空数据库。

我为登录REST API 编写了以下测试用例。并通过 python manage.py test 运行它。请告诉我如何解决上述问题。

注意:如果以下方法不正确,您可以建议其他方法。

from django.test import TestCase
from django.test.client import Client
from django.contrib.auth.models import User
import json

class TestAPI(TestCase):

      def setUp(self):
            self.c=Client() #Create Client object that simulates request to a url similar to a browser can
            User.objects.create_user(username="john", password="xxx")

      def test_login_api(self):
            credential_test=dict()
            c_test =Client()

            credential_test["username"]="john"
            credential_test["password"]="xxx"
            data=json.dumps(credential_test)
            #print 'data is'
            #print data
            response_test =c_test.put('/api/login', data)
            content_test=json.loads(response_test.content)
            print 'content'

2 个答案:

答案 0 :(得分:1)

尝试更改它:

User.objects.create(username="john", password="xxx")

为:

User.objects.create_user(username='john', password='xxx')

方法create_user使用set_password方法。

class UserManager(models.Manager):
    # ...   
    def create_user(self, username, email=None, password=None):
        """
        Creates and saves a User with the given username, email and password.
        """
        now = timezone.now()
        if not username:
            raise ValueError('The given username must be set')
        email = UserManager.normalize_email(email)
        user = self.model(username=username, email=email,
                          is_staff=False, is_active=True, is_superuser=False,
                          last_login=now, date_joined=now)

        user.set_password(password)
        user.save(using=self._db)
        return user

答案 1 :(得分:1)

两种方法:

  1. 扩展setUp()的使用,为其他模型创建记录,并在您创建的模型之间建立一组有效的关系。它是配置通过代码的方法。
  2. 使用灯具预填充您的测试dbs。如果你做一些研究,你可以找到如何使用现有的有效数据库创建一些灯具。但是,我建议您清理用于测试的任何生产数据。 Aka是一种通过数据配置的方法。