Merge pull request #9 from omidcodes/009-minor-fix-current-codes

Add some filter_backends to ModelViewset

Author: omidcodes

tasks/serializers.py

@@ -6,3 +6,12 @@ class TaskSerializer(serializers.ModelSerializer):
     class Meta:
         model = Task
         fields = "__all__"
+
+    def validate_title(self, value):
+        if not value.strip():
+            raise serializers.ValidationError("Title cannot be empty.")
+        return value
+
+    def create(self, validated_data):
+        task = Task.objects.create(**validated_data)
+        return task

tasks/tests/test_api.py

@@ -1,20 +1,31 @@
+from datetime import date
 import pytest
-from rest_framework.test import APIClient
 from django.urls import reverse
-from datetime import date
+from rest_framework.test import APIClient
+from tasks.models import Task
 
+@pytest.fixture
+def api_client():
+    return APIClient()
 
 @pytest.mark.django_db
-def test_create_task_api():
-    client = APIClient()
-    url = reverse("task-list")  # Adjust if you use a different route name
-    response = client.post(
-        url,
-        {
-            "title": "API Task",
+def test_create_task(api_client):
+    url = reverse("task-list")  # If you're using a DRF viewset with basename "task"
+    data = {
+            "title": "Test Task",
             "due_date": date.today(),
-        },
-        format="json",
-    )
+        }
+    
+    response = api_client.post(url, data, format="json")
     assert response.status_code == 201
-    assert response.data["title"] == "API Task"
+    assert Task.objects.count() == 1
+    assert Task.objects.first().title == "Test Task"
+
+@pytest.mark.django_db
+def test_get_tasks(api_client):
+    Task.objects.create(title="Task 1", due_date= date.today())
+    Task.objects.create(title="Task 2", due_date= date.today())
+    url = reverse("task-list")
+    response = api_client.get(url)
+    assert response.status_code == 200
+    assert len(response.data) == 2

tasks/views.py

@@ -1,4 +1,6 @@
 from rest_framework import viewsets
+from rest_framework import filters
+
 from tasks.models import Task
 from tasks.serializers import TaskSerializer
 
@@ -12,3 +14,6 @@ def home_view(request):
 class TaskViewSet(viewsets.ModelViewSet):
     queryset = Task.objects.all().order_by("-created_at")
     serializer_class = TaskSerializer
+
+    filter_backends = [filters.SearchFilter]
+    search_fields = ["title"]