Base
This commit is contained in:
@ -0,0 +1,30 @@
|
||||
from django.urls import reverse
|
||||
from django.contrib.auth import get_user_model
|
||||
from rest_framework import status
|
||||
from rest_framework.test import APITestCase
|
||||
from your_app.models import Suggestion
|
||||
|
||||
User = get_user_model()
|
||||
|
||||
class SuggestionTests(APITestCase):
|
||||
def setUp(self):
|
||||
self.user = User.objects.create_user(username='testuser', password='testpassword123')
|
||||
self.client.force_authenticate(user=self.user)
|
||||
|
||||
def test_create_suggestion(self):
|
||||
url = reverse('suggestion-list-create')
|
||||
data = {
|
||||
'url': 'https://example.com',
|
||||
'description': 'This is a test suggestion'
|
||||
}
|
||||
response = self.client.post(url, data, format='json')
|
||||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||||
self.assertEqual(Suggestion.objects.count(), 1)
|
||||
self.assertEqual(Suggestion.objects.get().url, 'https://example.com')
|
||||
|
||||
def test_list_suggestions(self):
|
||||
Suggestion.objects.create(url='https://example.com', suggested_by=self.user)
|
||||
url = reverse('suggestion-list-create')
|
||||
response = self.client.get(url)
|
||||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||||
self.assertEqual(len(response.data), 1)
|
||||
Reference in New Issue
Block a user