From 67a30ce114f02ac1d2e73870da167ca0e1623e88 Mon Sep 17 00:00:00 2001
From: sosokker <sirin_29@hotmail.co.th>
Date: Tue, 7 Nov 2023 03:38:51 +0700
Subject: [PATCH] Update test to match Todoviewset

---
 backend/tasks/tests/test_todo_creation.py | 18 ++++++------------
 1 file changed, 6 insertions(+), 12 deletions(-)

diff --git a/backend/tasks/tests/test_todo_creation.py b/backend/tasks/tests/test_todo_creation.py
index 0852fbe..9912126 100644
--- a/backend/tasks/tests/test_todo_creation.py
+++ b/backend/tasks/tests/test_todo_creation.py
@@ -1,21 +1,18 @@
 from datetime import datetime, timedelta
-
 from django.urls import reverse
 from rest_framework import status
 from rest_framework.test import APITestCase
-
 from tasks.tests.utils import create_test_user, login_user
-from ..models import Todo
+from tasks.models import Todo
 
-class TodoCreateViewTests(APITestCase):
+
+class TodoViewSetTests(APITestCase):
     def setUp(self):
-
         self.user = create_test_user()
         self.client = login_user(self.user)
-        self.url = reverse("add-task")
+        self.url = reverse("todo-list")
         self.due_date = datetime.now() + timedelta(days=5)
 
-
     def test_create_valid_todo(self):
         """
         Test creating a valid task using the API.
@@ -28,7 +25,7 @@ class TodoCreateViewTests(APITestCase):
             'priority': 1,
             'difficulty': 1,
             'user': self.user.id,
-            'end_event': self.due_date,
+            'end_event': self.due_date.strftime('%Y-%m-%dT%H:%M:%S'),
         }
         response = self.client.post(self.url, data, format='json')
         self.assertEqual(response.status_code, status.HTTP_201_CREATED)
@@ -42,7 +39,6 @@ class TodoCreateViewTests(APITestCase):
         data = {
             'type': 'invalid',  # Invalid task type
         }
-
         response = self.client.post(self.url, data, format='json')
         self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
         self.assertEqual(Todo.objects.count(), 0)  # No task should be created
@@ -55,7 +51,6 @@ class TodoCreateViewTests(APITestCase):
             'title': 'Incomplete Task',
             'type': 'habit',
         }
-
         response = self.client.post(self.url, data, format='json')
         self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
         self.assertEqual(Todo.objects.count(), 0)  # No task should be created
@@ -71,9 +66,8 @@ class TodoCreateViewTests(APITestCase):
             'priority': 1,
             'difficulty': 1,
             'user': 999,  # Invalid user ID
-            'end_event': self.due_date,
+            'end_event': self.due_date.strftime('%Y-%m-%dT%H:%M:%S'),
         }
-
         response = self.client.post(self.url, data, format='json')
         self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
         self.assertEqual(Todo.objects.count(), 0)  # No task should be created