Hướng dẫn hoàn chỉnh Python unittest: từ cơ bản đến nâng cao

1. Python unittest là gì?

Tầm quan trọng của kiểm thử đơn vị

2. Cách sử dụng cơ bản của unittest

Ví dụ kiểm thử cơ bản

import unittest

# Mã cần kiểm thử
def add(a, b):
    return a + b

# Lớp kiểm thử
class TestAddFunction(unittest.TestCase):

    def test_add_integers(self):
        result = add(2, 3)
        self.assertEqual(result, 5)

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

Mở rộng kiểm thử

def test_add_floats(self):
    result = add(2.5, 3.5)
    self.assertAlmostEqual(result, 6.0, places=2)

def test_add_strings(self):
    result = add("Hello, ", "World!")
    self.assertEqual(result, "Hello, World!")

3. Cách sử dụng setUp() và tearDown()

Ví dụ về setUp()

def setUp(self):
    self.temp_value = 42

Ví dụ về tearDown()

def tearDown(self):
    self.temp_value = None

4. Kiểm thử các phụ thuộc bằng mock

Ví dụ về mock

from unittest.mock import patch

class TestAddFunction(unittest.TestCase):

    @patch('my_module.time_consuming_function')
    def test_add_with_mock(self, mock_func):
        mock_func.return_value = 0
        result = add(2, 3)
        self.assertEqual(result, 5)

5. Xử lý ngoại lệ và assertion tùy chỉnh

Kiểm thử xử lý ngoại lệ

def test_divide_by_zero(self):
    with self.assertRaises(ZeroDivisionError):
        divide(1, 0)

Tạo assertion tùy chỉnh

def assertIsPositive(self, value):
    self.assertTrue(value > 0, f'{value} is not positive')

6. Tính năng phát hiện kiểm thử của unittest

Cách sử dụng tính năng phát hiện kiểm thử

python -m unittest discover

python -m unittest discover -s tests -p "test_*.py"

7. Mẹo nâng cao hiệu năng khi dùng unittest

Tối ưu hóa I/O tệp

from io import StringIO

class TestFileOperations(unittest.TestCase):

    def test_write_to_memory(self):
        output = StringIO()
        output.write('Hello, World!')
        self.assertEqual(output.getvalue(), 'Hello, World!')

Sử dụng mock

from unittest.mock import MagicMock

class TestApiCall(unittest.TestCase):

    def test_api_response(self):
        mock_api = MagicMock(return_value={'status': 'success'})
        response = mock_api()
        self.assertEqual(response['status'], 'success')

8. Tổng kết và các bước tiếp theo

Tóm tắt các điểm chính

• Cách sử dụng cơ bản: Kế thừa unittest.TestCase và tận dụng các phương thức assertion để tạo bài kiểm thử.
• setUp() / tearDown(): Gom và quản lý các xử lý chung trước và sau mỗi bài kiểm thử để cải thiện khả năng tái sử dụng và tính dễ đọc của mã.
• Sử dụng mock: Có thể kiểm thử chức năng mà không phụ thuộc vào tài nguyên bên ngoài, nhờ đó hiệu quả kiểm thử được cải thiện đáng kể.
• Test discovery: Tính năng hữu ích giúp đơn giản hóa việc quản lý kiểm thử trong các dự án lớn.
• Các kỹ thuật cải thiện hiệu năng: Nhờ xử lý trong bộ nhớ và sử dụng mock, có thể rút ngắn thời gian chạy kiểm thử.

Các bước tiếp theo