郑文峰的博客 郑文峰的博客
首页
  • python之路
  • go之路
  • 其他
  • redis
  • mysql
  • docker
  • k8s
读书破万卷
周刊
关于
  • 导航 (opens new window)
  • 代码片段 (opens new window)
  • 收藏
  • 友链
  • 外部页面

    • 开往 (opens new window)
  • 索引

    • 分类
    • 标签
    • 归档
GitHub (opens new window)

zhengwenfeng

穷则变,变则通,通则久
首页
  • python之路
  • go之路
  • 其他
  • redis
  • mysql
  • docker
  • k8s
读书破万卷
周刊
关于
  • 导航 (opens new window)
  • 代码片段 (opens new window)
  • 收藏
  • 友链
  • 外部页面

    • 开往 (opens new window)
  • 索引

    • 分类
    • 标签
    • 归档
GitHub (opens new window)
  • python

    • 基础

    • 第三方库

    • django

      • django celery 结合使用
      • django rest_framework使用jwt
      • django rest_framework Authentication
      • django rest_framework异常处理
      • django rest_framework 自定义文档
      • django压缩文件下载
      • django rest_framework使用pytest单元测试
        • djang自带测试
        • rest framework
      • django restframework choice 自定义输出数据
      • django Filtering 使用
      • django viewset 和 Router 配合使用时报的错
      • django model的序列化
      • django中使用AbStractUser
      • django.core.exceptions.ImproperlyConfigured Application labels aren't unique, duplicates users
      • django 中 media配置
      • django 外键引用自身和on_delete参数
      • django 警告 while time zone support is active
      • django rest_framework 分页
    • flask

    • tornado

    • 其他

  • go

  • 其他

  • 编程
  • python
  • django
zhengwenfeng
2022-08-10
目录

django rest_framework使用pytest单元测试

# djang自带测试

django本身自带了测试框架库,是基于unittest的。

执行 python manager.py test 会对路径所有test*.py 进行测试

from django.test import TestCase
from event_track.models.app import Appclass 

AppTestCase(TestCase):    
    def setUp(self):        
        App.objects.create(name="app1", package_name="package1")        
        App.objects.create(name="app2", package_name="package2")    
    def test_app(self):        
        app1 = App.objects.get(name="app1")        
        self.assertEqual(app1.package_name, "package1")        
        app1 = App.objects.get(name="app2")        
        self.assertEqual(app1.package_name, "package3")

1
2
3
4
5
6
7
8
9
10
11
12
13

# rest framework

下面是一个简单的测试案例。使用pytest对rest framework进行测试

1. 添加一个配置文件 具体看pytest-django官网 (opens new window)

[pytest]
DJANGO_SETTINGS_MODULE=event_track_root.settings
python_files = tests.py test_*.py *_tests.py
1
2
3

2. 创建一个model app.py

from django.db import models
class App(models.Model):    
    name = models.CharField(max_length=24)    
    package_name = models.CharField(max_length=50, unique=True)
1
2
3
4

对app的model类进行增删改查的测试 model测试必须添加@pytest.mark.django_db才可以启用数据库。 使用APITestCase对接口进行测试

3. 编写测试用例 test_app.py


@pytest.mark.django_db
@pytest.fixture(scope="module")
def init_app_data():
    App.objects.create(name="app1", package_name="package1")
    App.objects.create(name="app2", package_name="package2")
    App.objects.create(name="app3", package_name="package3")
    App.objects.create(name="app4", package_name="package4")

class AppTests(APITestCase):

    def test_create_app(self):
        url = reverse('event_track:App-list')
        data_list = [{"name": "app1", "package_name": "package1"},
                     {"name": "app2", "package_name": "package2"},
                     {"name": "app3", "package_name": "package3"},
                     {"name": "app4", "package_name": "package4"}
                     ]

        for data in data_list:
            response = self.client.post(url, data)
            self.assertEqual(response.status_code, status.HTTP_201_CREATED)

        self.assertEqual(App.objects.count(), 4)
        self.assertEqual(App.objects.get(name="app1").package_name, "package1")

    @pytest.mark.usefixtures('init_app_data')
    def test_delete_app(self):
        app = App.objects.get(package_name="package3")
        url = reverse('event_track:App-detail', [app.id])
        response = self.client.delete(url)

        self.assertEqual(App.objects.count(), 3)
        with pytest.raises(App.DoesNotExist):
            App.objects.get(package_name="package3")
        self.assertEqual(response.status_code, 204)

    @pytest.mark.usefixtures('init_app_data')
    def test_update_app(self):
        app = App.objects.get(name="app4")
        url = reverse('event_track:App-detail', [app.id])

        app.package_name = "package_update"
        response = self.client.put(url, AppSerializer(app).data)

        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.data['package_name'], 'package_update')
        self.assertEqual(App.objects.get(name="app4").package_name, 'package_update')

    @pytest.mark.usefixtures('init_app_data')
    def test_list_app(self):
        url = reverse('event_track:App-list')
        response = self.client.get(url, {'limit': 2, 'offset': 2})

        self.assertEqual(response.status_code, 200)
        self.assertEqual(len(response.data['results']), 2)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#python#django
上次更新: 2023/01/15, 15:47:48
django压缩文件下载
django restframework choice 自定义输出数据

← django压缩文件下载 django restframework choice 自定义输出数据→

最近更新
01
django rest_framework 分页
03-20
02
学习周刊-第03期-第09周
03-03
03
学习周刊-第02期-第08周
02-24
更多文章>
Theme by Vdoing | Copyright © 2022-2023 zhengwenfeng | MIT License
  • 跟随系统
  • 浅色模式
  • 深色模式
  • 阅读模式