Compare commits
No commits in common. "6e2ddad05937c78749cc87890a3d24138e2bfb13" and "2eec63daf6a779e389180a5076cbdab68c3d5aa5" have entirely different histories.
6e2ddad059
...
2eec63daf6
|
@ -7,93 +7,4 @@ from ..models import Puppy
|
|||
from ..serializers import PuppySerializer
|
||||
|
||||
# initialize the APIClient app
|
||||
client = Client()
|
||||
|
||||
class GetAllPuppiesTest(TestCase):
|
||||
"""
|
||||
Test module for GET all puppies API
|
||||
"""
|
||||
|
||||
def setUp(self):
|
||||
Puppy.objects.create(
|
||||
name='Casper', age=3, breed='Bull Dog', color='Black')
|
||||
Puppy.objects.create(
|
||||
name='Muffin', age=1, breed='Gradane', color='Brown')
|
||||
Puppy.objects.create(
|
||||
name='Rambo', age=2, breed='Labrador', color='Black')
|
||||
Puppy.objects.create(
|
||||
name='Ricky', age=6, breed='Labrador', color='Brown')
|
||||
|
||||
def test_get_all_puppies(self):
|
||||
# get API response
|
||||
response = client.get(reverse('get_post_puppies'))
|
||||
# get data from db
|
||||
puppies = Puppy.objects.all()
|
||||
serializer = PuppySerializer(puppies, many=True)
|
||||
self.assertEqual(response.data, serializer.data)
|
||||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||||
|
||||
class GetSinglePuppyTest(TestCase):
|
||||
""" Test module for GET single puppy API """
|
||||
|
||||
def setUp(self):
|
||||
self.casper = Puppy.objects.create(
|
||||
name='Casper', age=3, breed='Bull Dog', color='Black')
|
||||
self.muffin = Puppy.objects.create(
|
||||
name='Muffin', age=1, breed='Gradane', color='Brown')
|
||||
self.rambo = Puppy.objects.create(
|
||||
name='Rambo', age=2, breed='Labrador', color='Black')
|
||||
self.ricky = Puppy.objects.create(
|
||||
name='Ricky', age=6, breed='Labrador', color='Brown')
|
||||
|
||||
def test_get_valid_single_puppy(self):
|
||||
response = client.get(
|
||||
reverse('get_delete_update_puppy', kwargs={'pk': self.rambo.pk}))
|
||||
puppy = Puppy.objects.get(pk=self.rambo.pk)
|
||||
serializer = PuppySerializer(puppy)
|
||||
self.assertEqual(response.data, serializer.data)
|
||||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||||
|
||||
def test_get_invalid_single_puppy(self):
|
||||
response = client.get(
|
||||
reverse('get_delete_update_puppy', kwargs={'pk': 30}))
|
||||
self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)
|
||||
|
||||
|
||||
class CreateNewPuppyTest(TestCase):
|
||||
"""
|
||||
Test module for inserting a new puppy
|
||||
1. inserting a valid puppy
|
||||
2. inserting a invalid puppy
|
||||
"""
|
||||
|
||||
def setUp(self):
|
||||
self.valid_payload = {
|
||||
'name': 'Muffin',
|
||||
'age': 4,
|
||||
'breed': 'Pamerion',
|
||||
'color': 'White'
|
||||
}
|
||||
# Why this syntax works?
|
||||
self.invalid_payload = {
|
||||
'name': '',
|
||||
'age': 4,
|
||||
'breed': 'Pamerion',
|
||||
'color': 'White'
|
||||
}
|
||||
|
||||
def test_create_valid_puppy(self):
|
||||
response = client.post(
|
||||
reverse('get_post_puppies'),
|
||||
data=json.dumps(self.valid_payload),
|
||||
content_type='application/json'
|
||||
)
|
||||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||||
|
||||
def test_create_invalid_puppy(self):
|
||||
response = client.post(
|
||||
reverse('get_post_puppies'),
|
||||
data=json.dumps(self.invalid_payload),
|
||||
content_type='application/json'
|
||||
)
|
||||
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
|
||||
client = Client()
|
|
@ -7,7 +7,7 @@ from rest_framework import status
|
|||
from .models import Puppy
|
||||
from .serializers import PuppySerializer
|
||||
|
||||
@api_view(['GET', 'DELETE', 'UPDATE'])
|
||||
@api_view(['GET', 'DELETE', 'PUT'])
|
||||
def get_delete_update_puppy(request, pk):
|
||||
try:
|
||||
puppy = Puppy.objects.get(pk=pk)
|
||||
|
@ -16,9 +16,7 @@ def get_delete_update_puppy(request, pk):
|
|||
|
||||
# get details of a single puppy
|
||||
if request.method == 'GET':
|
||||
# we get the puppy using an ID. Run the tests to ensure they all pass.
|
||||
serializer = PuppySerializer(puppy)
|
||||
return Response(serializer.data)
|
||||
return Response({})
|
||||
# delete a single puppy
|
||||
elif request.method == 'DELETE':
|
||||
return Response({})
|
||||
|
@ -30,19 +28,7 @@ def get_delete_update_puppy(request, pk):
|
|||
def get_post_puppies(request):
|
||||
# get all puppies
|
||||
if request.method == 'GET':
|
||||
puppies = Puppy.objects.all()
|
||||
serializer = PuppySerializer(puppies, many=True) # generate a nested list of 'edit' items
|
||||
return Response(serializer.data)
|
||||
return Response({})
|
||||
# insert a new record for a puppy
|
||||
elif request.method == 'POST':
|
||||
data = {
|
||||
'name': request.data.get('name'),
|
||||
'age': int(request.data.get('age')),
|
||||
'breed': request.data.get('breed'),
|
||||
'color': request.data.get('color')
|
||||
}
|
||||
serializer = PuppySerializer(data=data)
|
||||
if serializer.is_valid():
|
||||
serializer.save()
|
||||
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
||||
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
||||
return Response({})
|
Loading…
Reference in New Issue