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

推荐订阅源

钛媒体:引领未来商业与生活新知
钛媒体:引领未来商业与生活新知
Hugging Face - Blog
Hugging Face - Blog
奇客Solidot–传递最新科技情报
奇客Solidot–传递最新科技情报
人人都是产品经理
人人都是产品经理
Microsoft Azure Blog
Microsoft Azure Blog
Engineering at Meta
Engineering at Meta
B
Blog RSS Feed
大猫的无限游戏
大猫的无限游戏
博客园_首页
雷峰网
雷峰网
V
Visual Studio Blog
爱范儿
爱范儿
A
About on SuperTechFans
量子位
N
Netflix TechBlog - Medium
Microsoft Security Blog
Microsoft Security Blog
让小产品的独立变现更简单 - ezindie.com
让小产品的独立变现更简单 - ezindie.com
MongoDB | Blog
MongoDB | Blog
U
Unit 42
Cyber Security Advisories - MS-ISAC
Cyber Security Advisories - MS-ISAC
月光博客
月光博客
S
SegmentFault 最新的问题
J
Java Code Geeks
H
Hackread – Cybersecurity News, Data Breaches, AI and More

郑文峰的博客

使用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