Django测试由于AssertionError:201!= 400而失败

时间:2020-06-03 10:30:02

标签: python django django-rest-framework django-tests

获得以下视图

class MyUserCreate(APIView):
    '''
    Creates the user. 
    '''
    def post(self, request, format='json'):
        serializer = MyUserSerializer(data=request.data)
        if serializer.is_valid():
            user = serializer.save()
            if user:
                return Response(serializer.data, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

并希望创建一个测试以确保我无法创建具有相同用户名的用户。

在serializers.py中,MyUserSerializer中的用户名看起来像这样

username = serializers.CharField(
        validators=[UniqueValidator(queryset=User.objects.all())],
        max_length=32
        )

和models.py中,MyUser中的用户名

username = models.CharField(db_column='username', unique=True, max_length=20)

在tests.py

class MyUserTest(APITestCase):

    ...

    def test_create_user_with_preexisting_username(self):
        data = {
                'username': 'testing',
                'password': 'test'
                }

        response = self.client.post(self.create_url, data, format='json')
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertEqual(MyUser.objects.count(), 1)
        self.assertEqual(len(response.data['username']), 1)

至此,对用户创建,没有用户名的用户,用户名过长的用户的测试...都可以正常进行。该问题仅在此特定测试中发生。

用户测试存在于数据库中

Users in the DB

如果我跑步

python manage.py tests

然后

FAIL: test_create_user_with_preexisting_username (codeLabsApp.tests.MyUserTest)
Ensure we can't create a user with preexisting username.
----------------------------------------------------------------------
Traceback (most recent call last):
  ...
AssertionError: 201 != 400

所以我得到了201创建请求,而不是400错误请求。

1 个答案:

答案 0 :(得分:2)

正如您在docs中所看到的,由于测试正在运行,因此没有本地数据库使用情况。
因此,我想,数据库表屏幕快照中的用户存在于本地数据库中。
运行测试时,Django将创建另一个数据库(仅适用于当前测试)。在该数据库中,没有用户testing,除了该用户是通过setUp测试类方法或Fixture创建的。
解决该问题的最简单方法是更改​​测试方法:

def test_create_user_with_preexisting_username(self):
    existed_username = "testing"
    existed_user = MyUser.objects.create(username=existed_username)
    data = {
            'username': existed_username,
            'password': 'test'
            }

    response = self.client.post(self.create_url, data, format='json')
    self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
    self.assertEqual(MyUser.objects.count(), 2)
    self.assertEqual(len(response.data['username']), 1)

此外,当您获得response.data的长度时请多加注意。如果您在某处进行了分页响应,则不会收到所有预期的实例(最大页面尺寸)。