运行开发服务器,在浏览器上自己点击测试,看写的代码是否正常,但是这样做很麻烦,因为以后如果有改动,可能会影响以前本来正常的功能,这样以前的功能又得测试一遍,非常不方便,Django中有完善的单元测试,我们可以对开发的每一个功能进行单元测试,这样只要运行一个命令 python manage.py test,就可以测试功能是否正常。
一言以蔽之,测试就是检查代码是否按照自己的预期那样运行。
测试驱动开发: 有时候,我们知道自己需要的功能(结果),并不知道代码如何书写,这时候就可以利用测试驱动开发(Test Driven Development),先写出我们期待得到的结果(把测试代码先写出来),再去完善代码,直到不报错,我们就完成了。
《改善Python的91个建议》一书中说:单元测试绝不是浪费时间的无用功,它是高质量代码的保障之一,在软件开发的一节中值得投入精力和时间去把好这一关。
1. Python 中 单元测试简介:
下面是一个 Python的单元测试简单的例子:
假如我们开发一个除法的功能,有的同学可能觉得很简单,代码是这样的:
1
2
|
def
division_funtion(x, y):
return
x
/
y
|
但是这样写究竟对还是不对呢,有些同学可以在代码下面这样测试:
1
2
3
4
5
6
7
8
|
def
division_funtion(x, y):
return
x
/
y
if
__name__
=
=
'__main__'
:
print
division_funtion(
2
,
1
)
print
division_funtion(
2
,
4
)
print
division_funtion(
8
,
3
)
|
但是这样运行后得到的结果,自己每次都得算一下去核对一遍,很不方便,Python中有 unittest 模块,可以很方便地进行测试,详情可以文章最后的链接,看官网文档的详细介绍。
下面是一个简单的示例:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
import
unittest
def
division_funtion(x, y):
return
x
/
y
class
TestDivision(unittest.TestCase):
def
test_int(
self
):
self
.assertEqual(division_funtion(
9
,
3
),
3
)
def
test_int2(
self
):
self
.assertEqual(division_funtion(
9
,
4
),
2.25
)
def
test_float(
self
):
self
.assertEqual(division_funtion(
4.2
,
3
),
1.4
)
if
__name__
=
=
'__main__'
:
unittest.main()
|
我简单地写了三个测试示例(不一定全面,只是示范,比如没有考虑除数是0的情况),运行后发现:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
F.F
======================================================================
FAIL: test_float (__main__.TestDivision)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/Users/tu/YunPan/mydivision.py", line 16, in test_float
self.assertEqual(division_funtion(4.2, 3), 1.4)
AssertionError: 1.4000000000000001 != 1.4
======================================================================
FAIL: test_int2 (__main__.TestDivision)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/Users/tu/YunPan/1.py", line 13, in test_int2
self.assertEqual(division_funtion(9, 4), 2.25)
AssertionError: 2 != 2.25
----------------------------------------------------------------------
Ran 3 tests in 0.001s
FAILED (failures=2)
|
汗!发现了没,竟然两个都失败了,测试发现:
4.2除以3 等于 1.4000000000000001 不等于期望值 1.4
9除以4等于2,不等于期望的 2.25
下面我们就是要修复这些问题,再次运行测试,直到运行不报错为止。
譬如根据实际情况,假设我们只需要保留到小数点后6位,可以这样改:
1
2
|
def
division_funtion(x, y):
return
round
(
float
(x)
/
y,
6
)
|
再次运行就不报错了:
1
2
3
4
5
|
...
----------------------------------------------------------------------
Ran 3 tests in 0.000s
OK
|
Python 单元测试 官方文档:
Python 2 (https://docs.python.org/2/library/unittest.html)
Python 3 (https://docs.python.org/3/library/unittest.html)
2. Django 中 单元测试:(不断完善中,后期会增加对前面讲解的内容的测试)
2.1 简单测试例子:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
from
django.test
import
TestCase
from
myapp.models
import
Animal
class
AnimalTestCase(TestCase):
def
setUp(
self
):
Animal.objects.create(name
=
"lion"
, sound
=
"roar"
)
Animal.objects.create(name
=
"cat"
, sound
=
"meow"
)
def
test_animals_can_speak(
self
):
"""Animals that can speak are correctly identified"""
lion
=
Animal.objects.get(name
=
"lion"
)
cat
=
Animal.objects.get(name
=
"cat"
)
self
.assertEqual(lion.speak(),
'The lion says "roar"'
)
self
.assertEqual(cat.speak(),
'The cat says "meow"'
)
|
这个例子是测试myapp.models 中的 Animal 类相关的方法功能。
2.2 用代码访问网址的方法:
1
2
3
4
5
6
7
8
|
>>>
from
django.test
import
Client
>>> c
=
Client()
>>> response
=
c.post(
'/login/'
, {
'username'
:
'john'
,
'password'
:
'smith'
})
>>> response.status_code
200
>>> response
=
c.get(
'/customer/details/'
)
>>> response.content
'<!DOCTYPE html...'
|
我们可以用 django.test.Client 的实例来实现 get 或 post 内容,检查一个网址返回的网页源代码。
默认情况下CSRF检查是被禁用的,如果测试需要,可以用下面的方法:
1
2
|
>>>
from
django.test
import
Client
>>> csrf_client
=
Client(enforce_csrf_checks
=
True
)
|
使用 csrf_client 这个实例进行请求即可。
指定浏览USER-AGENT:
1
|
>>> c
=
Client(HTTP_USER_AGENT
=
'Mozilla/5.0'
)
|
模拟post上传附件:
1
2
3
4
5
|
from
django.test
import
Client
c
=
Client()
with
open
(
'wishlist.doc'
) as fp:
c.post(
'/customers/wishes/'
, {
'name'
:
'fred'
,
'attachment'
: fp})
|
测试网页返回状态:
1
2
3
4
5
6
7
8
9
10
|
from
django.test
import
TestCase
class
SimpleTest(TestCase):
def
test_details(
self
):
response
=
self
.client.get(
'/customer/details/'
)
self
.assertEqual(response.status_code,
200
)
def
test_index(
self
):
response
=
self
.client.get(
'/customer/index/'
)
self
.assertEqual(response.status_code,
200
)
|
我们用 self.client 即可,不用 client = Client() 这样实例化,更方便,我们还可以继承 Client,添加一些其它方法:
1
2
3
4
5
6
7
8
9
10
11
12
|
from
django.test
import
TestCase, Client
class
MyTestClient(Client):
# Specialized methods for your environment
...
class
MyTest(TestCase):
client_class
=
MyTestClient
def
test_my_stuff(
self
):
# Here self.client is an instance of MyTestClient...
call_some_test_code()
|
定制 self.client 的方法:
1
2
3
4
5
6
7
8
9
10
11
|
from
django.test
import
Client, TestCase
class
MyAppTests(TestCase):
def
setUp(
self
):
super
(MyAppTests,
self
).setUp()
self
.client
=
Client(enforce_csrf_checks
=
True
)
def
test_home(
self
):
response
=
self
.client.get(
'/'
)
self
.assertEqual(response.status_code,
200
)
|