Django中的單元測試以及Python單元測試

Python單元測試

　　是用來對一個模塊、一個函數或者一個類進行正確性檢驗的測試工作。

　　在Python中unittest是它內置的單元測試框架，單元測試與功能測試都是日常開發中必不可少的部分。

　　比如對函數abs(),我們可以編寫出一下幾個測試用例：

• 　輸入正數，比如1，1.2，0.99，我們期待返回值與輸入相同
• 　輸入負數，比如-1，-1.2，-0.99，我們期待返回值與輸入值相反
• 　輸入0，我們期待返回0
• 　輸入非數值類型，比如None,[],{},我們期待拋出TypeError

　　　　把上面的測試用例放到一個測試模塊里，就是一個完整的單元測試。

一個簡單的測試用例

　　定義一個類，簡單的實現add、sub兩方法，並對其進行單元測試

　　待測試文件m1.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-


class MyClass(object):
    def __init__(self, x, y):
        self.x = int(x)
        self.y = int(y)

    def add(self):
        return self.x + self.y

    def sub(self):
        return self.x - self.y

　　在m1.py同級目錄下創建test.py測試文件，使用unittest單元測試框架對MyClass的方法進行測試

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import unittest
from m1 import MyClass


class MyclassTest(unittest.TestCase):
    def setUp(self):
        self.calc = MyClass(7, 5)

    def tearDown(self):
        pass

    def test_add(self):
        ret = self.calc.add()
        self.assertEqual(ret, 12)

    def test_sub(self):
        ret = self.calc.sub()
        self.assertEqual(ret, 2)


# if __name__ == '__main__':
#     suite = unittest.TestSuite()
#     suite.addTest(MyclassTest('test_add'))
#     suite.addTest(MyclassTest('test_sub'))
#
#     runner = unittest.TextTestRunner()
#     runner.run(suite)

if __name__ == '__main__':
    unittest.main()

 運行測試

python -m unittest test
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK

總結：

　　編寫單元測試時，我們需要編寫一個測試類，從unittest.TestCase繼承，unittest.TestCase提供了很多內置的條件判斷，我們只需要調用這些方法就可以斷言輸出是否是我們所期望的。最常用的斷言就是assertEqual()方法

運行單元測試：　

一旦編寫好單元測試，我們就可以運行單元測試。最簡單的運行方式是在test.py的最后加上兩行代碼：
if __name__ == '__main__':
    unittest.main()
這樣就可以把test.py當做正常的python腳本運行：
$ python test.py
另一種方法是在命令行通過參數-m unittest直接運行單元測試：
$ python -m unittest test

 unittest框架小知識點梳理

　　1. test fixture:是初始化和清理測試數據及環境，通過重寫TestCase的setUp()和tearDown()方法實現

　　　　兩方法使用：設想你的測試需要啟動一個數據庫，這時，就可以在setUp()方法中連接數據庫，在tearDown()方法中關閉數據庫，這樣，不必在每個測試方法中重復相同的代碼：　　　　

class MyclassTest(unittest.TestCase):

    def setUp(self):
        print('setUp...')

    def tearDown(self):
        print('tearDown...')

　　2.test case:是測試用例

　　3.test suite:是測試用例的集合，通過addTest加載TestCase到TestSuite中，返回一個TestSuite實例

　　4.test runner:是運行測試用例並返回結果，通過TextTestRunner類提供的run()方法來執行test suite或test case　　

 注意：

　　單元測試通過了並不意味着程序就沒有bug了，但是不通過程序肯定有bug

Django中的單元測試

　　在創建app的時候，在app目錄下自動生成tests.py文件

　　Model部分單元測試用例　　

from django.db import models

# Create your models here.


class Book(models.Model):
    title = models.CharField(max_length=32)
    price = models.DecimalField(max_digits=10, decimal_places=2)

　　測試用例代碼　　

from django.test import TestCase
from app01.models import Book
# Create your tests here.


class BookModelTest(TestCase):
    def setUp(self):
        Book.objects.create(title='斗破蒼穹', price=10.99)

    def test_book_model(self):
        from decimal import Decimal
        result = Book.objects.get(title='斗破蒼穹')
        self.assertEqual(result.price, Decimal('10.99'))

運行測試在項目目錄下運行：

python manage.py test
Creating test database for alias 'default'...
System check identified no issues (0 silenced).
.
----------------------------------------------------------------------
Ran 1 test in 0.002s

OK
Destroying test database for alias 'default'...

視圖部分單元測試用例　　

from django.shortcuts import render

# Create your views here.


def index(request):
    return render(request, 'index.html') 

 在app01/tests.py文件中添加測試用例代碼：

class IndexPageTest(TestCase):
    """測試index頁面"""
    def test_index_page_renders_index_template(self):
        """測試index視圖"""
        response = self.client.get('/index/')
        self.assertEqual(response.status_code, 200)  # 判斷狀態碼
        self.assertTemplateUsed(response, 'index.html')  # 判斷渲染的模板是否正確

運行單元測試：

python manage.py test
Creating test database for alias 'default'...
System check identified no issues (0 silenced).
..
----------------------------------------------------------------------
Ran 2 tests in 0.044s

OK
Destroying test database for alias 'default'...