使用 Django 和 HTMX 建立 To-Do 應用程式 - 使用 TDD 新增 Todo 模型部分

這是我們關於如何使用 HTMX 和 Django 建立 Todo 應用程式系列的第二篇文章。按此處查看第 1 部分。

在第 2 部分中，我們將建立 Todo 模型並透過單元測試實現其基本功能。

創建 Todo 模型

在 models.py 中，我們建立 Todo 模型及其基本屬性。我們希望將 Todo 專案與 UserProfile 關聯，以便使用者只能看到自己的專案。待辦事項還將有一個標題和一個布林屬性 is_completed。我們對 Todo 模型有很多未來的想法，例如除了已完成或未開始之外還可以將任務設置為“正在進行”，以及截止日期，但那是以後的事了。現在讓我們保持簡單，盡快在螢幕上顯示一些內容。

注意：在現實世界的應用程式中，我們可能應該考慮使用 UUID 作為 UserProfile 和 Todo 模型的主鍵，但我們現在會保持簡單。

# core/models.py

from django.contrib.auth.models import AbstractUser
from django.db import models  # 



<p>讓我們運行新模型的遷移：<br>
</p>

<pre class="brush:php;toolbar:false">❯ uv run python manage.py makemigrations
Migrations for 'core':
  core/migrations/0002_todo.py
    + Create model Todo
❯ uv run python manage.py migrate
Operations to perform:
  Apply all migrations: admin, auth, contenttypes, core, sessions
Running migrations:
  Applying core.0002_todo... OK

寫我們的第一個測試

讓我們為我們的專案寫第一個測驗。我們希望確保使用者只能看到自己的待辦事項，而看不到其他使用者的項目。

為了幫助我們編寫測試，我們將向我們的專案 model-bakery 新增新的開發依賴項，這簡化了建立虛擬 Django 模型實例的過程。我們還將添加 pytest-django。

❯ uv add model-bakery pytest-django --dev
Resolved 27 packages in 425ms
Installed 2 packagez in 12ms
 + model-bakery==1.20.0
 + pytest-django==4.9.0

在 pyproject.toml 中，我們需要透過在檔案末尾新增一些行來設定 pytest：

# pyproject.toml

# NEW
[tool.pytest.ini_options]
DJANGO_SETTINGS_MODULE = "todomx.settings"
python_files = ["test_*.py", "*_test.py", "testing/python/*.py"]

現在讓我們來寫第一個測試，以確保使用者只能存取自己的待辦事項。

# core/tests/test_todo_model.py

import pytest


@pytest.mark.django_db
class TestTodoModel:
    def test_todo_items_are_associated_to_users(self, make_todo, make_user):
        [user1, user2] = make_user(_quantity=2)

        for i in range(3):
            make_todo(user=user1, title=f"user1 todo {i}")

        make_todo(user=user2, title="user2 todo")

        assert {todo.title for todo in user1.todos.all()} == {
            "user1 todo 0",
            "user1 todo 1",
            "user1 todo 2",
        }

        assert {todo.title for todo in user2.todos.all()} == {"user2 todo"}

我們使用 pytest 中的 conftest.py 檔案來存放我們計劃在測試中使用的所有裝置。 model_bakery 函式庫讓使用最少的樣板檔案建立 UserProfile 和 Todo 的實例變得簡單。

#core/tests/conftest.py

import pytest
from model_bakery import baker


@pytest.fixture
def make_user(django_user_model):
    def _make_user(**kwargs):
        return baker.make("core.UserProfile", **kwargs)

    return _make_user


@pytest.fixture
def make_todo(make_user):
    def _make_todo(user=None, **kwargs):
        return baker.make("core.Todo", user=user or make_user(), **kwargs)

    return _make_todo

讓我們執行測試！

❯ uv run pytest
Test session starts (platform: darwin, Python 3.12.8, pytest 8.3.4, pytest-sugar 1.0.0)
django: version: 5.1.4, settings: todomx.settings (from ini)
configfile: pyproject.toml
plugins: sugar-1.0.0, django-4.9.0
collected 1 item

 core/tests/test_todo_model.py ✓                                                                                                                                   100% ██████████

Results (0.25s):
       1 passed

註冊 Todos 的管理頁面

最後，我們可以為其註冊一個管理頁面：

# core/admin.py

from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from .models import Todo, UserProfile # 



<p>我們現在可以從管理員添加一些待辦事項！ </p>

<p><img src="/static/imghwm/default1.png" data-src="https://img.php.cn/upload/article/000/000/000/173594553634650.jpg?x-oss-process=image/resize,p_40" class="lazy" alt="Creating a To-Do app with Django and HTMX - Part Adding the Todo model with TDD"></p>

<p>如果你想查看第 2 部分結束之前的整個程式碼，可以在 Github 的第 02 部分分支上查看。 </p>


          

            
        

以上是使用 Django 和 HTMX 建立 To-Do 應用程式 - 使用 TDD 新增 Todo 模型部分的詳細內容。更多資訊請關注PHP中文網其他相關文章！