62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
from django.test import TestCase
|
|
from rest_framework.test import APIClient
|
|
|
|
from apps.dashboard.models import Item, List
|
|
from apps.lyric.models import User
|
|
|
|
class BaseAPITest(TestCase):
|
|
# Helper fns
|
|
def setUp(self):
|
|
self.client = APIClient()
|
|
self.user = User.objects.create_user("test@example.com")
|
|
self.client.force_authenticate(user=self.user)
|
|
|
|
class ListDetailAPITest(BaseAPITest):
|
|
def test_returns_list_with_items(self):
|
|
list_ = List.objects.create(owner=self.user)
|
|
Item.objects.create(text="item 1", list=list_)
|
|
Item.objects.create(text="item 2", list=list_)
|
|
|
|
response = self.client.get(f"/api/lists/{list_.id}/")
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.data["id"], str(list_.id))
|
|
self.assertEqual(len(response.data["items"]), 2)
|
|
|
|
class ListItemsAPITest(BaseAPITest):
|
|
|
|
def test_can_add_item_to_list(self):
|
|
list_ = List.objects.create(owner=self.user)
|
|
|
|
response = self.client.post(
|
|
f"/api/lists/{list_.id}/items/",
|
|
{"text": "a new item"},
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 201)
|
|
self.assertEqual(Item.objects.count(), 1)
|
|
self.assertEqual(Item.objects.first().text, "a new item")
|
|
|
|
class ListsAPITest(BaseAPITest):
|
|
def test_get_returns_only_users_lists(self):
|
|
list1 = List.objects.create(owner=self.user)
|
|
Item.objects.create(text="item 1", list=list1)
|
|
other_user = User.objects.create_user("other@example.com")
|
|
List.objects.create(owner=other_user)
|
|
|
|
response = self.client.get("/api/lists/")
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(len(response.data), 1)
|
|
self.assertEqual(response.data[0]["id"], str(list1.id))
|
|
|
|
def test_post_creates_list_with_item(self):
|
|
response = self.client.post(
|
|
"/api/lists/",
|
|
{"text": "first item"},
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 201)
|
|
self.assertEqual(List.objects.count(), 1)
|
|
self.assertEqual(List.objects.first().owner, self.user)
|
|
self.assertEqual(Item.objects.first().text, "first item") |