惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

D
Docker
爱范儿
爱范儿
Vercel News
Vercel News
OSCHINA 社区最新新闻
OSCHINA 社区最新新闻
让小产品的独立变现更简单 - ezindie.com
让小产品的独立变现更简单 - ezindie.com
Martin Fowler
Martin Fowler
博客园 - 司徒正美
大猫的无限游戏
大猫的无限游戏
酷 壳 – CoolShell
酷 壳 – CoolShell
T
Tailwind CSS Blog
Blog — PlanetScale
Blog — PlanetScale
博客园 - 【当耐特】
人人都是产品经理
人人都是产品经理
F
Fortinet All Blogs
freeCodeCamp Programming Tutorials: Python, JavaScript, Git & More
博客园 - Franky
T
The Blog of Author Tim Ferriss
Y
Y Combinator Blog
Recent Announcements
Recent Announcements
奇客Solidot–传递最新科技情报
奇客Solidot–传递最新科技情报
Google DeepMind News
Google DeepMind News
H
Help Net Security
钛媒体:引领未来商业与生活新知
钛媒体:引领未来商业与生活新知
G
Google Developers Blog

郑文峰的博客

使用dify对接飞书多维表格 使用n8n对接飞书多维表格 服务启动时出现 OOM 一次服务升级时pg表DDL执行超时失败 Go语言高效IO缓冲技术详解 Go语言延迟初始化(Lazy Initialization)最佳实践 Go语言字符串拼接性能对比与优化指南 Go语言结构体内存对齐完全指南 Go语言空结构体:零内存消耗的高效编程 Go语言堆栈分配与逃逸分析深度解析 Go语言原子操作完全指南 Go语言内存预分配完全指南 Go语言不可变数据共享:无锁并发编程实践 Go语言零拷贝技术完全指南 Go语言遍历性能深度解析:从原理到优化实践 Go语言Interface Boxing原理与性能优化指南 Go协程池深度解析:原理、实现与最佳实践 使用etcd分布式锁导致的协程泄露与死锁问题 基于pre-commit的Python代码规范落地实践 初识 MCP Server pulsar阻塞导致logstash无法接入日志 django-prometheus使用及源码分析 kube-proxy源码分析 kubernetes service如何通过iptables转发 tcp缓存引起的日志丢失 django-apschedule定时任务异常停止 理解calico容器网络通信方案原理 理解flannel的三种容器网络方案原理 理解Linux IPIP隧道 理解VXLAN网络
django rest_framework使用pytest单元测试
zhengwenfeng · 2022-08-10 · via 郑文峰的博客

# 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